obscura/ollama-python
1
0
Fork 0
mirror of https://github.com/ollama/ollama-python.git synced 2026-08-14 08:57:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

httpx: kwargs

Browse Source
This commit is contained in:
Michael Yang
2024-01-10 10:52:50 -08:00
parent cea391a041
commit 2804a03d82
1 changed files with 18 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ollama/_client.py
+18 -7
View File
@@ -20,14 +20,25 @@ from ollama._types import Message, Options
class BaseClient:
def __init__(self, client, base_url: Optional[str] = None) -> None:
base_url = base_url or os.getenv('OLLAMA_HOST', 'http://127.0.0.1:11434')
self._client = client(base_url=base_url, follow_redirects=True, timeout=None)
def __init__(
self,
client,
base_url: Optional[str] = None,
follow_redirects: bool = True,
timeout: Any = None,
**kwargs,
) -> None:
self._client = client(
base_url=base_url or os.getenv('OLLAMA_HOST', 'http://127.0.0.1:11434'),
follow_redirects=follow_redirects,
timeout=timeout,
**kwargs,
)
class Client(BaseClient):
def __init__(self, base_url: Optional[str] = None) -> None:
super().__init__(httpx.Client, base_url)
def __init__(self, base_url: Optional[str] = None, **kwargs) -> None:
super().__init__(httpx.Client, base_url, **kwargs)
def _request(self, method: str, url: str, **kwargs) -> httpx.Response:
response = self._client.request(method, url, **kwargs)
@@ -247,8 +258,8 @@ class Client(BaseClient):
class AsyncClient(BaseClient):
def __init__(self, base_url: Optional[str] = None) -> None:
super().__init__(httpx.AsyncClient, base_url)
def __init__(self, base_url: Optional[str] = None, **kwargs) -> None:
super().__init__(httpx.AsyncClient, base_url, **kwargs)
async def _request(self, method: str, url: str, **kwargs) -> httpx.Response:
response = await self._client.request(method, url, **kwargs)