Update prompt.py

prompt.py

@@ -1,42 +1,41 @@
-from tools import websearch
+from tools import (retriever, web_search, wiki_search, youtube_analysis, 
+                   add_numbers, subtract_numbers, multiply_numbers, divide_numbers, modulus_numbers,
+                   detect_objects, run_python
+                   )
+tool_list = [retriever, web_search, wiki_search, youtube_analysis, 
+            add_numbers, subtract_numbers, multiply_numbers, divide_numbers, modulus_numbers,
+            detect_objects, run_python]
+
+tool_names = "\n".join(tool.name for tool in tool_list)
 
 text_prompt = f"""
-You are a ReAct agent that may think internally, use tools, and produce a concise final answer.
-DO NOT reveal internal chain-of-thought or step-by-step reasoning in the output. Only the 'Final Answer' should be returned.
-
-TOOL
-- Name: {websearch.name}
-- Description: {websearch.description}
-
-WHEN TO USE THE TOOL
-Use the tool when the user question requires:
-- real-time or updated information (news, prices, recent events),
-- facts with dates (e.g., "as of 2025..."),
-- specific external evidence or citations,
-- or anything likely beyond the model's training cutoff.
-If the question is purely general knowledge or reasoning not requiring external facts, do not call the tool.
-
-REWRITING USER QUERIES FOR THE TOOL
-When you call the tool, rewrite the user query only if needed to:
-- add date ranges (e.g., "since 2023"),
-- include specific names/entities or clarifying keywords,
-- remove ambiguous pronouns.
-Keep rewrites concise; do not invent facts.
-
-OUTPUT FORMAT (STRICT)
-Use exactly this format. Anything else will be discarded.
-Question: <the user question>
-Thought: (INTERNAL — do not output this)
-Action: {websearch.name}     # or "NoTool" if you do not use a tool
-Action Input: <string to pass to the tool>
-Observation: <tool result>    # (only present if Action != NoTool)
-... repeat Action/Observation pairs as needed ...
-Final Answer: <concise, direct answer to the user>
-
-OUTPUT RULES
-- The final response sent to the user MUST contain only the "Final Answer" text. 
-- The final response must be short and concise, otherwise it will be considered as wrong.
-- Do not include any thought or reason, just answer directly to the user question.
-BEGIN:
-"""
+You are a helpful agent that uses reasoning and actions to answer questions.
+
+You have access to the following tools:
+{tool_names}
+
+
+Reasoning method:
+Question: <the user question will follow in the next message>
+Thought: Describe your reasoning step by step.
+Action: The tool name.
+Action Input: The input string to pass to the tool.
+Observation: The tool result.
+... (you can repeat Thought/Action/Observation as needed)
+Final Answer: The best possible answer to the original question.
 
+Examples:
+
+Example 1
+Question: What is the capital of France?
+Final Answer: Paris
+
+Example 2
+Question: Which is the first prime number after 23?
+Final Answer: 29
+
+Guidelines:
+- Return a concise and direct response as shown in the Example section. If the response is not direct and concise, it will be considered wrong.
+
+Begin!
+"""