Update agent.py
Browse files
agent.py
CHANGED
|
@@ -198,14 +198,14 @@ model_with_tools = model.bind_tools(tools)
|
|
| 198 |
|
| 199 |
def assistant(state: AgentState):
|
| 200 |
response = model_with_tools.invoke(state["messages"])
|
|
|
|
|
|
|
|
|
|
| 201 |
print("\n===== ASSISTANT RESPONSE =====")
|
| 202 |
print("TYPE:", type(response))
|
| 203 |
print("CONTENT:", response.content)
|
| 204 |
print("TOOL CALLS:", response.tool_calls)
|
| 205 |
print("===============================\n")
|
| 206 |
-
# print("--- Thinking Section ---")
|
| 207 |
-
# print(response.content[0].get("thinking").strip())
|
| 208 |
-
# print("--- End of Thinking ---\n\n")
|
| 209 |
return {
|
| 210 |
"messages": [response],
|
| 211 |
# "file_path": state["file_path"],
|
|
|
|
| 198 |
|
| 199 |
def assistant(state: AgentState):
|
| 200 |
response = model_with_tools.invoke(state["messages"])
|
| 201 |
+
print("\n===== Thinking Section =====")
|
| 202 |
+
print(response.content[0].get("thinking").strip())
|
| 203 |
+
print("===============================\n")
|
| 204 |
print("\n===== ASSISTANT RESPONSE =====")
|
| 205 |
print("TYPE:", type(response))
|
| 206 |
print("CONTENT:", response.content)
|
| 207 |
print("TOOL CALLS:", response.tool_calls)
|
| 208 |
print("===============================\n")
|
|
|
|
|
|
|
|
|
|
| 209 |
return {
|
| 210 |
"messages": [response],
|
| 211 |
# "file_path": state["file_path"],
|