- Fix topics extract (#126)

- Add schemas to extraction
This commit is contained in:
cybermaggedon 2024-10-27 08:22:12 +00:00 committed by GitHub
parent 1e137768ca
commit b07d539802
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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
]