g\ddlmZddlmZddlmZmZddlm Z ddl m Z ddl mZmZmZ dd lmZdd lmZed Z eeefZ edGddeZy #e$rZed ed Z[wwxYw)) annotations) dataclass)LiteralUnion) AsyncClient)ToolDefinition) AgentModelModelcached_async_http_client) AsyncOpenAIuPlease install `openai` to use the OpenAI model, you can use the `openai` optional group — `pip install 'pydantic-ai-slim[openai]'`N) OpenAIModel) codellamagemmagemma2llama3zllama3.1zllama3.2zllama3.2-visionzllama3.3mistralz mistral-nemomixtralphi3qwqqwenqwen2zqwen2.5 starcoder2F)initcleZdZUdZded<ded<ddddd  d d Z dd Zdd Zy) OllamaModelaA model that implements Ollama using the OpenAI API. Internally, this uses the [OpenAI Python client](https://github.com/openai/openai-python) to interact with the Ollama server. Apart from `__init__`, all methods are private or match those of the base class. OllamaModelName model_namer openai_modelzhttp://localhost:11434/v1/ollamaN)base_urlapi_key openai_client http_clientc||_|%|Jd|Jdt|||_y|xs t}t |||}t|||_y)aRInitialize an Ollama model. Ollama has built-in compatability for the OpenAI chat completions API ([source](https://ollama.com/blog/openai-compatibility)), so we reuse the [`OpenAIModel`][pydantic_ai.models.openai.OpenAIModel] here. Args: model_name: The name of the Ollama model to use. List of models available [here](https://ollama.com/library) You must first download the model (`ollama pull `) in order to use the model base_url: The base url for the ollama requests. The default value is the ollama default api_key: The API key to use for authentication. Defaults to 'ollama' for local instances, but can be customized for proxy setups that require authentication openai_client: An existing [`AsyncOpenAI`](https://github.com/openai/openai-python?tab=readme-ov-file#async-usage) client to use, if provided, `base_url` and `http_client` must be `None`. http_client: An existing `httpx.AsyncClient` to use for making HTTP requests. Nz2Cannot provide both `openai_client` and `base_url`z5Cannot provide both `openai_client` and `http_client`)rr$)r"r#r%)rrr r r)selfrr"r#r$r% http_client_ oai_clients N/var/www/openai/venv/lib/python3.12/site-packages/pydantic_ai/models/ollama.py__init__zOllamaModel.__init__Est2%  $# Y%Y Y#& _(_ _& +zQ^ _D 'D*B*DL$hUabJ +zQ[ \D cZK|jj|||d{S7w)N)function_toolsallow_text_result result_tools)r agent_model)r'r.r/r0s r*r1zOllamaModel.agent_modelis9&&22)/%3    s "+)+c d|jS)Nzollama:)r)r's r*namezOllamaModel.namevs)**r,) rrr"z str | Noner#strr$zAsyncOpenAI | Noner%zAsyncHTTPClient | None)r.list[ToolDefinition]r/boolr0r5returnr )r7r4)__name__ __module__ __qualname____doc____annotations__r+r1r3r,r*rr9s  <,0.2"]#"] "]  "] * "],"]H  -    +    +r,r) __future__r _annotations dataclassesrtypingrrhttpxrAsyncHTTPClienttoolsr r r r openair ImportErrorerCommonOllamaModelNamesr4rrr=r,r*rJs2!!0"  "  &.34 =+%=+=+S  _    sAA1# A,,A1