fix: pydantic ai proper schema and events payload

cristipufu · cristipufu · commit 5ee2ee0e9dc2 · 2026-03-01T17:12:32.000+02:00
diff --git a/README.md b/README.md
@@ -13,6 +13,7 @@ All packages extend the [UiPath Python SDK](https://github.com/UiPath/uipath-pyt
 | [LlamaIndex](https://www.llamaindex.ai/) | [![PyPI](https://img.shields.io/pypi/v/uipath-llamaindex)](https://pypi.org/project/uipath-llamaindex/) | [![Downloads](https://img.shields.io/pypi/dm/uipath-llamaindex.svg)](https://pypi.org/project/uipath-llamaindex/) | [README](packages/uipath-llamaindex/README.md) · [Docs](https://uipath.github.io/uipath-python/llamaindex/quick_start/) · [Samples](packages/uipath-llamaindex/samples/) |
 | [Microsoft Agent Framework](https://github.com/microsoft/agent-framework) | [![PyPI](https://img.shields.io/pypi/v/uipath-agent-framework)](https://pypi.org/project/uipath-agent-framework/) | [![Downloads](https://img.shields.io/pypi/dm/uipath-agent-framework.svg)](https://pypi.org/project/uipath-agent-framework/) | [README](packages/uipath-agent-framework/README.md) · [Samples](packages/uipath-agent-framework/samples/) |
 | [OpenAI Agents](https://github.com/openai/openai-agents-python) | [![PyPI](https://img.shields.io/pypi/v/uipath-openai-agents)](https://pypi.org/project/uipath-openai-agents/) | [![Downloads](https://img.shields.io/pypi/dm/uipath-openai-agents.svg)](https://pypi.org/project/uipath-openai-agents/) | [README](packages/uipath-openai-agents/README.md) · [Docs](https://uipath.github.io/uipath-python/openai-agents/quick_start/) · [Samples](packages/uipath-openai-agents/samples/) |
+| [PydanticAI](https://github.com/pydantic/pydantic-ai) | [![PyPI](https://img.shields.io/pypi/v/uipath-pydantic-ai)](https://pypi.org/project/uipath-pydantic-ai/) | [![Downloads](https://img.shields.io/pypi/dm/uipath-pydantic-ai.svg)](https://pypi.org/project/uipath-pydantic-ai/) | [README](packages/uipath-pydantic-ai/README.md) · [Samples](packages/uipath-pydantic-ai/samples/) |
 
 
 ## Structure
@@ -25,7 +26,8 @@ uipath-integrations-python/
     ├── uipath-llamaindex/      # LlamaIndex runtime
     ├── uipath-openai-agents/   # OpenAI Agents runtime
     ├── uipath-google-adk/      # Google ADK runtime
-    └── uipath-agent-framework/ # Microsoft Agent Framework runtime
+    ├── uipath-agent-framework/ # Microsoft Agent Framework runtime
+    └── uipath-pydantic-ai/     # PydanticAI runtime
 ```
 
 ## Development
diff --git a/packages/uipath-pydantic-ai/pyproject.toml b/packages/uipath-pydantic-ai/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "uipath-pydantic-ai"
-version = "0.0.1"
+version = "0.0.2"
 description = "Python SDK that enables developers to build and deploy PydanticAI agents to the UiPath Cloud Platform"
 readme = "README.md"
 requires-python = ">=3.11"
diff --git a/packages/uipath-pydantic-ai/samples/graph-flow/README.md b/packages/uipath-pydantic-ai/samples/graph-flow/README.md
@@ -34,7 +34,7 @@ flowchart TB
 ## Run
 
 ```
-uipath run agent '{"messages": "Write a blog post about the future of AI agents"}'
+uipath run agent '{"messages": [{"contentParts": [{"data": {"inline": "Write a blog post about the future of AI agents"}}], "role": "user"}]}'
 ```
 
 ## Debug
diff --git a/packages/uipath-pydantic-ai/samples/multi-agent/README.md b/packages/uipath-pydantic-ai/samples/multi-agent/README.md
@@ -19,7 +19,7 @@ flowchart TB
 ## Run
 
 ```
-uipath run agent '{"messages": "Write a report about machine learning"}'
+uipath run agent '{"messages": [{"contentParts": [{"data": {"inline": "Write a report about machine learning"}}], "role": "user"}]}'
 ```
 
 ## Debug
diff --git a/packages/uipath-pydantic-ai/samples/programmatic-handoff/README.md b/packages/uipath-pydantic-ai/samples/programmatic-handoff/README.md
@@ -24,7 +24,7 @@ flowchart TB
 ## Run
 
 ```
-uipath run agent '{"messages": "I was charged twice for my subscription last month"}'
+uipath run agent '{"messages": [{"contentParts": [{"data": {"inline": "I was charged twice for my subscription last month"}}], "role": "user"}]}'
 ```
 
 ## Debug
diff --git a/packages/uipath-pydantic-ai/samples/quickstart-agent/README.md b/packages/uipath-pydantic-ai/samples/quickstart-agent/README.md
@@ -19,11 +19,11 @@ flowchart TB
 ## Run
 
 ```
-uipath run agent '{"messages": "What is the weather in San Francisco?"}'
+uipath run agent '{"messages": [{"contentParts": [{"data": {"inline": "What is the weather in San Francisco?"}}], "role": "user"}]}'
 ```
 
 ## Debug
 
 ```
-uipath dev web
+uipath dev
 ```
diff --git a/packages/uipath-pydantic-ai/samples/quickstart-agent/main.py b/packages/uipath-pydantic-ai/samples/quickstart-agent/main.py
@@ -1,10 +1,10 @@
 import httpx
-from pydantic_ai import Agent
+from pydantic_ai import Agent, RunContext
 
 from uipath_pydantic_ai.chat import UiPathChatOpenAI
 
 
-def get_weather(ctx, location: str) -> str:
+def get_weather(ctx: RunContext[None], location: str) -> str:
     """Get the current weather for a location using the Open-Meteo API.
 
     Args:
diff --git a/packages/uipath-pydantic-ai/samples/quickstart-agent/pyproject.toml b/packages/uipath-pydantic-ai/samples/quickstart-agent/pyproject.toml
@@ -8,7 +8,7 @@ requires-python = ">=3.11"
 dependencies = [
     "uipath",
     "uipath-pydantic-ai",
-    "pydantic-ai>=0.1.0",
+    "pydantic-ai>=1.63.0",
 ]
 
 [dependency-groups]
diff --git a/packages/uipath-pydantic-ai/src/uipath_pydantic_ai/chat/openai.py b/packages/uipath-pydantic-ai/src/uipath_pydantic_ai/chat/openai.py
@@ -5,7 +5,7 @@
 
 import httpx
 from openai import AsyncOpenAI, OpenAI
-from pydantic_ai.models.openai import OpenAIModel
+from pydantic_ai.models.openai import OpenAIChatModel
 from pydantic_ai.providers.openai import OpenAIProvider
 from uipath._utils._ssl_context import get_httpx_client_kwargs
 from uipath.utils import EndpointManager
@@ -74,7 +74,7 @@ class UiPathChatOpenAI:
 
     This client wraps the OpenAI SDK and configures it to use UiPath's
     LLM Gateway endpoints with proper authentication and headers.
-    Returns PydanticAI-compatible OpenAIModel instances.
+    Returns PydanticAI-compatible OpenAIChatModel instances.
 
     Example:
         ```python
@@ -173,7 +173,7 @@ def __init__(
         )
 
         # Create PydanticAI-compatible model
-        self._model = OpenAIModel(
+        self._model = OpenAIChatModel(
             self._model_name,
             provider=OpenAIProvider(openai_client=self._async_client),
         )
@@ -225,8 +225,8 @@ def _build_base_url(self) -> str:
             raise ValueError("UIPATH_URL environment variable is required")
 
     @property
-    def model(self) -> OpenAIModel:
-        """Get the PydanticAI-compatible OpenAIModel."""
+    def model(self) -> OpenAIChatModel:
+        """Get the PydanticAI-compatible OpenAIChatModel."""
         return self._model
 
     @property
diff --git a/packages/uipath-pydantic-ai/src/uipath_pydantic_ai/runtime/factory.py b/packages/uipath-pydantic-ai/src/uipath_pydantic_ai/runtime/factory.py
@@ -58,6 +58,7 @@ def _setup_instrumentation(self) -> None:
                 self.context.trace_manager.add_span_processor(
                     OpenInferenceSpanProcessor()
                 )
+                Agent.instrument_all()
         except Exception:
             pass
 
diff --git a/packages/uipath-pydantic-ai/src/uipath_pydantic_ai/runtime/runtime.py b/packages/uipath-pydantic-ai/src/uipath_pydantic_ai/runtime/runtime.py
diff --git a/packages/uipath-pydantic-ai/src/uipath_pydantic_ai/runtime/schema.py b/packages/uipath-pydantic-ai/src/uipath_pydantic_ai/runtime/schema.py
diff --git a/packages/uipath-pydantic-ai/tests/test_runtime.py b/packages/uipath-pydantic-ai/tests/test_runtime.py
diff --git a/packages/uipath-pydantic-ai/tests/test_schema.py b/packages/uipath-pydantic-ai/tests/test_schema.py
diff --git a/packages/uipath-pydantic-ai/uv.lock b/packages/uipath-pydantic-ai/uv.lock

Original file line number	Diff line number	Diff line change
`@@ -8,7 +8,7 @@ requires-python = ">=3.11"`
`8`	`8`	`dependencies = [`
`9`	`9`	`"uipath",`
`10`	`10`	`"uipath-pydantic-ai",`
`11`		`- "pydantic-ai>=0.1.0",`
	`11`	`+ "pydantic-ai>=1.63.0",`
`12`	`12`	`]`
`13`	`13`
`14`	`14`	`[dependency-groups]`
Original file line number	Diff line number	Diff line change
`@@ -58,6 +58,7 @@ def _setup_instrumentation(self) -> None:`
`58`	`58`	`self.context.trace_manager.add_span_processor(`
`59`	`59`	`OpenInferenceSpanProcessor()`
`60`	`60`	`)`
	`61`	`+ Agent.instrument_all()`
`61`	`62`	`except Exception:`
`62`	`63`	`pass`
`63`	`64`