sahanacp commited on
Commit
450c8d7
·
verified ·
1 Parent(s): b8705cb

Update agent/planning_agent.py

Browse files
Files changed (1) hide show
  1. 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
- clearml_callback.flush_tracker(name="Planning agent", finish=True)
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)}")