Update sys prompt and requirements

basic_agent.py

@@ -9,7 +9,7 @@ from llama_index.readers.web import SimpleWebPageReader
 from llama_index.core.tools.ondemand_loader_tool import OnDemandLoaderTool
 from langfuse.llama_index import LlamaIndexInstrumentor
 from llama_index.llms.ollama import Ollama
-from llama_index.core.agent.workflow import ReActAgent
+from llama_index.core.agent.workflow import ReActAgent, FunctionAgent
 
 class BasicAgent:
     def __init__(self, ollama=False, langfuse=True):
@@ -48,15 +48,21 @@ class BasicAgent:
             description="A tool for loading the content of a webpage and querying it for information",
         )
 
-        # Create Alfred with all the tools
-        self.agent = AgentWorkflow.from_tools_or_functions( # ReActAgent( 
-            wiki_spec.to_tool_list() + [search_tool], # webpage_tool does not work properly - cookies etc
+        self.agent = AgentWorkflow.from_tools_or_functions( # ReActAgent(
+            tools=[search_tool], # webpage_tool does not work properly - cookies etc
             llm=llm,
             verbose=True,
             system_prompt = (
-                "You are a helpful assistant that can search the web and wikipedia for information.\n"
-                "Only output the shortest possible answer to the question. "
-                "Avoid additional information or explanations."
+                "You are a general AI assistant. I will ask you a question. "
+                "Report your thoughts, and finish your answer with the following template: "
+                "FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number "
+                "OR as few words as possible OR a comma separated list of numbers and/or "
+                "strings. If you are asked for a number, don't use comma to write your "
+                "number neither use units such as $ or percent sign unless specified otherwise. "
+                "If you are asked for a string, don't use articles, neither abbreviations (e.g. "
+                "for cities), and write the digits in plain text unless specified otherwise. If "
+                "you are asked for a comma separated list, apply the above rules depending of "
+                "whether the element to be put in the list is a number or a string."
             )
         )
 
@@ -68,4 +74,4 @@ class BasicAgent:
         if self.langfuse:
             self.instrumentor.flush()
 
-        return response.response.content.strip()
+        return response.response.content.replace("FINAL ANSWER:", "").strip()

requirements.txt

@@ -6,4 +6,5 @@ llama_index-tools-duckduckgo
 llama_index-tools-wikipedia
 llama-index-embeddings-huggingface
 llama-index-readers-web
+llama-index-llms-ollama
 langfuse