Change file tree

agent/agent.py → agent.py

@@ -1,13 +1,13 @@
-from smolagents import Tool, tool
+from smolagents import Tool
 import os
 import asyncio
 
-from Final_Assignment_Template.agent.ressources_getter import save_file, \
+from ressources_getter import save_file, \
     get_prompt_with_file_content, extract_text_from_file, convert_excel_to_text, delete_tmp_files
-from get_question.hf_api import get_file_by_task_id
-from Final_Assignment_Template.agent.agent_websearch import create_wikipedia_agent
-from Final_Assignment_Template.agent.agent_multimedia import create_multimedia_agent
-from Final_Assignment_Template.agent.agent_code import create_code_analyzer_agent
+from hf_api import get_file_by_task_id
+from agent_websearch import create_wikipedia_agent
+from agent_multimedia import create_multimedia_agent
+from agent_code import create_code_analyzer_agent
 from smolagents import ToolCallingAgent, LiteLLMModel
 from dotenv import load_dotenv
 

agent/agent_websearch.py → agent_websearch.py

@@ -1,13 +1,11 @@
 import wikipedia
 import requests
 from bs4 import BeautifulSoup
-import pandas as pd
-from typing import List, Dict, Any, Optional
+from typing import List, Dict, Any
 import os
 from dotenv import load_dotenv
 from smolagents import CodeAgent, LiteLLMModel, tool, DuckDuckGoSearchTool
-import re
-from datetime import datetime
+
 
 # Configuration de Wikipedia en français
 wikipedia.set_lang("fr")

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import pandas as pd
 
-from .agent.agent import BasicAgent
+from agent import BasicAgent
 
 # (Keep Constants as is)
 # --- Constants ---

requirements.txt

@@ -11,7 +11,6 @@ litellm
 python-dotenv
 wikipedia-api
 beautifulsoup4
-duckduckgo-search
 yt_dlp
 google-generativeai
 openpyxl