- Fix topics extract

- Add schemas to extraction
This commit is contained in:
Cyber MacGeddon 2024-10-27 00:12:01 +01:00
parent 1e137768ca
commit 2caa4fc1ca
2 changed files with 12 additions and 2 deletions

View file

@ -28,7 +28,7 @@ class Relationship:
@dataclasses.dataclass
class Topic:
topic: str
name: str
definition: str
class PromptClient(BaseClient):
@ -118,7 +118,7 @@ class PromptClient(BaseClient):
)
return [
Topic(topic=d["topic"], definition=d["definition"])
Topic(name=d["topic"], definition=d["definition"])
for d in topics
]