Spaces:
Sleeping
Sleeping
Update agent/planning_agent.py
Browse files- agent/planning_agent.py +2 -3
agent/planning_agent.py
CHANGED
|
@@ -64,8 +64,7 @@ def initialize_planning_agent(llm_instance, chat_memory_instance, query_memory_i
|
|
| 64 |
name="compose_response",
|
| 65 |
func=compose_response,
|
| 66 |
description="Use this to only format the response. After this step, return the formatted response to main.py"
|
| 67 |
-
)
|
| 68 |
-
ltool
|
| 69 |
]
|
| 70 |
|
| 71 |
system_prompt = """You are an efficient AI planning agent. Follow these rules strictly:
|
|
@@ -193,7 +192,7 @@ def execute(query):
|
|
| 193 |
query_memory.memories['original_query'] = query
|
| 194 |
response = agent.run(
|
| 195 |
f"Process this user query: {query}")
|
| 196 |
-
|
| 197 |
return response
|
| 198 |
except Exception as e:
|
| 199 |
logger.error(f"Error in planning agent: {str(e)}")
|
|
|
|
| 64 |
name="compose_response",
|
| 65 |
func=compose_response,
|
| 66 |
description="Use this to only format the response. After this step, return the formatted response to main.py"
|
| 67 |
+
)
|
|
|
|
| 68 |
]
|
| 69 |
|
| 70 |
system_prompt = """You are an efficient AI planning agent. Follow these rules strictly:
|
|
|
|
| 192 |
query_memory.memories['original_query'] = query
|
| 193 |
response = agent.run(
|
| 194 |
f"Process this user query: {query}")
|
| 195 |
+
# clearml_callback.flush_tracker(name="Planning agent", finish=True)
|
| 196 |
return response
|
| 197 |
except Exception as e:
|
| 198 |
logger.error(f"Error in planning agent: {str(e)}")
|