apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-18 17:51:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Topics is working, added a test

Browse source
This commit is contained in:
Cyber MacGeddon 2024-10-26 16:25:38 +01:00
parent 629ab0d3ea
commit 4984a205d9
2 changed files with 30 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

12
trustgraph-base/trustgraph/clients/prompt_client.py
View file

@ -26,6 +26,11 @@ class Relationship:
o: str
o_entity: str
@dataclasses.dataclass
class Topic:
topic: str
definition: str
class PromptClient(BaseClient):
def __init__(
@ -104,13 +109,18 @@ class PromptClient(BaseClient):
def request_topics(self, chunk, timeout=300):
return self.request(
topics = self.request(
id="extract-topics",
terms={
"text": chunk
},
timeout=timeout
)
return [
Topic(topic=d["topic"], definition=d["definition"])
for d in topics
]
def request_rows(self, schema, chunk, timeout=300):