Spaces:
Sleeping
Sleeping
File size: 1,458 Bytes
c19d193 e53e81b 8fe992b e53e81b 9b5b26a 84f336a e53e81b 9b5b26a 8c01ffb ae7a494 e53e81b f70b4f0 13d500a 8c01ffb 84f336a e53e81b a0b8af5 e53e81b 861422e e53e81b 8c01ffb 8fe992b 84f336a e53e81b 84f336a 8c01ffb e53e81b 861422e 8fe992b e53e81b | 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 | import yaml
import os
from smolagents import GradioUI, CodeAgent, InferenceClientModel
# Get current directory path
CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
from tools.web_search import WebSearchTool as WebSearch
from tools.visit_webpage import VisitWebpageTool as VisitWebpage
from tools.suggest_menu import SimpleTool as SuggestMenu
from tools.catering_service_tool import SimpleTool as CateringServiceTool
from tools.superhero_party_theme_generator import SuperheroPartyThemeTool as SuperheroPartyThemeGenerator
from tools.final_answer import FinalAnswerTool as FinalAnswer
model = InferenceClientModel(
model_id='zai-org/GLM-5.1',
)
web_search = WebSearch()
visit_webpage = VisitWebpage()
suggest_menu = SuggestMenu()
catering_service_tool = CateringServiceTool()
superhero_party_theme_generator = SuperheroPartyThemeGenerator()
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=[web_search, visit_webpage, suggest_menu, catering_service_tool, superhero_party_theme_generator],
managed_agents=[],
max_steps=10,
verbosity_level=2,
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()
|