refac: ollama cloud web search count support
This commit is contained in:
parent
3d2f7e3143
commit
fe65fe0b97
|
@ -26,11 +26,9 @@ def search_ollama_cloud(
|
||||||
filter_list (Optional[list[str]]): List of domains to filter results by
|
filter_list (Optional[list[str]]): List of domains to filter results by
|
||||||
"""
|
"""
|
||||||
log.info(f"Searching with Ollama for query: {query}")
|
log.info(f"Searching with Ollama for query: {query}")
|
||||||
headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
|
|
||||||
|
|
||||||
payload = {
|
headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
|
||||||
"query": query,
|
payload = {"query": query, "max_results": count}
|
||||||
}
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
response = requests.post(f"{url}/api/web_search", headers=headers, json=payload)
|
response = requests.post(f"{url}/api/web_search", headers=headers, json=payload)
|
||||||
|
|
Loading…
Reference in New Issue