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

Update agent.py

Browse Source
This commit is contained in:
maglore9900 2024-10-04 21:57:03 -04:00
parent 42960c66b7
commit 5366daf9c8
1 changed files with 1 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
modules/agent.py
View File

@ -215,30 +215,6 @@ class Agent:
# for chunk in self.llm.stream(self.char_prompt.format(query=output_value)): # for chunk in self.llm.stream(self.char_prompt.format(query=output_value)):
# yield(chunk) # yield(chunk)
async def rag_final_answer(self, state: list):
print("> rag final_answer")
print(f"state: {state}")
try:
#! if AgentFinish and no intermediate steps then return the answer without rag_final_answer (need to develop)
context = state.get("agent_out").return_values['output']
if not context:
context = state.get("agent_out")['answer']
if not context:
context = state.get("intermediate_steps")[-1]
except:
context = ""
if "return_values" in str(state.get("agent_out")) and state["intermediate_steps"] == []:
print("bypassing rag_final_answer")
print(f"context: {context}")
return {"agent_out": {"answer":context, "source": "Quick Response"}}
else:
prompt = f"You are a helpful assistant, Ensure the answer to user's question is in natural language, using the context provided.\n\nCONTEXT: {context}\nQUESTION: {state['input']}"
loop = asyncio.get_running_loop()
# Run the synchronous method in an executor
out = await loop.run_in_executor(None, self.final_answer_llm.invoke, prompt)
function_call = out.additional_kwargs["tool_calls"][-1]["function"]["arguments"]
return {"agent_out": function_call}
async def router(self, state): async def router(self, state):
print("> router") print("> router")
print(f"----router agent state: {state}") print(f"----router agent state: {state}")