mirror of
https://github.com/Alishahryar1/free-claude-code.git
synced 2026-07-03 14:05:26 +02:00
Alishahryar1
40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
"""Ollama provider implementation."""
|
|
|
|
import httpx
|
|
|
|
from providers.anthropic_messages import AnthropicMessagesTransport
|
|
from providers.base import ProviderConfig
|
|
from providers.defaults import OLLAMA_DEFAULT_BASE
|
|
from providers.model_listing import extract_ollama_model_ids
|
|
|
|
|
|
class OllamaProvider(AnthropicMessagesTransport):
|
|
"""Ollama provider using native Anthropic Messages API."""
|
|
|
|
def __init__(self, config: ProviderConfig):
|
|
super().__init__(
|
|
config,
|
|
provider_name="OLLAMA",
|
|
default_base_url=OLLAMA_DEFAULT_BASE,
|
|
)
|
|
self._api_key = config.api_key or "ollama"
|
|
|
|
async def _send_stream_request(self, body: dict) -> httpx.Response:
|
|
"""Create a streaming native Anthropic messages response."""
|
|
request = self._client.build_request(
|
|
"POST",
|
|
"/v1/messages",
|
|
json=body,
|
|
headers=self._request_headers(),
|
|
)
|
|
return await self._client.send(request, stream=True)
|
|
|
|
async def _send_model_list_request(self) -> httpx.Response:
|
|
"""Query Ollama's native local model-list endpoint."""
|
|
return await self._client.get(f"{self._base_url}/api/tags")
|
|
|
|
def _extract_model_ids_from_model_list_payload(
|
|
self, payload: object
|
|
) -> frozenset[str]:
|
|
return extract_ollama_model_ids(payload, provider_name=self._provider_name)
|