Spaces:
Sleeping
Sleeping
Update gemini_agent.py
Browse files- 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
|
| 10 |
-
from
|
| 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-
|
| 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 |
|