dorivanfernandes/convert_data_to_insert_sql

1import yaml2import os3from smolagents import GradioUI, CodeAgent, HfApiModel4 5# Get current directory path6CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))7 8from tools.web_search import DuckDuckGoSearchTool as WebSearch9from tools.visit_webpage import VisitWebpageTool as VisitWebpage10from tools.data_to_sql import SimpleTool as DataToSql11from tools.final_answer import FinalAnswerTool as FinalAnswer12 13 14 15model = HfApiModel(16model_id='Qwen/Qwen2.5-Coder-32B-Instruct',17provider=None,18)19 20web_search = WebSearch()21visit_webpage = VisitWebpage()22data_to_sql = DataToSql()23final_answer = FinalAnswer()24 25 26with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:27    prompt_templates = yaml.safe_load(stream)28 29agent = CodeAgent(30    model=model,31    tools=[web_search, visit_webpage, data_to_sql],32    managed_agents=[],33    max_steps=10,34    verbosity_level=2,35    grammar=None,36    planning_interval=None,37    name=None,38    description=None,39    prompt_templates=prompt_templates40)41if __name__ == "__main__":42    GradioUI(agent).launch()43