Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -1028,14 +1028,37 @@ def search_knowledge_base(query):
|
|
| 1028 |
|
| 1029 |
def google_search(query):
|
| 1030 |
try:
|
|
|
|
| 1031 |
search_client = serpapi.Client(api_key=serper_api_key)
|
| 1032 |
-
|
| 1033 |
-
|
| 1034 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1035 |
except Exception as e:
|
| 1036 |
logger.error(f"Error in Google search: {e}")
|
| 1037 |
return ["Error occurred during Google search"]
|
| 1038 |
-
|
| 1039 |
|
| 1040 |
|
| 1041 |
# RAG response function
|
|
|
|
| 1028 |
|
| 1029 |
def google_search(query):
|
| 1030 |
try:
|
| 1031 |
+
# Initialize SerpAPI client and set parameters
|
| 1032 |
search_client = serpapi.Client(api_key=serper_api_key)
|
| 1033 |
+
search_params = {
|
| 1034 |
+
"engine": "google", # Use Google as the search engine
|
| 1035 |
+
"q": query, # Query
|
| 1036 |
+
"sort": "date", # Sort by latest date
|
| 1037 |
+
"hl": "en", # Language set to English
|
| 1038 |
+
"gl": "us" # Country/region set to US
|
| 1039 |
+
}
|
| 1040 |
+
|
| 1041 |
+
# Perform the search
|
| 1042 |
+
results = search_client.search(search_params)
|
| 1043 |
+
|
| 1044 |
+
# Process the results to extract links and snippets
|
| 1045 |
+
organic_results = results.get("organic_results", [])
|
| 1046 |
+
valid_results = []
|
| 1047 |
+
for result in organic_results:
|
| 1048 |
+
link = result.get("link", "")
|
| 1049 |
+
snippet = result.get("snippet", "")
|
| 1050 |
+
# Validate the link (basic check to avoid blank or invalid links)
|
| 1051 |
+
if link.startswith("http"):
|
| 1052 |
+
valid_results.append({"link": link, "snippet": snippet})
|
| 1053 |
+
|
| 1054 |
+
# Ensure we return meaningful results
|
| 1055 |
+
if valid_results:
|
| 1056 |
+
return valid_results
|
| 1057 |
+
return ["No valid links found"]
|
| 1058 |
+
|
| 1059 |
except Exception as e:
|
| 1060 |
logger.error(f"Error in Google search: {e}")
|
| 1061 |
return ["Error occurred during Google search"]
|
|
|
|
| 1062 |
|
| 1063 |
|
| 1064 |
# RAG response function
|