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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -24,6 +24,11 @@ class BasicAgent:
24
  self.api_url = f"https://api-inference.huggingface.co/models/{model}"
25
  self.headers = {"Authorization": f"Bearer {os.getenv('HF_API_KEY')}"}
26
  print("BasicAgent initialized.")
 
 
 
 
 
27
 
28
  #usage
29
  #agent = HuggingFaceAgent("google/gemma-7b") # Same architecture as Gemini
@@ -32,7 +37,7 @@ class BasicAgent:
32
 
33
  def __call__(self, question: str) -> str:
34
  print(f"Agent received question (first 50 chars): {question[:50]}...")
35
- fixed_answer = self.agent.generate_response(question)
36
  print(f"Agent returning answer: {fixed_answer}")
37
  return fixed_answer
38
 
 
24
  self.api_url = f"https://api-inference.huggingface.co/models/{model}"
25
  self.headers = {"Authorization": f"Bearer {os.getenv('HF_API_KEY')}"}
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
 
37
 
38
  def __call__(self, question: str) -> str:
39
  print(f"Agent received question (first 50 chars): {question[:50]}...")
40
+ fixed_answer = self.generate_response(question)
41
  print(f"Agent returning answer: {fixed_answer}")
42
  return fixed_answer
43