FIX: removed LocalAgent

agent.py

@@ -3,10 +3,6 @@
 import os
 from typing import Optional
 import pandas as pd
-import torch
-
-# Local agent specific
-from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 
 # Smolagents imports
 from smolagents import (
@@ -191,72 +187,3 @@ class GeminiAgent:
             )
             
         return self.gemini_agent.run(prompt)
-
-class LocalAgent:
-    def __init__(self):
-        checkpoint = "Qwen/Qwen2.5-7B-Instruct"
-        quantized_model_dir = "./quantized_model"
-        
-        # Define the quantized configuration
-        bnb_config = BitsAndBytesConfig(
-            load_in_4bit = True,
-            bnb_4bit_quant_type = "nf4",
-            bnb_4bit_compute_dtype = torch.bfloat16,
-            bnb_4bit_use_double_quant = True,
-        )
-        
-        # Load quantized model and tokenizer
-        temp_model = AutoModelForCausalLM.from_pretrained(
-            checkpoint,
-            quantization_config = bnb_config,
-            device_map="auto" # use multiple GPUs if available
-        )        
-        temp_tokenizer = AutoTokenizer.from_pretrained(checkpoint)
-        
-        # Save the model in local path (seems like it's the only way to make it work with TransformersModel)
-        temp_model.save_pretrained(quantized_model_dir)
-        temp_tokenizer.save_pretrained(quantized_model_dir)
-        
-        self.system_prompt = SYSTEM_PROMPT
-        
-        self.model = TransformersModel(
-            model_path = quantized_model_dir, 
-            temperature = 0.1,    
-            top_p = 0.95,   
-            device_map = "auto", 
-            max_new_tokens = 8196    # https://github.com/huggingface/smolagents/issues/414#:~:text=Running%20with%20TransformersModel%20does%20not%20work
-        )
-        self.tools = AGENT_TOOLS 
-        
-        self.local_agent = CodeAgent(
-            model=self.model,
-            tools=tools,
-            add_base_tools=True,      # probably redundant, but it does not hurt
-            max_steps=5,
-            additional_authorized_imports = ['numpy','subprocess', 're', 'pandas',
-                                             'json', 'os', 'pathlib', 'tempfile', 
-                                             # 'matplotlib.pyplot', 'seaborn'
-                                            ],
-            verbosity_level = 1,
-            max_print_outputs_length=1_000_000
-        )
-        
-        print("✅ Local (quantized) agent initialized.")
-        
-    def __call__(self, question: str, file_path: Optional[str] = None) -> str:
-
-        if file_path:
-            # Inject system prompt + question and (optional) file path
-            prompt = (
-                f"{self.system_prompt}\n\n"
-                f"Question: {question}\n\n"
-                f"There is an associated file at path: {file_path}.\n"
-                f"Use the appropriate tool to download it (if necessary) and read it before answering"
-            )
-        else:
-            prompt = (
-                f"{self.system_prompt}\n\n"
-                f"Question: {question}\n\n"
-            )
-            
-        return self.local_agent.run(prompt)

requirements.txt

@@ -25,9 +25,4 @@ pytubefix==10.3.6
 openai-whisper==20250625
 
 # OCR (OPTIONAL, disabled)
-# pytesseract==0.3.13
-
-# Additional for LocalAgent (optional)
-transformers==4.1.0
-bitsandbytes==0.49.0 
-torch==2.6.0
+# pytesseract==0.3.13