Merge pull request #341 from flyingice/main
fix(grounding): align Parallel AI search with current API schema
This commit is contained in:
@@ -140,7 +140,10 @@ def parallel_search(
|
|||||||
data = http.request(
|
data = http.request(
|
||||||
"POST", "https://api.parallel.ai/v1/search",
|
"POST", "https://api.parallel.ai/v1/search",
|
||||||
headers={"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"},
|
headers={"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"},
|
||||||
json_data={"query": query, "max_results": count},
|
json_data={
|
||||||
|
"search_queries": [query],
|
||||||
|
"advanced_settings": {"max_results": count},
|
||||||
|
},
|
||||||
timeout=15,
|
timeout=15,
|
||||||
)
|
)
|
||||||
items = []
|
items = []
|
||||||
@@ -150,7 +153,7 @@ def parallel_search(
|
|||||||
url = r.get("url", "")
|
url = r.get("url", "")
|
||||||
if not url:
|
if not url:
|
||||||
continue
|
continue
|
||||||
raw_date = r.get("published_date") or ""
|
raw_date = r.get("publish_date") or ""
|
||||||
pub_date = _normalize_date(raw_date[:10]) if raw_date else None
|
pub_date = _normalize_date(raw_date[:10]) if raw_date else None
|
||||||
if not _in_date_range(pub_date, date_range):
|
if not _in_date_range(pub_date, date_range):
|
||||||
continue
|
continue
|
||||||
@@ -159,7 +162,7 @@ def parallel_search(
|
|||||||
"title": r.get("title", ""),
|
"title": r.get("title", ""),
|
||||||
"url": url,
|
"url": url,
|
||||||
"source_domain": _domain(url),
|
"source_domain": _domain(url),
|
||||||
"snippet": r.get("snippet", ""),
|
"snippet": ((r.get("excerpts") or [""])[0] or "")[:500],
|
||||||
"date": pub_date,
|
"date": pub_date,
|
||||||
"relevance": 0.8,
|
"relevance": 0.8,
|
||||||
"why_relevant": "Parallel AI web search",
|
"why_relevant": "Parallel AI web search",
|
||||||
|
|||||||
Reference in New Issue
Block a user