o i$ @sddlmZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZdd lmZdd lmZdd lmZzdd lmZWney_ZzededZ[wwddlmZGdddeeZdS)) annotationsN)overload) AsyncClient) ModelProfile) UserError)deepseek_model_profile)google_model_profile)meta_model_profile)mistral_model_profile)moonshotai_model_profile)qwen_model_profile)AsyncInferenceClientuPlease install the `huggingface_hub` package to use the HuggingFace provider, you can use the `huggingface` optional group — `pip install 'pydantic-ai-slim[huggingface]'`)Providerc@seZdZdZed"ddZed"ddZed#d d Zd$d dZe ddd%ddZ e ddd&ddZ e ddd'ddZ e ddd(ddZ e ddd)ddZ e ddd*ddZ     d+d,d!dZ dS)-HuggingFaceProviderzProvider for Hugging Face.returnstrcCsdS)N huggingfaceselfrrg/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_ai/providers/huggingface.pynameszHuggingFaceProvider.namecCs|jjSN)clientmodelrrrrbase_url#szHuggingFaceProvider.base_urlr cCs|jSr)_clientrrrrr'szHuggingFaceProvider.client model_nameModelProfile | NonecCsNttttttd}d|vrdS|}|dd\}}||vr%|||SdS)N)z deepseek-aigoogleqwenz meta-llama mistralai moonshotai/r)rrr r r r lowersplit)rrprovider_to_profileproviderrrr model_profile+s  z!HuggingFaceProvider.model_profileN)api_keyrr* str | NoneNonecCdSrr)rrr*rrr__init__?zHuggingFaceProvider.__init__ provider_namecCr-rr)rr0r*rrrr.Ar/ hf_clientcCr-rr)rr1r*rrrr.Cr/cCr-rr)rr1rr*rrrr.Er/cCr-rr)rr1r0r*rrrr.Gr/cCr-rr)rr*rrrr.Ir/AsyncInferenceClient | None http_clientAsyncClient | NonecCsl|ptd}|durtd|durtd|dur#|dur#td|dur1t|||d|_dS||_dS)aCreate a new Hugging Face provider. Args: base_url: The base url for the Hugging Face requests. api_key: The API key to use for authentication, if not provided, the `HF_TOKEN` environment variable will be used if available. hf_client: An existing [`AsyncInferenceClient`](https://huggingface.co/docs/huggingface_hub/v0.29.3/en/package_reference/inference_client#huggingface_hub.AsyncInferenceClient) client to use. If not provided, a new instance will be created. http_client: (currently ignored) An existing `httpx.AsyncClient` to use for making HTTP requests. provider_name : Name of the provider to use for inference. available providers can be found in the [HF Inference Providers documentation](https://huggingface.co/docs/inference-providers/index#partners). defaults to "auto", which will select the first available provider for the model, the first of the providers available for the model, sorted by the user's order in https://hf.co/settings/inference-providers. If `base_url` is passed, then `provider_name` is not used. HF_TOKENNzySet the `HF_TOKEN` environment variable or pass it via `HuggingFaceProvider(api_key=...)`to use the HuggingFace provider.zR`http_client` is ignored for HuggingFace provider, please use `hf_client` instead.z3Cannot provide both `base_url` and `provider_name`.)r*r(r)osgetenvr ValueErrorr r)rrr*r1r3r0rrrr.Ls )rr)rr )rrrr)rrr*r+rr,)r0rr*r+rr,)r1r r*r+rr,)r1r rrr*r+rr,)r1r r0rr*r+rr,)r*r+rr,)NNNNN) rr+r*r+r1r2r3r4r0r+rr,) __name__ __module__ __qualname____doc__propertyrrrr)rr.rrrrrs6    r) __future__r _annotationsr6typingrhttpxr pydantic_airpydantic_ai.exceptionsrpydantic_ai.profiles.deepseekrpydantic_ai.profiles.googlerpydantic_ai.profiles.metar pydantic_ai.profiles.mistralr pydantic_ai.profiles.moonshotair pydantic_ai.profiles.qwenr huggingface_hubr ImportError _import_errorrrrrrrs0