Spaces:
Sleeping
Sleeping
File size: 3,040 Bytes
c52d511 39e9071 5724962 5952adf c52d511 39e9071 5952adf 39e9071 5952adf 3792131 5724962 67d97f3 5952adf ec08c09 5724962 ec08c09 268976d ecb2900 3792131 69dd942 5952adf 67d97f3 5952adf 67d97f3 c52d511 39e9071 67d97f3 ff69472 5952adf c52d511 5952adf 67d97f3 |
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 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 |
# tools/final_answer.py
from smolagents import Tool
from typing import Optional, Dict, Any
import json
import logging
logger = logging.getLogger(__name__)
class FinalAnswerTool(Tool):
"""Tool for providing final answers to user queries with improved error handling and validation."""
name = "final_answer"
description = "Tool for providing the final answer to the agent's task with structured output"
inputs: Dict[str, Any] = {
"answer": {
"type": "string",
"description": "The final answer to be returned in JSON format"
}
}
output_type = "string"
def __init__(self, description: Optional[str] = None):
super().__init__()
self.description = description or self.description
def validate_json(self, answer: str) -> bool:
"""Validate if the answer is proper JSON format."""
try:
if isinstance(answer, str):
json.loads(answer)
return True
except json.JSONDecodeError:
return False
def format_response(self, answer: str) -> str:
"""Format the response to ensure consistent structure."""
try:
if isinstance(answer, str):
# Try to parse as JSON first
if self.validate_json(answer):
return answer
# If not JSON, create a structured response
return json.dumps({
'clean_text': answer,
'summary': '',
'sentiment': '',
'topics': ''
})
# If answer is already a dict, convert to JSON
if isinstance(answer, dict):
return json.dumps(answer)
raise ValueError("Invalid answer format")
except Exception as e:
logger.error(f"Error formatting response: {str(e)}")
return json.dumps({
'error': str(e),
'clean_text': str(answer),
'summary': '',
'sentiment': '',
'topics': ''
})
def forward(self, answer: str) -> str:
"""Process and return the final answer with improved error handling.
Args:
answer: The answer text to be returned
Returns:
str: The processed answer in JSON format
"""
try:
return self.format_response(answer)
except Exception as e:
logger.error(f"Error in forward method: {str(e)}")
return json.dumps({
'error': str(e),
'clean_text': 'An error occurred while processing the response',
'summary': '',
'sentiment': '',
'topics': ''
})
def __call__(self, answer: str) -> str:
"""Alias for forward method to maintain compatibility"""
return self.forward(answer) |