AC-Angelo93 commited on
Commit
14f3941
·
verified ·
1 Parent(s): 2336c3b

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -1,7 +1,7 @@
1
  # agent.py
2
  import os
3
  #from supabase import create_client
4
- from sentence_transformers import SentenceTransformers
5
  from serpapi import GoogleSearch
6
  import pandas as pd
7
  import faiss
@@ -80,7 +80,7 @@ def retrieve_docs(query: str, k: int = 3) -> str:
80
  return "\n\n---\n\n".join(hits)
81
 
82
 
83
- SERPAPI_KEY = os.getenv("SERPAPY_KEY")
84
  # ---- web_search tool
85
  @tool
86
  def web_search(query: str, num_results: int = 5) -> str:
 
1
  # agent.py
2
  import os
3
  #from supabase import create_client
4
+ from sentence_transformers import SentenceTransformer
5
  from serpapi import GoogleSearch
6
  import pandas as pd
7
  import faiss
 
80
  return "\n\n---\n\n".join(hits)
81
 
82
 
83
+ SERPAPI_KEY = os.getenv("SERPAPI_KEY")
84
  # ---- web_search tool
85
  @tool
86
  def web_search(query: str, num_results: int = 5) -> str: