Kackle commited on
Commit
074144b
·
verified ·
1 Parent(s): e40eaa5

Update gemini_agent.py

Browse files
Files changed (1) hide show
  1. gemini_agent.py +3 -3
gemini_agent.py CHANGED
@@ -6,8 +6,8 @@ import re
6
  import time
7
  import asyncio
8
  # Add LangChain tools for Wikipedia and DuckDuckGo
9
- from langchain_community.tools import DuckDuckGoSearchRun, WikipediaQueryRun
10
- from langchain_community.utilities import WikipediaAPIWrapper
11
 
12
  load_dotenv()
13
 
@@ -19,7 +19,7 @@ class GeminiAgent:
19
  api_key = os.getenv('GOOGLE_API_KEY')
20
  genai.configure(api_key=api_key)
21
 
22
- self.model = genai.GenerativeModel('gemini-1.5-pro-latest')
23
  self.last_request_time = 0
24
  self.min_request_interval = 6.0 # 6 seconds between requests (10 per minute limit)
25
 
 
6
  import time
7
  import asyncio
8
  # Add LangChain tools for Wikipedia and DuckDuckGo
9
+ from langchain.tools import DuckDuckGoSearchRun, WikipediaQueryRun
10
+ from langchain.utilities import WikipediaAPIWrapper
11
 
12
  load_dotenv()
13
 
 
19
  api_key = os.getenv('GOOGLE_API_KEY')
20
  genai.configure(api_key=api_key)
21
 
22
+ self.model = genai.GenerativeModel('gemini-2.0-flash-exp')
23
  self.last_request_time = 0
24
  self.min_request_interval = 6.0 # 6 seconds between requests (10 per minute limit)
25