Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ import requests
|
|
6 |
from typing import List, Dict, Union
|
7 |
import requests
|
8 |
import wikipediaapi
|
9 |
-
|
10 |
from typing import List, Dict, Union
|
11 |
import requests
|
12 |
import pandas as pd
|
@@ -17,6 +17,8 @@ load_dotenv()
|
|
17 |
# --- Constants ---
|
18 |
DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
19 |
|
|
|
|
|
20 |
|
21 |
# --- Basic Agent Definition ---
|
22 |
class BasicAgent:
|
@@ -26,10 +28,11 @@ class BasicAgent:
|
|
26 |
print("BasicAgent initialized.")
|
27 |
|
28 |
# Initialize other required components
|
29 |
-
self.searx_url = "https://
|
30 |
self.wiki = wikipediaapi.Wikipedia('en') # Requires wikipedia-api package
|
31 |
|
32 |
-
|
|
|
33 |
#usage
|
34 |
#agent = HuggingFaceAgent("google/gemma-7b") # Same architecture as Gemini
|
35 |
#print(agent.generate("Explain quantum computing"))
|
|
|
6 |
from typing import List, Dict, Union
|
7 |
import requests
|
8 |
import wikipediaapi
|
9 |
+
import google.generativeai as genai
|
10 |
from typing import List, Dict, Union
|
11 |
import requests
|
12 |
import pandas as pd
|
|
|
17 |
# --- Constants ---
|
18 |
DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
19 |
|
20 |
+
# Configure Gemini
|
21 |
+
genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
|
22 |
|
23 |
# --- Basic Agent Definition ---
|
24 |
class BasicAgent:
|
|
|
28 |
print("BasicAgent initialized.")
|
29 |
|
30 |
# Initialize other required components
|
31 |
+
self.searx_url = "https://searx.space/search" # Set your SearxNG instance URL
|
32 |
self.wiki = wikipediaapi.Wikipedia('en') # Requires wikipedia-api package
|
33 |
|
34 |
+
genai.configure(api_key=os.getenv('GEMINI_API_KEY'))
|
35 |
+
self.model = genai.GenerativeModel(model)
|
36 |
#usage
|
37 |
#agent = HuggingFaceAgent("google/gemma-7b") # Same architecture as Gemini
|
38 |
#print(agent.generate("Explain quantum computing"))
|