|
|
import yaml
|
|
|
import os
|
|
|
from smolagents import GradioUI, CodeAgent, LiteLLMModel
|
|
|
|
|
|
|
|
|
CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
|
|
from tools.get_today_datetime import SimpleTool as GetTodayDatetime
|
|
|
from tools.final_answer import FinalAnswerTool as FinalAnswer
|
|
|
|
|
|
|
|
|
|
|
|
model = LiteLLMModel(
|
|
|
model_id='gemini/gemini-2.5-flash',
|
|
|
api_base=None,
|
|
|
)
|
|
|
|
|
|
get_today_datetime = GetTodayDatetime()
|
|
|
final_answer = FinalAnswer()
|
|
|
|
|
|
|
|
|
with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:
|
|
|
prompt_templates = yaml.safe_load(stream)
|
|
|
|
|
|
agent = CodeAgent(
|
|
|
model=model,
|
|
|
tools=[get_today_datetime],
|
|
|
managed_agents=[],
|
|
|
max_steps=20,
|
|
|
verbosity_level=1,
|
|
|
planning_interval=None,
|
|
|
name=None,
|
|
|
description=None,
|
|
|
executor_type='local',
|
|
|
executor_kwargs={},
|
|
|
max_print_outputs_length=None,
|
|
|
prompt_templates=prompt_templates
|
|
|
)
|
|
|
if __name__ == "__main__":
|
|
|
GradioUI(agent).launch()
|
|
|
|
