add more modes

agent.py

@@ -20,7 +20,14 @@ import json
 from langchain_core.messages import SystemMessage
 from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
 from langchain_ollama import ChatOllama
+from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain_groq import ChatGroq
 from langgraph.prebuilt import create_react_agent
+import warnings
+from bs4 import GuessedAtParserWarning
+
+# Suppress the specific BeautifulSoup warning from the wikipedia library
+warnings.filterwarnings("ignore", category=GuessedAtParserWarning)
 
 # (Keep Constants as is)
 # --- Constants -- -
@@ -292,16 +299,35 @@ class BasicAgent:
         space_id = os.getenv("SPACE_ID")
 
         if space_id:
-            print("Detected Hugging Face Space environment. Initializing Qwen2.5-72B-Instruct via Inference Endpoint...")
-            llm_engine = HuggingFaceEndpoint(
-                repo_id="Qwen/Qwen2.5-72B-Instruct",
-                huggingfacehub_api_token=hf_token,
-                task="text-generation",
-                max_new_tokens=2048,
-                do_sample=False,
-                timeout=300,
-            )
-            self.llm = ChatHuggingFace(llm=llm_engine)
+            print("Detected Hugging Face Space environment.")
+            google_api_key = os.getenv("GOOGLE_API_KEY")
+            groq_api_key = os.getenv("GROQ_API_KEY")
+            
+            if google_api_key:
+                print("Initializing Gemini 1.5 Flash...")
+                self.llm = ChatGoogleGenerativeAI(
+                    model="gemini-1.5-flash",
+                    google_api_key=google_api_key,
+                    temperature=0.0,
+                )
+            elif groq_api_key:
+                print("Initializing Llama-3-70b via Groq (Free Tier)...")
+                self.llm = ChatGroq(
+                    model="llama3-70b-8192",
+                    groq_api_key=groq_api_key,
+                    temperature=0.0,
+                )
+            else:
+                 print("Warning: No Gemini or Groq key found. Falling back to HF Endpoint...")
+                 llm_engine = HuggingFaceEndpoint(
+                    repo_id="Qwen/Qwen2.5-72B-Instruct",
+                    huggingfacehub_api_token=hf_token,
+                    task="text-generation",
+                    max_new_tokens=2048,
+                    do_sample=False,
+                    timeout=300,
+                )
+                 self.llm = ChatHuggingFace(llm=llm_engine)
         else:
             print("Detected local environment. Initializing local Ollama with Qwen3...")
             self.llm = ChatOllama(

requirements.txt

@@ -26,4 +26,6 @@ torchvision
 av
 ffmpeg-python
 Pillow
-huggingface_hub
+huggingface_hub
+langchain-google-genai
+langchain-groq