mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-04-27 17:36:23 +02:00
Fix/document rag (#506)
* Fix missing document RAG user/collection params * Added test
This commit is contained in:
parent
266454e75f
commit
6ac8a7c2d9
2 changed files with 83 additions and 1 deletions
|
|
@ -92,7 +92,12 @@ class Processor(FlowProcessor):
|
|||
else:
|
||||
doc_limit = self.doc_limit
|
||||
|
||||
response = await self.rag.query(v.query, doc_limit=doc_limit)
|
||||
response = await self.rag.query(
|
||||
v.query,
|
||||
user=v.user,
|
||||
collection=v.collection,
|
||||
doc_limit=doc_limit
|
||||
)
|
||||
|
||||
await flow("response").send(
|
||||
DocumentRagResponse(
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue