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

Fix error, putting flow stuff into API

Browse source
This commit is contained in:
Cyber MacGeddon 2025-05-03 10:00:16 +01:00
parent 031da4842d
commit b212dd3aae
11 changed files with 155 additions and 134 deletions
Download patch file Download diff file Expand all files Collapse all files

13
trustgraph-cli/scripts/tg-invoke-document-rag
View file

@ -13,11 +13,11 @@ default_user = 'trustgraph'
default_collection = 'default'
default_doc_limit = 10
def question(url, question, user, collection, doc_limit):
def question(url, flow_id, question, user, collection, doc_limit):
rag = Api(url)
api = Api(url)
resp = rag.document_rag(
resp = api.flow(flow_id).document_rag(
question=question, user=user, collection=collection,
doc_limit=doc_limit,
)
@ -43,12 +43,6 @@ def main():
help=f'Flow ID (default: 0000)'
)
# parser.add_argument(
# '--pulsar-api-key',
# default=default_pulsar_api_key,
# help=f'Pulsar API key',
# )
parser.add_argument(
'-q', '--question',
required=True,
@ -79,6 +73,7 @@ def main():
question(
url=args.url,
flow_id = args.flow_id,
question=args.question,
user=args.user,
collection=args.collection,