o i @sddlmZddlZddlmZddlZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZzddlm Z Wne!yuZ"ze!de"dZ"["wwdddZ#dddZ$Gdddee Z%dS)) annotationsN)overload) ModelProfile) UserError)cached_async_http_client)deepseek_model_profile)google_model_profile)groq_model_profile)meta_model_profile)mistral_model_profile)moonshotai_model_profile)openai_model_profile)qwen_model_profile)Provider) AsyncGroquPlease install the `groq` package to use the Groq provider, you can use the `groq` optional group — `pip install "pydantic-ai-slim[groq]"` model_namestrreturnModelProfile | NonecCstdddt|S)zJGet the model profile for an MoonshotAI model used with the Groq provider.Tsupports_json_object_outputsupports_json_schema_output)rupdater rr`/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_ai/providers/groq.pygroq_moonshotai_model_profiles rcCs&|dvrtdddt|St|S)zCGet the model profile for a Meta model used with the Groq provider.>llama-4-scout-17b-16e-instruct"llama-4-maverick-17b-128e-instructTr)rrr rrrrmeta_groq_model_profile%s  rc@seZdZdZedddZedddZed d d Zd!d dZe ddd"ddZ e ddddd#ddZ dddddd$ddZ dS)% GroqProviderzProvider for Groq API.rrcCsdS)Ngroqrselfrrrname2szGroqProvider.namecCs t|jjSN)rclientbase_urlr"rrrr'6s zGroqProvider.base_urlrcCs|jSr%)_clientr"rrrr&:szGroqProvider.clientrrc Csftttttttttd }| D] \}}| }| |r0| dr*|t |d}||SqdS)N) llamaz meta-llama/gemmaqwendeepseekmistralz moonshotai/z compound-zopenai//)r rrrrr rr r itemslower startswithendswithlen)r#rprefix_to_profileprefix profile_funcrrr model_profile>s$    zGroqProvider.model_profileN) groq_clientr8AsyncGroq | NoneNonecCdSr%r)r#r8rrr__init__TszGroqProvider.__init__)api_keyr' http_clientr= str | Noner'r>httpx.AsyncClient | NonecCr;r%r)r#r=r'r>rrrr<Ws)r=r'r8r>cCs|dur!|dus Jd|dusJd|dusJd||_dS|p'td}|p/tdd}|s6td|durDt|||d |_dStd d }t|||d |_dS) aCreate a new Groq provider. Args: api_key: The API key to use for authentication, if not provided, the `GROQ_API_KEY` environment variable will be used if available. base_url: The base url for the Groq requests. If not provided, the `GROQ_BASE_URL` environment variable will be used if available. Otherwise, defaults to Groq's base url. groq_client: An existing [`AsyncGroq`](https://github.com/groq/groq-python?tab=readme-ov-file#async-usage) client to use. If provided, `api_key` and `http_client` must be `None`. http_client: An existing `AsyncHTTPClient` to use for making HTTP requests. Nz3Cannot provide both `groq_client` and `http_client`z/Cannot provide both `groq_client` and `api_key`z0Cannot provide both `groq_client` and `base_url` GROQ_API_KEY GROQ_BASE_URLzhttps://api.groq.comzoSet the `GROQ_API_KEY` environment variable or pass it via `GroqProvider(api_key=...)`to use the Groq provider.)r'r=r>r!)provider)r(osgetenvrrr)r#r=r'r8r>rrrr<\s  )rr)rrrrrr)r8r9rr:)r=r?r'r?r>r@rr:) r=r?r'r?r8r9r>r@rr:) __name__ __module__ __qualname____doc__propertyr$r'r&r7rr<rrrrr /s&    r rF)& __future__r _annotationsrDtypingrhttpx pydantic_airpydantic_ai.exceptionsrpydantic_ai.modelsrpydantic_ai.profiles.deepseekrpydantic_ai.profiles.googlerpydantic_ai.profiles.groqr pydantic_ai.profiles.metar pydantic_ai.profiles.mistralr pydantic_ai.profiles.moonshotair pydantic_ai.profiles.openair pydantic_ai.profiles.qwenrpydantic_ai.providersrr!r ImportError _import_errorrrr rrrrs: