klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-02 13:18:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Truncate online query to Serper if query exceeds max supported length

Browse Source 
Previously query to serper with longer than max supported would throw
error instead of returning at least some results.

Truncating the onlien search query to serper to max supported length
mitigates that issue.
This commit is contained in:
Debanjum
2025-03-31 01:16:06 +05:30
parent db7eba56f6
commit 9b7442f28f
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/khoj/processor/tools/online_search.py
View File

@@ -251,9 +251,16 @@ async def search_with_google(query: str, location: LocationData) -> Tuple[str, D
async def search_with_serper(query: str, location: LocationData) -> Tuple[str, Dict[str, List[Dict]]]:
country_code = location.country_code.lower() if location and location.country_code else "us"
payload = json.dumps({"q": query, "gl": country_code})
headers = {"X-API-KEY": SERPER_DEV_API_KEY, "Content-Type": "application/json"}
country_code = location.country_code.lower() if location and location.country_code else "us"
max_query_length = 2048
if len(query) > max_query_length:
logger.warning(
f"Truncate online query. Query length {len(query)} exceeds {max_query_length} supported by Serper. Query: {query}"
)
query = query[:max_query_length]
payload = json.dumps({"q": query, "gl": country_code})
async with aiohttp.ClientSession() as session:
async with session.post(SERPER_DEV_URL, headers=headers, data=payload) as response: