diff --git a/core/graphAgent.py b/core/graphAgent.py index f5f62ff..c6605e5 100644 --- a/core/graphAgent.py +++ b/core/graphAgent.py @@ -29,7 +29,7 @@ def __init__(self): self.workflow.add_node("perplexity_agent", perplexity_agent) self.workflow.add_node("calendar_tool", ToolNode(get_tools())) self.workflow.add_node("use_calendar_tool", calendar_tool_decider) - self.workflow.add_node("calendar_decider", calendar_desicion_agent) + self.workflow.add_node("calendar_decider", calendar_decision_agent) self.workflow.add_node("other_agent", other_agent) self.workflow.add_edge(START, "jarvis_agent") diff --git a/core/noder.py b/core/noder.py index 83719d9..d3d8d5b 100644 --- a/core/noder.py +++ b/core/noder.py @@ -52,8 +52,8 @@ def tool_agent_decider(state: GraphState): Data: {data} Your options for agents are the following: - - "perplexity": This agent has access to tools that use the perplexity API. - These tools are the following: {perplexity_tools} + - "perplexity": This agent has access to tools that use the perplexity API and tools + for doing a RAG-search. These tools are the following: {perplexity_tools} - "calendar": This agent has access to calendar tools These tools are the following: {calendar_tools} - "other": Other tools available: {other_tools} @@ -112,8 +112,8 @@ def perplexity_agent(state: GraphState): """Agent that handles tools using the perplexity api""" prompt = PromptTemplate( template= """ - Your job is to create tool_calls to tools using the perplexity API. - The tool or tools you decide + Your job is to create tool_calls to tools using the perplexity API or + to tools that do a RAG-search. The tool or tools you decide to call should help answer the users question. Here are previous messages: @@ -134,11 +134,11 @@ def perplexity_agent(state: GraphState): return {"messages": [response]} -def calendar_desicion_agent(state: GraphState): +def calendar_decision_agent(state: GraphState): """Agent that decides what to do with the calendar""" prompt = PromptTemplate( template= """ - Your job is to determine if you wich calendar related tools you need to answer the + Your job is to determine if you which calendar related tools you need to answer the jarvis agents question and answer with only the name of the option choose.