sahanacp commited on
Commit
e417ed2
·
verified ·
1 Parent(s): 96c0fcf

Update agent/planning_agent.py

Browse files
Files changed (1) hide show
  1. agent/planning_agent.py +4 -2
agent/planning_agent.py CHANGED
@@ -39,6 +39,8 @@ def initialize_planning_agent(llm_instance, chat_memory_instance, query_memory_i
39
  product_review_agent.initialize_product_review_agent(llm, chat_memory)
40
  generic_agent.initialize_generic_agent(llm, chat_memory)
41
  # composer_agent.initialize_composer_agent(llm, memory)
 
 
42
 
43
  tools = [
44
  Tool(
@@ -60,10 +62,10 @@ def initialize_planning_agent(llm_instance, chat_memory_instance, query_memory_i
60
  name="compose_response",
61
  func=compose_response,
62
  description="Use this to only format the response. After this step, return the formatted response to main.py"
63
- )
 
64
  ]
65
 
66
-
67
  system_prompt = """You are an efficient AI planning agent. Follow these rules strictly:
68
 
69
  CRITICAL INSTRUCTION:
 
39
  product_review_agent.initialize_product_review_agent(llm, chat_memory)
40
  generic_agent.initialize_generic_agent(llm, chat_memory)
41
  # composer_agent.initialize_composer_agent(llm, memory)
42
+
43
+ ltool = load_tools(["serpapi", "llm-math"], llm=agent, callbacks=clearml_callback)
44
 
45
  tools = [
46
  Tool(
 
62
  name="compose_response",
63
  func=compose_response,
64
  description="Use this to only format the response. After this step, return the formatted response to main.py"
65
+ ),
66
+ ltool
67
  ]
68
 
 
69
  system_prompt = """You are an efficient AI planning agent. Follow these rules strictly:
70
 
71
  CRITICAL INSTRUCTION: