add Groq client support (#84)

Co-authored-by: Samuel Colvin <[email protected]>

Author: ricklamers

docs/api/models/groq.md

@@ -0,0 +1,3 @@
+# `pydantic_ai.models.groq`
+
+::: pydantic_ai.models.groq

docs/index.md

@@ -11,7 +11,7 @@ PydanticAI is a Python Agent Framework designed to make it less painful to build
 ## Why use PydanticAI
 
 * Built by the team behind Pydantic (the validation layer of the OpenAI SDK, the Anthropic SDK, LangChain, LlamaIndex, AutoGPT, Transformers, Instructor and many more)
-* Model-agnostic — currently both OpenAI and Gemini are supported, and Anthropic [is coming soon](https://github.com/pydantic/pydantic-ai/issues/63). And there is a simple interface to implement and customize support for other models.
+* Model-agnostic — currently both OpenAI, Gemini, and Groq are supported, Anthropic [is coming soon](https://github.com/pydantic/pydantic-ai/issues/63). And there is a simple interface to implement support for other models.
 * Type-safe
 * Control flow and composing agents is done with vanilla python, allowing you to make use of the same Python development best practices you'd use in any other (non-AI) project
 * [Structured response](results.md#structured-result-validation) validation with Pydantic

mkdocs.yml

@@ -39,6 +39,7 @@ nav:
     - api/models/openai.md
     - api/models/gemini.md
     - api/models/vertexai.md
+    - api/models/groq.md
     - api/models/test.md
     - api/models/function.md
 

pydantic_ai/models/__init__.py

@@ -30,6 +30,20 @@
     'openai:o1-preview',
     'openai:o1-mini',
     'openai:gpt-3.5-turbo',
+    'groq:llama-3.1-70b-versatile',
+    'groq:llama3-groq-70b-8192-tool-use-preview',
+    'groq:llama3-groq-8b-8192-tool-use-preview',
+    'groq:llama-3.1-70b-specdec',
+    'groq:llama-3.1-8b-instant',
+    'groq:llama-3.2-1b-preview',
+    'groq:llama-3.2-3b-preview',
+    'groq:llama-3.2-11b-vision-preview',
+    'groq:llama-3.2-90b-vision-preview',
+    'groq:llama3-70b-8192',
+    'groq:llama3-8b-8192',
+    'groq:mixtral-8x7b-32768',
+    'groq:gemma2-9b-it',
+    'groq:gemma-7b-it',
     'gemini-1.5-flash',
     'gemini-1.5-pro',
     'test',
@@ -227,6 +241,10 @@ def infer_model(model: Model | KnownModelName) -> Model:
 
         # noinspection PyTypeChecker
         return GeminiModel(model)  # pyright: ignore[reportArgumentType]
+    elif model.startswith('groq:'):
+        from .groq import GroqModel
+
+        return GroqModel(model[5:])  # pyright: ignore[reportArgumentType]
     else:
         from ..exceptions import UserError