BestLocalModelAgent / tools /generate_recommendation_report.py
maubilli's picture
Upload agent
65d450b verified
Raw History Blame Contribute Delete
1.15 kB
from smolagents import Tool
from typing import Any, Optional
class SimpleTool(Tool):
name = "generate_recommendation_report"
description = "Generates the complete formatted hardware analysis, recommendation report, and live Hugging Face model list."
inputs = {'focus': {'type': 'string', 'nullable': True, 'description': "The user's focus preference: 'balanced', 'coding', 'reasoning', or 'lightweight'."}, 'force_refresh': {'type': 'boolean', 'nullable': True, 'description': 'If True, fetches fresh model data from Hugging Face.'}}
output_type = "string"
def forward(self, focus: 'str' = 'balanced', force_refresh: 'bool' = False) -> 'str':
"""Generates the complete formatted hardware analysis, recommendation report, and live Hugging Face model list.
Args:
focus: The user's focus preference: 'balanced', 'coding', 'reasoning', or 'lightweight'.
force_refresh: If True, fetches fresh model data from Hugging Face.
"""
from agents.bestLocalModel import format_recommendation_report
return format_recommendation_report(focus=focus, force_refresh=force_refresh)