Spaces:
Sleeping
Sleeping
Update agent/planning_agent.py
Browse files- agent/planning_agent.py +1 -1
agent/planning_agent.py
CHANGED
|
@@ -189,7 +189,7 @@ def execute(query):
|
|
| 189 |
query_memory.memories['original_query'] = query
|
| 190 |
response = agent.run(
|
| 191 |
f"Process this user query: {query}")
|
| 192 |
-
clearml_callback.flush_tracker(
|
| 193 |
return response
|
| 194 |
except Exception as e:
|
| 195 |
logger.error(f"Error in planning agent: {str(e)}")
|
|
|
|
| 189 |
query_memory.memories['original_query'] = query
|
| 190 |
response = agent.run(
|
| 191 |
f"Process this user query: {query}")
|
| 192 |
+
clearml_callback.flush_tracker(name="Planning agent", finish=True)
|
| 193 |
return response
|
| 194 |
except Exception as e:
|
| 195 |
logger.error(f"Error in planning agent: {str(e)}")
|