fix: tests

Author: cosmyo

packages/uipath-openai-agents/src/uipath_openai_agents/runtime/_serialize.py

@@ -16,12 +16,12 @@ def serialize_output(output: Any) -> Any:
     if output is None:
         return {}
 
-    # Handle Pydantic models
-    if hasattr(output, "model_dump"):
+    # Handle Pydantic models (but not Pydantic model classes)
+    if hasattr(output, "model_dump") and not isinstance(output, type):
         return serialize_output(output.model_dump(by_alias=True))
-    elif hasattr(output, "dict"):
+    elif hasattr(output, "dict") and not isinstance(output, type):
         return serialize_output(output.dict())
-    elif hasattr(output, "to_dict"):
+    elif hasattr(output, "to_dict") and not isinstance(output, type):
         return serialize_output(output.to_dict())
 
     # Handle dataclasses (but not dataclass types)

packages/uipath-openai-agents/src/uipath_openai_agents/runtime/runtime.py

@@ -219,52 +219,10 @@ def _convert_stream_event_to_runtime_event(
         return None
 
     def _prepare_agent_input(self, input: dict[str, Any] | None) -> str | list[Any]:
-        """
-        Prepare agent input from UiPath input dictionary.
-
-        Supports two input formats:
-        - {"message": "text"} → returns string for Runner.run()
-        - {"messages": [...]} → returns list of message dicts for Runner.run()
-
-        Note: When using sessions, string input is preferred as it doesn't
-        require a session_input_callback.
-
-        Args:
-            input: Input dictionary from UiPath
-
-        Returns:
-            String or list for Runner.run() input parameter
-
-        Raises:
-            ValueError: If input doesn't contain "message" or "messages" field
-        """
-        if not input:
-            raise ValueError(
-                "Input is required. Provide either 'message' (string) or 'messages' (list of message dicts)"
-            )
-
-        # Check for "messages" field (list of message dicts)
-        if "messages" in input:
-            messages = input["messages"]
-            # Ensure it's a list
-            if isinstance(messages, list):
-                return messages
-            else:
-                raise ValueError(
-                    "'messages' field must be a list of message dictionaries"
-                )
-
-        # Check for "message" field (simple string)
-        if "message" in input:
-            message = input["message"]
-            # Return as string (OpenAI Agents SDK handles string → message conversion)
-            return str(message)
-
-        # No valid field found
-        raise ValueError(
-            "Input must contain either 'message' (string) or 'messages' (list of message dicts). "
-            f"Got keys: {list(input.keys())}"
-        )
+        """Prepare agent input from UiPath input dictionary."""
+        if input and "messages" in input and isinstance(input["messages"], list):
+            return input.get("messages", [])
+        return input.get("message", "") if input else ""
 
     def _serialize_message(self, message: Any) -> dict[str, Any]:
         """