|
|
|
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() |
|
|