From 88c4c7e9691a6cbc516a2b482195814d777c6f36 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Wed, 26 Nov 2025 17:27:11 +0000 Subject: [PATCH] Fix missing params --- trustgraph-base/trustgraph/base/prompt_client.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/trustgraph-base/trustgraph/base/prompt_client.py b/trustgraph-base/trustgraph/base/prompt_client.py index 274d4834..307a118a 100644 --- a/trustgraph-base/trustgraph/base/prompt_client.py +++ b/trustgraph-base/trustgraph/base/prompt_client.py @@ -112,7 +112,7 @@ class PromptClient(RequestResponse): timeout = timeout, ) - async def kg_prompt(self, query, kg, timeout=600): + async def kg_prompt(self, query, kg, timeout=600, streaming=False, chunk_callback=None): return await self.prompt( id = "kg-prompt", variables = { @@ -123,9 +123,11 @@ class PromptClient(RequestResponse): ] }, timeout = timeout, + streaming = streaming, + chunk_callback = chunk_callback, ) - async def document_prompt(self, query, documents, timeout=600): + async def document_prompt(self, query, documents, timeout=600, streaming=False, chunk_callback=None): return await self.prompt( id = "document-prompt", variables = { @@ -133,6 +135,8 @@ class PromptClient(RequestResponse): "documents": documents, }, timeout = timeout, + streaming = streaming, + chunk_callback = chunk_callback, ) async def agent_react(self, variables, timeout=600, streaming=False, chunk_callback=None):