From 42960c66b7fa659f658078bcb1b6e4ae5ce4a3ba Mon Sep 17 00:00:00 2001 From: maglore9900 Date: Fri, 4 Oct 2024 15:32:26 -0400 Subject: [PATCH] Update agent.py --- modules/agent.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/modules/agent.py b/modules/agent.py index 890ff15..2c185d2 100644 --- a/modules/agent.py +++ b/modules/agent.py @@ -211,8 +211,9 @@ class Agent: agent_out = answer.get('agent_out') output_value = agent_out.return_values.get('output', None) max = self.llm.invoke(self.char_prompt.format(query=output_value)) - # print(f"max: {max.content}") return {"agent_out": max.content} + # for chunk in self.llm.stream(self.char_prompt.format(query=output_value)): + # yield(chunk) async def rag_final_answer(self, state: list): print("> rag final_answer")