Update agent.py
Browse files
agent.py
CHANGED
|
@@ -31,12 +31,12 @@ llm = ChatGroq(
|
|
| 31 |
)
|
| 32 |
llm_with_tools = llm.bind_tools(tools)
|
| 33 |
|
| 34 |
-
|
| 35 |
system_prompt = SystemMessage(
|
| 36 |
content = text_prompt
|
| 37 |
)
|
| 38 |
query = agent_state["messages"][-1] # HumanMessage
|
| 39 |
-
response =
|
| 40 |
return {"messages": [response]}
|
| 41 |
|
| 42 |
graph_builder = StateGraph(State)
|
|
|
|
| 31 |
)
|
| 32 |
llm_with_tools = llm.bind_tools(tools)
|
| 33 |
|
| 34 |
+
def ask_agent(agent_state: State):
|
| 35 |
system_prompt = SystemMessage(
|
| 36 |
content = text_prompt
|
| 37 |
)
|
| 38 |
query = agent_state["messages"][-1] # HumanMessage
|
| 39 |
+
response = llm_with_tools.invoke(query.content)
|
| 40 |
return {"messages": [response]}
|
| 41 |
|
| 42 |
graph_builder = StateGraph(State)
|