pydantic-ai-slim 0.0.28__tar.gz → 0.0.29__tar.gz

{pydantic_ai_slim-0.0.28 → pydantic_ai_slim-0.0.29}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai-slim
-Version: 0.0.28
+Version: 0.0.29
 Summary: Agent Framework / shim to use Pydantic with LLMs, slim package
 Author-email: Samuel Colvin <samuel@pydantic.dev>
 License-Expression: MIT
@@ -29,7 +29,7 @@ Requires-Dist: exceptiongroup; python_version < '3.11'
 Requires-Dist: griffe>=1.3.2
 Requires-Dist: httpx>=0.27
 Requires-Dist: logfire-api>=1.2.0
-Requires-Dist: pydantic-graph==0.0.28
+Requires-Dist: pydantic-graph==0.0.29
 Requires-Dist: pydantic>=2.10
 Provides-Extra: anthropic
 Requires-Dist: anthropic>=0.40.0; extra == 'anthropic'

{pydantic_ai_slim-0.0.28 → pydantic_ai_slim-0.0.29}/pydantic_ai/agent.py

@@ -1,6 +1,5 @@
 from __future__ import annotations as _annotations
 
-import asyncio
 import dataclasses
 import inspect
 from collections.abc import AsyncIterator, Awaitable, Iterator, Sequence
@@ -13,6 +12,7 @@ import logfire_api
 from typing_extensions import TypeVar, deprecated
 
 from pydantic_graph import BaseNode, End, Graph, GraphRun, GraphRunContext
+from pydantic_graph._utils import get_event_loop
 
 from . import (
     _agent_graph,
@@ -538,7 +538,7 @@ class Agent(Generic[AgentDepsT, ResultDataT]):
         """
         if infer_name and self.name is None:
             self._infer_name(inspect.currentframe())
-        return asyncio.get_event_loop().run_until_complete(
+        return get_event_loop().run_until_complete(
             self.run(
                 user_prompt,
                 result_type=result_type,

{pydantic_ai_slim-0.0.28 → pydantic_ai_slim-0.0.29}/pydantic_ai/common_tools/duckduckgo.py

@@ -1,3 +1,4 @@
+import functools
 from dataclasses import dataclass
 from typing import TypedDict
 
@@ -39,6 +40,9 @@ class DuckDuckGoSearchTool:
     client: DDGS
     """The DuckDuckGo search client."""
 
+    max_results: int | None = None
+    """The maximum number of results. If None, returns results only from the first response."""
+
     async def __call__(self, query: str) -> list[DuckDuckGoResult]:
         """Searches DuckDuckGo for the given query and returns the results.
 
@@ -48,16 +52,22 @@ class DuckDuckGoSearchTool:
         Returns:
             The search results.
         """
-        results = await anyio.to_thread.run_sync(self.client.text, query)
+        search = functools.partial(self.client.text, max_results=self.max_results)
+        results = await anyio.to_thread.run_sync(search, query)
         if len(results) == 0:
             raise RuntimeError('No search results found.')
         return duckduckgo_ta.validate_python(results)
 
 
-def duckduckgo_search_tool(duckduckgo_client: DDGS | None = None):
-    """Creates a DuckDuckGo search tool."""
+def duckduckgo_search_tool(duckduckgo_client: DDGS | None = None, max_results: int | None = None):
+    """Creates a DuckDuckGo search tool.
+
+    Args:
+        duckduckgo_client: The DuckDuckGo search client.
+        max_results: The maximum number of results. If None, returns results only from the first response.
+    """
     return Tool(
-        DuckDuckGoSearchTool(client=duckduckgo_client or DDGS()).__call__,
+        DuckDuckGoSearchTool(client=duckduckgo_client or DDGS(), max_results=max_results).__call__,
         name='duckduckgo_search',
         description='Searches DuckDuckGo for the given query and returns the results.',
     )

{pydantic_ai_slim-0.0.28 → pydantic_ai_slim-0.0.29}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "pydantic-ai-slim"
-version = "0.0.28"
+version = "0.0.29"
 description = "Agent Framework / shim to use Pydantic with LLMs, slim package"
 authors = [{ name = "Samuel Colvin", email = "samuel@pydantic.dev" }]
 license = "MIT"
@@ -37,7 +37,7 @@ dependencies = [
     "httpx>=0.27",
     "logfire-api>=1.2.0",
     "pydantic>=2.10",
-    "pydantic-graph==0.0.28",
+    "pydantic-graph==0.0.29",
     "exceptiongroup; python_version < '3.11'",
 ]