Adjusting wrong model and prompt

custom_agent.py

@@ -13,8 +13,11 @@ import os
 load_dotenv(find_dotenv())
 
 DEFAULT_PROMPT = """
-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.
-
+You are a helpful assistant tasked with answering questions using a set of tools. 
+Now, 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.
+Your answer should only start with "FINAL ANSWER: ", then follows with the answer. 
 """
 
 
@@ -130,8 +133,6 @@ class CustomAgent:
         self.embeddings = HuggingFaceEmbeddings(
             model_name="sentence-transformers/all-mpnet-base-v2"
         )
-        print(os.environ.get("SUPABASE_URL"))
-        print(os.environ.get("SUPABASE_SERVICE_KEY"))
 
         self.supabase: Client = create_client(
             os.environ.get("SUPABASE_URL"), os.environ.get("SUPABASE_SERVICE_ROLE_KEY")
@@ -146,7 +147,7 @@ class CustomAgent:
 
         # Create the agent
         self.agent = create_react_agent(
-            model="openai:gpt-4-1106-preview",
+            model="openai:gpt-4.1",
             tools=[
                 web_search,
                 add,
@@ -164,7 +165,7 @@ class CustomAgent:
         """Retriever"""
         similar_question = self.vector_store.similarity_search(query)
         return HumanMessage(
-            content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].page_content}",
+            content=f"Here I provide a similar question and answer for reference, you can use it to answer the question: \n\n{similar_question[0].page_content}",
         )
 
     def __call__(self, question: str) -> str:
@@ -191,5 +192,5 @@ class CustomAgent:
 if __name__ == "__main__":
     agent = CustomAgent()
     agent(
-        'What was the volume in m^3 of the fish bag that was calculated in the University of Leicester paper "Can Hiccup Supply Enough Fish to Maintain a Dragon\u2019s Diet?"'
+        "How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia."
     )