Upload 2 files

agent.py

@@ -0,0 +1,142 @@
+import os
+
+from dotenv import load_dotenv
+load_dotenv() # Loads variables from .env into the environment
+
+# os.environ["LANGFUSE_HOST"] = os.getenv("LANGFUSE_HOST")
+# os.environ["LANGFUSE_PUBLIC_KEY"] = os.getenv("LANGFUSE_PUBLIC_KEY")
+# os.environ["LANGFUSE_SECRET_KEY"] = os.getenv("LANGFUSE_SECRET_KEY")
+
+# os.environ["HF_TOKEN"] = os.getenv("HF_token")
+# os.environ["GEMINI_API_KEY"] = os.getenv("GEMINI_API_KEY")
+#os.environ["TAVILY_API_KEY"] = os.getenv("TRAVILY_SECRET_KEY")
+
+from langfuse import get_client
+langfuse = get_client()
+# Verify connection
+if langfuse.auth_check():
+    print("Langfuse client is authenticated and ready!")
+else:
+    print("Authentication failed. Please check your credentials and host.")
+
+
+from smolagents import OpenAIServerModel
+gemini_model = OpenAIServerModel(
+    model_id="gemini-2.5-flash",
+    api_base="https://generativelanguage.googleapis.com/v1beta/openai/",
+    api_key=os.getenv("GEMINI_API_KEY")
+)
+
+# from smolagents import TransformersModel
+# VL_model = TransformersModel(
+#     model_id="Qwen/Qwen2.5-VL-32B-Instruct",
+#     device_map="auto",
+#     torch_dtype="auto",
+# )
+
+# coder_model = TransformersModel(
+#     model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+#     device_map="auto",
+#     torch_dtype="auto",
+# )
+
+
+#from Tools.searchtool import WebSearchTool
+from Tools.mywebpagevisit import CustomVisitWebpageTool
+from smolagents import (
+    CodeAgent,
+    WebSearchTool,
+    #WikipediaSearchTool
+)
+
+web_search = WebSearchTool()
+visit_webpage = CustomVisitWebpageTool()
+#wikipedia_search = WikipediaSearchTool()
+
+from openinference.instrumentation.smolagents import SmolagentsInstrumentor
+SmolagentsInstrumentor().instrument()
+
+web_agent = CodeAgent(
+    tools=[WebSearchTool,CustomVisitWebpageTool],
+    model= gemini_model,
+    additional_authorized_imports = ['requests', 'bs4','json'],
+    name="web_agent",
+    description="This is an agent that can do web search using 'WebSearchTool' and visit a webpage using 'CustomVisitWebpageTool'."
+)
+ 
+from Tools.ytdownload import youtube_audio_downloader,youtube_video_downloader
+
+yt_agent = CodeAgent(
+    tools=[youtube_audio_downloader,youtube_video_downloader],
+    model= gemini_model,
+    name="youtube_downloader_agent",
+    description="This is an agent that can download youtube audio file using 'youtube_audio_downloader'and youtube video file using 'youtube_video_downloader'."
+)
+
+from Tools.transcriber import transcriber
+
+transcriber_agent = CodeAgent(
+    tools=[transcriber],
+    model= gemini_model,
+    name="transcriber_agent",
+    description="This is an agent that can transcribe a local audio file using 'transcriber'.\n\nDo not remove adjectives or descriptors of ingredients if not specifically mentioned."
+)
+
+from Tools.visual_reasoner import video_reasoner, image_reasoner
+VL_agent = CodeAgent(
+    tools=[video_reasoner,image_reasoner],
+    model= gemini_model,
+    name="VL_agent",
+    description="This is an agent that can answer questions on a video using 'video_reasoner' and on a image using 'image_reasoner'."
+)
+
+excel_agent = CodeAgent(
+    tools=[],
+    model= gemini_model,
+    name="excel_agent",
+    description="This is an agent that can work with excel.",
+    add_base_tools=False,
+    additional_authorized_imports = ['pandas', 'numpy', 'math']
+)
+
+from Tools.code_reader import code_reader
+manager_agent = CodeAgent(
+    tools=[code_reader],
+    model=gemini_model,
+    managed_agents=[web_agent, yt_agent, transcriber_agent, VL_agent],
+    additional_authorized_imports=[
+        "numpy","os","pandas",'scipy','regex','math'
+    ],
+    planning_interval=5,
+    verbosity_level=2,
+    max_steps=10,
+    description= "You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."
+)
+
+# manager_agent.run(
+#     "How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia."
+# )
+
+class MyAgent:
+    def __init__(self):
+        print("BasicAgent initialized.")
+
+    def __call__(self, question: str) -> str:
+        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        try:
+            answer = manager_agent.run(question)
+            return answer
+        except Exception as e:
+            error = f"An error occurred while processing the question: {e}"
+            print(error)
+            return error
+
+
+# if __name__ == "__main__":
+#     question = """
+# What was the actual enrollment of the Malko competition in 2023?
+# """
+#     agent = MyAgent()
+#     answer = agent(question)
+#     print(f"Answer: {answer}")
+

app.py

@@ -181,4 +181,4 @@ if __name__ == "__main__":
     print("-"*(60 + len(" App Starting ")) + "\n")
 
     print("Launching Gradio Interface for Basic Agent Evaluation...")
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=True)