github/max_headroom
1
0
Fork 0
mirror of https://github.com/maglore9900/max_headroom.git synced 2025-06-06 19:45:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update agent.py

Browse Source
This commit is contained in:
maglore9900 2024-10-04 15:32:26 -04:00
parent a213f30084
commit 42960c66b7
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/agent.py
View File

@ -211,8 +211,9 @@ class Agent:
agent_out = answer.get('agent_out') agent_out = answer.get('agent_out')
output_value = agent_out.return_values.get('output', None) output_value = agent_out.return_values.get('output', None)
max = self.llm.invoke(self.char_prompt.format(query=output_value)) max = self.llm.invoke(self.char_prompt.format(query=output_value))
# print(f"max: {max.content}")
return {"agent_out": 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): async def rag_final_answer(self, state: list):
print("> rag final_answer") print("> rag final_answer")