File size: 2,151 Bytes
2c23242 8fe992b 2c23242 8fe992b 2c23242 8fe992b 2c23242 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
# 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 = "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
formatted_output += (
f"**{idx}. {title}**\n"
f" - **Company:** {company}\n"
f" - **Location:** {location}\n"
f" - **🔗 [Apply Here]({link})**\n\n"
)
return formatted_output
# 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)}```"
|