o i} @sddlmZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZmZddlmZddlmZzddlmZWne ywZ!ze de!dZ![!wwGdddeeZ"dS)) annotationsN) AsyncOpenAI) ModelProfile) UserError)cached_async_http_client)cohere_model_profile)deepseek_model_profile)google_model_profile)harmony_model_profile)meta_model_profile)mistral_model_profile)OpenAIJsonSchemaTransformerOpenAIModelProfile)qwen_model_profile)ProvideruPlease install the `openai` package to use the Ollama provider, you can use the `openai` optional group — `pip install "pydantic-ai-slim[openai]"`c@sXeZdZdZedddZedddZedd d Zdd dZ    ddddZ dS)OllamaProviderz(Provider for local or remote Ollama API.returnstrcCsdS)Nollamaselfrrb/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_ai/providers/ollama.pyname!szOllamaProvider.namecCs t|jjSN)rclientbase_urlrrrrr%s zOllamaProvider.base_urlrcCs|jSr)_clientrrrrr)szOllamaProvider.client model_nameModelProfile | Nonec CsVttttttttd}d}|D]\}}|}| |r"||}qt t d |S)N)llamagemmaqwenqwqdeepseekmistralcommandzgpt-oss)json_schema_transformer) r r rrr rr itemslower startswithrr update)rrprefix_to_profileprofileprefix profile_funcrrr model_profile-s   zOllamaProvider.model_profileNr str | Noneapi_key openai_clientAsyncOpenAI | None http_clienthttpx.AsyncClient | NoneNonecCs|dur!|dus Jd|dusJd|dusJd||_dS|p'td}|s.td|p6tdp6d}|durEt|||d |_dStd d }t|||d |_dS) aCreate a new Ollama provider. Args: base_url: The base url for the Ollama requests. If not provided, the `OLLAMA_BASE_URL` environment variable will be used if available. api_key: The API key to use for authentication, if not provided, the `OLLAMA_API_KEY` environment variable will be used if available. openai_client: An existing [`AsyncOpenAI`](https://github.com/openai/openai-python?tab=readme-ov-file#async-usage) client to use. If provided, `base_url`, `api_key`, 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`z1Cannot provide both `openai_client` and `api_key`OLLAMA_BASE_URLzwSet the `OLLAMA_BASE_URL` environment variable or pass it via `OllamaProvider(base_url=...)`to use the Ollama provider.OLLAMA_API_KEYzapi-key-not-set)rr2r5r)provider)rosgetenvrrr)rrr2r3r5rrr__init__Cs  zOllamaProvider.__init__)rr)rr)rrrr)NNNN) rr1r2r1r3r4r5r6rr7) __name__ __module__ __qualname____doc__propertyrrrr0r=rrrrrs    r)# __future__r _annotationsr;httpxopenair pydantic_airpydantic_ai.exceptionsrpydantic_ai.modelsrpydantic_ai.profiles.cohererpydantic_ai.profiles.deepseekrpydantic_ai.profiles.googler pydantic_ai.profiles.harmonyr pydantic_ai.profiles.metar pydantic_ai.profiles.mistralr pydantic_ai.profiles.openair rpydantic_ai.profiles.qwenrpydantic_ai.providersr ImportError _import_errorrrrrrs6