Skip to content

Commit

Permalink
Update agent.py
Browse files Browse the repository at this point in the history
  • Loading branch information
maglore9900 committed Oct 4, 2024
1 parent a213f30 commit 42960c6
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion modules/agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -211,8 +211,9 @@ async def respond(self, answer: str):
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")
Expand Down

0 comments on commit 42960c6

Please sign in to comment.