Add support on Groq

Author: Kludex

.pre-commit-config.yaml

@@ -60,6 +60,6 @@ repos:
     rev: v2.3.0
     hooks:
     - id: codespell
-      args: ['--skip', 'tests/models/cassettes/*,docs/a2a/fasta2a.md']
+      args: ['--skip', 'tests/models/cassettes/*,docs/a2a/fasta2a.md,tests/models/test_groq.py']
       additional_dependencies:
       - tomli

pydantic_ai_slim/pydantic_ai/models/groq.py

@@ -10,7 +10,7 @@
 from typing_extensions import assert_never
 
 from .. import ModelHTTPError, UnexpectedModelBehavior, _utils, usage
-from .._utils import guard_tool_call_id as _guard_tool_call_id
+from .._utils import generate_tool_call_id, guard_tool_call_id as _guard_tool_call_id
 from ..messages import (
     BinaryContent,
     DocumentUrl,
@@ -21,6 +21,8 @@
     ModelResponsePart,
     ModelResponseStreamEvent,
     RetryPromptPart,
+    ServerToolCallPart,
+    ServerToolReturnPart,
     SystemPromptPart,
     TextPart,
     ToolCallPart,
@@ -205,7 +207,7 @@ async def _completions_create(
             extra_headers = model_settings.get('extra_headers', {})
             extra_headers.setdefault('User-Agent', get_user_agent())
             return await self.client.chat.completions.create(
-                model=str(self._model_name),
+                model=self._model_name,
                 messages=groq_messages,
                 n=1,
                 parallel_tool_calls=model_settings.get('parallel_tool_calls', NOT_GIVEN),
@@ -234,7 +236,15 @@ def _process_response(self, response: chat.ChatCompletion) -> ModelResponse:
         timestamp = datetime.fromtimestamp(response.created, tz=timezone.utc)
         choice = response.choices[0]
         items: list[ModelResponsePart] = []
-        # TODO(Marcelo): The `choice.message.executed_tools` has the server tools executed by Groq.
+        if choice.message.executed_tools:
+            for tool in choice.message.executed_tools:
+                tool_call_id = generate_tool_call_id()
+                items.append(
+                    ServerToolCallPart(
+                        tool_name=tool.type, args=tool.arguments, model_name='groq', tool_call_id=tool_call_id
+                    )
+                )
+                items.append(ServerToolReturnPart(tool_name=tool.type, content=tool.output, tool_call_id=tool_call_id))
         if choice.message.content is not None:
             items.append(TextPart(content=choice.message.content))
         if choice.message.tool_calls is not None:

pydantic_ai_slim/pyproject.toml

@@ -66,7 +66,7 @@ cohere = ["cohere>=5.13.11; platform_system != 'Emscripten'"]
 vertexai = ["google-auth>=2.36.0", "requests>=2.32.2"]
 google = ["google-genai>=1.15.0"]
 anthropic = ["anthropic>=0.52.0"]
-groq = ["groq>=0.15.0"]
+groq = ["groq>=0.25.0"]
 mistral = ["mistralai>=1.2.5"]
 bedrock = ["boto3>=1.35.74"]
 # Tools

tests/models/test_groq.py

@@ -22,6 +22,8 @@
     ModelRequest,
     ModelResponse,
     RetryPromptPart,
+    ServerToolCallPart,
+    ServerToolReturnPart,
     SystemPromptPart,
     TextPart,
     ToolCallPart,
@@ -692,3 +694,105 @@ async def test_groq_model_web_search_tool(allow_model_requests: None, groq_api_k
 
     result = await agent.run('What day is today?')
     assert result.output == snapshot('The current day is Tuesday.')
+    assert result.all_messages() == snapshot(
+        [
+            ModelRequest(parts=[UserPromptPart(content='What day is today?', timestamp=IsDatetime())]),
+            ModelResponse(
+                parts=[
+                    ServerToolCallPart(
+                        tool_name='search',
+                        args='{"query": "What is the current date?"}',
+                        tool_call_id='0',
+                        model_name='groq',
+                    ),
+                    ServerToolReturnPart(
+                        tool_name='search',
+                        content="""\
+Title: Today's Date - Find Out Quickly What's The Date Today ️
+URL: https://calendarhours.com/todays-date/
+Content: The current date in RFC 2822 Format with shortened day of week, numerical date, three-letter month abbreviation, year, time, and time zone is: Tue, 13 May 2025 06:07:56 -0400; The current date in Unix Epoch Format with number of seconds that have elapsed since January 1, 1970 (midnight UTC/GMT) is:
+Score: 0.8299
+
+Title: Today's Date | Current date now - MaxTables
+URL: https://maxtables.com/tools/todays-date.html
+Content: The current date, including day of the week, month, day, and year. The exact time, down to seconds. Details on the time zone, its location, and its GMT difference. A tool to select the present date. A visual calendar chart. Why would I need to check Today's Date on this platform instead of my device?
+Score: 0.7223
+
+Title: Current Time and Date - Exact Time!
+URL: https://time-and-calendar.com/
+Content: The actual time is: Mon May 12 2025 22:14:39 GMT-0700 (Pacific Daylight Time) Your computer time is: 22:14:38 The time of your computer is synchronized with our web server. This mean that it is synchonizing in real time with our server clock.
+Score: 0.6799
+
+Title: Today's Date - CalendarDate.com
+URL: https://www.calendardate.com/todays.htm
+Content: Details about today's date with count of days, weeks, and months, Sun and Moon cycles, Zodiac signs and holidays. Monday May 12, 2025 . Home; Calendars. 2025 Calendar; ... Current Season Today: Spring with 40 days until the start of Summer. S. Hemishpere flip seasons - i.e. Winter is Summer.
+Score: 0.6416
+
+Title: What is the date today | Today's Date
+URL: https://www.datetoday.info/
+Content: Master time tracking with Today's Date. Stay updated with real-time information on current date, time, day of the week, days left in the week, current day and remaining days of the year. Explore time in globally accepted formats. Keep up with the current week and month, along with the remaining weeks and months for the year. Embrace efficient time tracking with Today's Date.
+Score: 0.6282
+
+Title: Explore Today's Date, Time Zones, Holidays & More
+URL: https://whatdateis.today/
+Content: Check what date and time it is today (May 8, 2025). View current time across different time zones, upcoming holidays, and use our date calculator. Your one-stop destination for all date and time information.
+Score: 0.6181
+
+Title: Today's Date and Time - Date and Time Tools
+URL: https://todaysdatetime.com/
+Content: Discover today's exact date and time, learn about time zones, date formats, and explore our comprehensive collection of date and time tools including calculators, converters, and calendars. ... Get the exact current date and time, along with powerful calculation tools for all your scheduling needs. 12h. Today. Day 76 of year (366) Yesterday
+Score: 0.5456
+
+Title: Current Time Now - What time is it? - RapidTables.com
+URL: https://www.rapidtables.com/tools/current-time.html
+Content: This page includes the following information: Current time: hours, minutes, seconds. Today's date: day of week, month, day, year. Time zone with location and GMT offset.
+Score: 0.4255
+
+Title: Current Time
+URL: https://www.timeanddate.com/
+Content: Welcome to the world's top site for time, time zones, and astronomy. Organize your life with free online info and tools you can rely on. No sign-up needed. Sign in. News. News Home; Astronomy News; ... Current Time. Monday May 12, 2025 Roanoke Rapids, North Carolina, USA. Set home location. 11:27: 03 pm. World Clock.
+Score: 0.3876
+
+Title: Current local time in the United States - World clock
+URL: https://dateandtime.info/country.php?code=US
+Content: Time and Date of DST Change Time Change; DST started: Sunday, March 9, 2025 at 2:00 AM: The clocks were put forward an hour to 3:00 AM. DST ends: Sunday, November 2, 2025 at 2:00 AM: The clocks will be put back an hour to 1:00 AM. DST starts: Sunday, March 8, 2026 at 2:00 AM: The clocks will be put forward an hour to 3:00 AM.
+Score: 0.3042
+
+Title: Time.is - exact time, any time zone
+URL: https://time.is/
+Content: 7 million locations, 58 languages, synchronized with atomic clock time. Time.is. Get Time.is Ad-free! Exact time now: 05:08:45. Tuesday, 13 May, 2025, week 20. Sun: ↑ 05:09 ↓ 20:45 (15h 36m) - More info - Make London time default - Remove from favorite locations
+Score: 0.2796
+
+Title: Time in United States now
+URL: https://time.is/United_States
+Content: Exact time now, time zone, time difference, sunrise/sunset time and key facts for United States. Time.is. Get Time.is Ad-free! Time in United States now . 11:17:42 PM. Monday, May 12, 2025. United States (incl. dependent territories) has 11 time zones. The time zone for the capital Washington, D.C. is used here.
+Score: 0.2726
+
+Title: Current Local Time in the United States - timeanddate.com
+URL: https://www.timeanddate.com/worldclock/usa
+Content: United States time now. USA time zones and time zone map with current time in each state.
+Score: 0.2519
+
+Title: Current local time in United States - World Time Clock & Map
+URL: https://24timezones.com/United-States/time
+Content: Check the current time in United States and time zone information, the UTC offset and daylight saving time dates in 2025.
+Score: 0.2221
+
+Title: The World Clock — Worldwide - timeanddate.com
+URL: https://www.timeanddate.com/worldclock/
+Content: World time and date for cities in all time zones. International time right now. Takes into account all DST clock changes.
+Score: 0.2134
+
+""",
+                        tool_call_id='0',
+                        timestamp=IsDatetime(),
+                    ),
+                    TextPart(content='The current day is Tuesday.'),
+                ],
+                usage=Usage(requests=1, request_tokens=4287, response_tokens=117, total_tokens=4404),
+                model_name='compound-beta',
+                timestamp=IsDatetime(),
+                vendor_id='stub',
+            ),
+        ]
+    )