apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-04-25 16:36:21 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Fix/improve command line help (#145)

Browse source 
* Make command line consistent, fix incorrect documentation.

* Improve tg-invoke-prompt help
This commit is contained in:
cybermaggedon 2024-11-08 18:14:14 +00:00 • committed by GitHub
parent f97856245c
commit ae8661fe2e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 36 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -57,13 +57,13 @@ class PromptClient(BaseClient):
output_schema=PromptResponse,
)
def request(self, id, terms, timeout=300):
def request(self, id, variables, timeout=300):
resp = self.call(
id=id,
terms={
k: json.dumps(v)
for k, v in terms.items()
for k, v in variables.items()
},
timeout=timeout
)
@ -76,7 +76,7 @@ class PromptClient(BaseClient):
defs = self.request(
id="extract-definitions",
terms={
variables={
"text": chunk
},
timeout=timeout
@ -91,7 +91,7 @@ class PromptClient(BaseClient):
rels = self.request(
id="extract-relationships",
terms={
variables={
"text": chunk
},
timeout=timeout
@ -111,7 +111,7 @@ class PromptClient(BaseClient):
topics = self.request(
id="extract-topics",
terms={
variables={
"text": chunk
},
timeout=timeout
@ -126,7 +126,7 @@ class PromptClient(BaseClient):
return self.request(
id="extract-rows",
terms={
variables={
"chunk": chunk,
"row-schema": {
"name": schema.name,
@ -148,7 +148,7 @@ class PromptClient(BaseClient):
return self.request(
id="kg-prompt",
terms={
variables={
"query": query,
"knowledge": [
{ "s": v[0], "p": v[1], "o": v[2] }
@ -162,7 +162,7 @@ class PromptClient(BaseClient):
return self.request(
id="document-prompt",
terms={
variables={
"query": query,
"documents": documents,
},