Spaces:
Build error
Build error
Update agent/planning_agent.py
Browse files- 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:
|