Edited final_answer

tools/.ipynb_checkpoints/final_answer-checkpoint.py

@@ -1,14 +1,54 @@
-from typing import Any, Optional
+# from typing import Any, Optional
+# from smolagents.tools import Tool
+
+# class FinalAnswerTool(Tool):
+#     name = "final_answer"
+#     description = "Provides a final answer to the given problem."
+#     inputs = {'answer': {'type': 'any', 'description': 'The final answer to the problem'}}
+#     output_type = "any"
+
+#     def forward(self, answer: Any) -> Any:
+#         return answer
+
+#     def __init__(self, *args, **kwargs):
+#         self.is_initialized = False
+
+
+from typing import Any
 from smolagents.tools import Tool
 
 class FinalAnswerTool(Tool):
     name = "final_answer"
-    description = "Provides a final answer to the given problem."
-    inputs = {'answer': {'type': 'any', 'description': 'The final answer to the problem'}}
-    output_type = "any"
+    description = "Formats and presents final answers in a human-readable format."
+    inputs = {'answer': {'type': 'any', 'description': 'The final answer, which could be job listings or a general response'}}
+    output_type = "string"
+
+    def forward(self, answer: Any) -> str:
+        """
+        Determines the type of answer and formats it accordingly.
+        """
+        # Case 1: If the answer is a string, return it as is
+        if isinstance(answer, str):
+            return f"📌 **Final Answer:**\n\n{answer}"
+        
+        # Case 2: If the answer is a list of job listings, format them properly
+        elif isinstance(answer, list) and all(isinstance(job, dict) for job in answer):
+            formatted_output = "**🔍 Remote Data Science Jobs in Europe**\n\n"
+
+            for idx, job in enumerate(answer, start=1):
+                title = job.get("Title", "N/A")
+                company = job.get("Company", "N/A")
+                location = job.get("Location", "Anywhere")
+                link = job.get("Link", "#")  # Provide a default link if missing
 
-    def forward(self, answer: Any) -> Any:
-        return answer
+                formatted_output += (
+                    f"**{idx}. {title}**\n"
+                    f"   - **Company:** {company}\n"
+                    f"   - **Location:** {location}\n"
+                    f"   - **🔗 [Apply Here]({link})**\n\n"
+                )
+            return formatted_output
 
-    def __init__(self, *args, **kwargs):
-        self.is_initialized = False
+        # Case 3: If it's an unknown format (list, dict, etc.), return it as a formatted string
+        else:
+            return f"📌 **Final Answer:**\n\n```{str(answer)}```"

tools/final_answer.py

@@ -1,14 +1,54 @@
-from typing import Any, Optional
+# from typing import Any, Optional
+# from smolagents.tools import Tool
+
+# class FinalAnswerTool(Tool):
+#     name = "final_answer"
+#     description = "Provides a final answer to the given problem."
+#     inputs = {'answer': {'type': 'any', 'description': 'The final answer to the problem'}}
+#     output_type = "any"
+
+#     def forward(self, answer: Any) -> Any:
+#         return answer
+
+#     def __init__(self, *args, **kwargs):
+#         self.is_initialized = False
+
+
+from typing import Any
 from smolagents.tools import Tool
 
 class FinalAnswerTool(Tool):
     name = "final_answer"
-    description = "Provides a final answer to the given problem."
-    inputs = {'answer': {'type': 'any', 'description': 'The final answer to the problem'}}
-    output_type = "any"
+    description = "Formats and presents final answers in a human-readable format."
+    inputs = {'answer': {'type': 'any', 'description': 'The final answer, which could be job listings or a general response'}}
+    output_type = "string"
+
+    def forward(self, answer: Any) -> str:
+        """
+        Determines the type of answer and formats it accordingly.
+        """
+        # Case 1: If the answer is a string, return it as is
+        if isinstance(answer, str):
+            return f"📌 **Final Answer:**\n\n{answer}"
+        
+        # Case 2: If the answer is a list of job listings, format them properly
+        elif isinstance(answer, list) and all(isinstance(job, dict) for job in answer):
+            formatted_output = "**🔍 Remote Data Science Jobs in Europe**\n\n"
+
+            for idx, job in enumerate(answer, start=1):
+                title = job.get("Title", "N/A")
+                company = job.get("Company", "N/A")
+                location = job.get("Location", "Anywhere")
+                link = job.get("Link", "#")  # Provide a default link if missing
 
-    def forward(self, answer: Any) -> Any:
-        return answer
+                formatted_output += (
+                    f"**{idx}. {title}**\n"
+                    f"   - **Company:** {company}\n"
+                    f"   - **Location:** {location}\n"
+                    f"   - **🔗 [Apply Here]({link})**\n\n"
+                )
+            return formatted_output
 
-    def __init__(self, *args, **kwargs):
-        self.is_initialized = False
+        # Case 3: If it's an unknown format (list, dict, etc.), return it as a formatted string
+        else:
+            return f"📌 **Final Answer:**\n\n```{str(answer)}```"