kanshan/ollama-python
1
0
Fork 0
mirror of https://github.com/ollama/ollama-python.git synced 2026-01-13 21:57:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

client: load OLLAMA_API_KEY on init

Browse Source
This commit is contained in:
ParthSareen 2025-09-22 13:11:02 -07:00
parent b22c5fdabb
commit 17fb82e215
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
ollama/_client.py
View File

@ -94,6 +94,8 @@ class BaseClient:
`kwargs` are passed to the httpx client.
"""
api_key = os.getenv('OLLAMA_API_KEY', None)
self._client = client(
base_url=_parse_host(host or os.getenv('OLLAMA_HOST')),
follow_redirects=follow_redirects,
@ -106,6 +108,7 @@ class BaseClient:
'Content-Type': 'application/json',
'Accept': 'application/json',
'User-Agent': f'ollama-python/{__version__} ({platform.machine()} {platform.system().lower()}) Python/{platform.python_version()}',
'Authorization': f'Bearer {api_key}' if api_key else '',
}.items()
},
**kwargs,
@ -638,7 +641,13 @@ class Client(BaseClient):
Returns:
WebSearchResponse with the search results
Raises:
ValueError: If OLLAMA_API_KEY environment variable is not set
"""
api_key = os.getenv('OLLAMA_API_KEY')
if not api_key:
raise ValueError('OLLAMA_API_KEY environment variable is required for web search')
return self._request(
WebSearchResponse,
'POST',
@ -658,7 +667,13 @@ class Client(BaseClient):
Returns:
WebCrawlResponse with the crawl results
Raises:
ValueError: If OLLAMA_API_KEY environment variable is not set
"""
api_key = os.getenv('OLLAMA_API_KEY')
if not api_key:
raise ValueError('OLLAMA_API_KEY environment variable is required for web fetch')
return self._request(
WebCrawlResponse,
'POST',