Update crew.py

crew.py

@@ -20,12 +20,13 @@ tracer_provider = register(
 def get_crew():
     search_tool = SerperDevTool()
     web_rag_tool = WebsiteSearchTool()
+    final_answer_tool = FinalAnswerTool()
     
     researcher = Agent(
         role="Web Research Agent",
         goal="Search the web for topic {topic} and scrape the most relevant web page.",
         backstory="As an expert web research assistant, you are search the web for topic {topic} and scraping the most relevant web page.",
-        tools=[search_tool, web_rag_tool],
+        tools=[search_tool, web_rag_tool, final_answer_tool],
         verbose=True
     )
     
@@ -41,4 +42,29 @@ def get_crew():
         planning=True,
         process = Process.sequential,
         verbose=True
-    )
+    )
+
+class FinalAnswerTool(BaseTool):
+    name: str ="Final Answer Tool"
+    description: str = ("Given a question and answer, gets the final answer.")
+    
+    def _run(question, answer) -> str:
+        prompt_template = """
+            You are given a question and context. You must **precisely** answer the question based on the context. Do not include explanations, steps, reasoning, or additional text.
+            **Question:** """ + question + """
+            **Context:** """ + answer + """
+            **Example 1:** What is the capital of France? Paris
+            **Example 2:** What is the superlative of good? Best
+            **Example 3:** What is the opposite of left? Right
+            **Answer:**:
+            """
+    
+        client = OpenAI()
+        completion = client.chat.completions.create(
+                    messages = [{"role": "user",
+                                 "content": [{"type": "text", 
+                                              "text": prompt_template}],
+                    model = "gpt-4o-mini"
+                )
+            
+        return completion.choices[0].message.content