File size: 727 Bytes
8fe992b cc003cf 8fe992b cc003cf 8fe992b cc003cf 8fe992b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from smolagents import GradioUI, {{ class_name }}, {{ agent_dict['model']['class'] }}
{% for tool in tools.values() %}
from scripts.{{ tool.name }} import {{ tool.__class__.__name__ }}
{% endfor %}
model = {{ agent_dict['model']['class'] }}()
{% for tool in tools.values() %}
{{ tool.name }} = {{ tool.__class__.__name__ }}()
{% endfor %}
agent = {{ class_name }}(
model=model,
tools=[{% for tool in tools.keys() %}{{ tool }}{% if not loop.last %}, {% endif %}{% endfor %}],
{% for attribute_name, value in agent_dict.items() if attribute_name not in ["model", "tools", "prompt_templates"] %}
{{ attribute_name }}={{ value }},
{% endfor %}
prompts_path='./prompts.yaml'
)
GradioUI(agent).launch()
|