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

Graph RAG and document RAG working and configurable

Browse source
This commit is contained in:
Cyber MacGeddon 2025-03-12 23:46:14 +00:00
parent 5160d5bf2f
commit c8df9a4fa2
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -44,7 +44,7 @@ def main():
# ) # )
parser.add_argument( parser.add_argument(
'-q', '--query', '-q', '--question',
required=True, required=True,
help=f'Question to answer', help=f'Question to answer',
) )

2
trustgraph-flow/trustgraph/gateway/document_rag.py
View file

@ -23,7 +23,7 @@ class DocumentRagRequestor(ServiceRequestor):
query=body["query"], query=body["query"],
user=body.get("user", "trustgraph"), user=body.get("user", "trustgraph"),
collection=body.get("collection", "default"), collection=body.get("collection", "default"),
doc_limit=body.get("doc-limit", 20), doc_limit=int(body.get("doc-limit", 20)),
) )
def from_response(self, message): def from_response(self, message):