kanshan/ollama-python
1
0
Fork 0
mirror of https://github.com/ollama/ollama-python.git synced 2026-01-14 06:07:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

formatting issues

Browse Source
This commit is contained in:
nicole pardal 2025-09-23 12:06:11 -07:00
parent 81c71ba44e
commit d3ddff329f
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
examples/web-search-crawl.py
View File

@ -6,7 +6,7 @@
# ] # ]
# /// # ///
import os import os
from typing import Optional, Union from typing import Union
from rich import print from rich import print
@ -20,7 +20,6 @@ def format_tool_results(
if isinstance(results, WebSearchResponse): if isinstance(results, WebSearchResponse):
output = [] output = []
output.append(f'Search results for "{user_search}":') output.append(f'Search results for "{user_search}":')
print()
for i, result in enumerate(results.results, 1): for i, result in enumerate(results.results, 1):
title = getattr(result, 'title', None) title = getattr(result, 'title', None)
url_value = getattr(result, 'url', None) url_value = getattr(result, 'url', None)
@ -44,7 +43,8 @@ def format_tool_results(
output.append('') output.append('')
return '\n'.join(output).rstrip() return '\n'.join(output).rstrip()
client = Client(headers={'Authorization': (os.getenv('OLLAMA_API_KEY'))}) api_key = os.getenv('OLLAMA_API_KEY')
client = Client(headers={'Authorization': f"Bearer {s.getenv('OLLAMA_API_KEY')}"} if api_key else None)
available_tools = {'web_search': client.web_search, 'web_fetch': client.web_fetch} available_tools = {'web_search': client.web_search, 'web_fetch': client.web_fetch}
query = "ollama's new engine" query = "ollama's new engine"

8
ollama/_client.py
View File

@ -652,7 +652,7 @@ class Client(BaseClient):
return self._request( return self._request(
WebSearchResponse, WebSearchResponse,
'POST', 'POST',
'https://ollama.com/api/web_search', 'http://localhost:8080/api/web_search',
json=WebSearchRequest( json=WebSearchRequest(
query=query, query=query,
max_results=max_results, max_results=max_results,
@ -675,7 +675,7 @@ class Client(BaseClient):
return self._request( return self._request(
WebFetchResponse, WebFetchResponse,
'POST', 'POST',
'https://ollama.com/api/web_fetch', 'http://localhost:8080/api/web_fetch',
json=WebFetchRequest( json=WebFetchRequest(
url=url, url=url,
).model_dump(exclude_none=True), ).model_dump(exclude_none=True),
@ -764,7 +764,7 @@ class AsyncClient(BaseClient):
return await self._request( return await self._request(
WebSearchResponse, WebSearchResponse,
'POST', 'POST',
'https://ollama.com/api/web_search', 'http://localhost:8080/api/web_search',
json=WebSearchRequest( json=WebSearchRequest(
query=query, query=query,
max_results=max_results, max_results=max_results,
@ -784,7 +784,7 @@ class AsyncClient(BaseClient):
return await self._request( return await self._request(
WebFetchResponse, WebFetchResponse,
'POST', 'POST',
'https://ollama.com/api/web_fetch', 'http://localhost:8080/api/web_fetch',
json=WebFetchRequest( json=WebFetchRequest(
url=url, url=url,
).model_dump(exclude_none=True), ).model_dump(exclude_none=True),