Update app.py

app.py

@@ -2,12 +2,11 @@
 import os
 import gradio as gr
 import requests
-from smolagents import DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool, LiteLLMModel, CodeAgent, tool
+from smolagents import DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool, LiteLLMModel, CodeAgent, tool , OpenAIServerModel
 from pathlib import Path
 import pathlib
 import tempfile
 import PyPDF2   
-from opik import track
 import pytesseract             
 from PIL import Image       
 from smolagents.tools import PipelineTool, Tool
@@ -16,8 +15,6 @@ import pandas as pd
 from tabulate import tabulate  # pragma: no cover – fallback path
 import re
 import opik
-import litellm   
-from litellm.integrations.opik.opik import OpikLogger
 import time
 from tenacity import retry, stop_after_attempt, wait_exponential, retry_if_exception_type, before_sleep_log
 import logging
@@ -183,12 +180,6 @@ def analyze_image_tool(image_path: str) -> str:
         return f"Error analyzing image: {e}"
 
 
-litellm.callbacks = [OpikLogger()]
-llm = LiteLLMModel(
-    model_id="groq/llama-3.3-70b-versatile",
-    client=litellm,
-)
-
 # Define the retry decorator to handle rate limits
 @retry(
     retry=retry_if_exception_type(litellm.RateLimitError),
@@ -209,7 +200,7 @@ def call_llm_with_retry(agent, question: str) -> str:
 class BasicAgent:
     def __init__(self):
         self.agent = CodeAgent(
-            model=llm,
+            model=OpenAIServerModel(model_id="gpt-4o"),
             tools=[DuckDuckGoSearchTool(), WikipediaSearchTool(), ExcelToTextTool(), pdf_to_text_tool, analyze_image_tool],
             add_base_tools=True,
             additional_authorized_imports=['pandas', 'numpy', 'csv', 'subprocess']