wt002 commited on
Commit
4eb08aa
·
verified ·
1 Parent(s): 001fcdb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -6,7 +6,7 @@ import requests
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,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://your-searx-instance.com" # Set your SearxNG instance URL
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"))