Update gemini_agent.py

gemini_agent.py

@@ -5,10 +5,9 @@ from excel_parser import ExcelParser
 import re
 import time
 import asyncio
-from google.generativeai.types import HarmCategory, HarmBlockThreshold
 # Add LangChain tools for Wikipedia and DuckDuckGo
-from langchain_community.tools import DuckDuckGoSearchRun, WikipediaQueryRun
-from langchain_community.utilities import WikipediaAPIWrapper
+from langchain.tools import DuckDuckGoSearchRun, WikipediaQueryRun
+from langchain.utilities import WikipediaAPIWrapper
 
 load_dotenv()
 
@@ -20,7 +19,7 @@ class GeminiAgent:
         api_key = os.getenv('GOOGLE_API_KEY')
         genai.configure(api_key=api_key)
         
-        self.model = genai.GenerativeModel('gemini-2.5-pro')
+        self.model = genai.GenerativeModel('gemini-2.0-flash-exp')
         self.last_request_time = 0
         self.min_request_interval = 6.0  # 6 seconds between requests (10 per minute limit)
         
@@ -78,13 +77,7 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
                 generation_config=genai.types.GenerationConfig(
                     max_output_tokens=50,
                     temperature=0.0
-                ),
-                safety_settings={
-                    HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
-                    HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
-                    HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
-                    HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
-                }
+                )
             )
             answer = response.text.strip()
             
@@ -152,13 +145,7 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
                 generation_config=genai.types.GenerationConfig(
                     max_output_tokens=150,
                     temperature=0.0
-                ),
-                safety_settings={
-                    HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
-                    HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
-                    HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
-                    HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
-                }
+                )
             )
             answer = response.text.strip()
             
@@ -251,13 +238,7 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
             generation_config=genai.types.GenerationConfig(
                 max_output_tokens=100,
                 temperature=0.0
-            ),
-            safety_settings={
-                HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
-                HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
-                HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
-                HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
-            }
+            )
         )
         answer = response.text.strip()
         
@@ -322,13 +303,7 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
                 generation_config=genai.types.GenerationConfig(
                     max_output_tokens=100,
                     temperature=0.0
-                ),
-                safety_settings={
-                    HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
-                    HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
-                    HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
-                    HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
-                }
+                )
             )
             answer = response.text.strip()