fix: tests messages

Author: cosmyo

packages/uipath-openai-agents/tests/cli/test_init.py

@@ -4,7 +4,7 @@
 import os
 
 from click.testing import CliRunner
-from uipath._cli.cli_init import init
+from uipath._cli import cli
 
 
 class TestInit:
@@ -30,7 +30,7 @@ def test_init_basic_config_generation(
             with open("openai_agents.json", "w") as f:
                 f.write(openai_agents_config)
 
-            result = runner.invoke(init)
+            result = runner.invoke(cli, ["init"])
             assert result.exit_code == 0
             assert os.path.exists("entry-points.json")
 
@@ -54,9 +54,9 @@ def test_init_basic_config_generation(
                 assert isinstance(input_schema["properties"], dict)
                 assert isinstance(input_schema["required"], list)
 
-                # OpenAI agents use default message input
-                assert "message" in input_schema["properties"]
-                assert "message" in input_schema["required"]
+                # OpenAI agents use default messages input
+                assert "messages" in input_schema["properties"]
+                assert "messages" in input_schema["required"]
 
                 # Verify output schema (default since no output_type specified)
                 assert "output" in entry
@@ -87,7 +87,7 @@ def test_init_translation_agent_config_generation(
             with open("openai_agents.json", "w") as f:
                 f.write(openai_agents_config)
 
-            result = runner.invoke(init)
+            result = runner.invoke(cli, ["init"])
             assert result.exit_code == 0
             assert os.path.exists("entry-points.json")
 
@@ -112,8 +112,8 @@ def test_init_translation_agent_config_generation(
                 input_schema = translation_entry["input"]
                 assert input_schema["type"] == "object"
                 assert "properties" in input_schema
-                assert "message" in input_schema["properties"]
-                assert "message" in input_schema["required"]
+                assert "messages" in input_schema["properties"]
+                assert "messages" in input_schema["required"]
 
                 # Verify output schema from agent's output_type
                 assert "output" in translation_entry

packages/uipath-openai-agents/tests/test_agent_as_tools_schema.py

@@ -33,17 +33,17 @@ def test_agent_as_tools_input_schema():
 
     # OpenAI Agents use messages as input (not custom types)
     input_props = schema["input"]["properties"]
-    assert "message" in input_props
+    assert "messages" in input_props
 
-    # Verify message field accepts string or array
-    assert "anyOf" in input_props["message"]
-    types = [t.get("type") for t in input_props["message"]["anyOf"]]
+    # Verify messages field accepts string or array
+    assert "anyOf" in input_props["messages"]
+    types = [t.get("type") for t in input_props["messages"]["anyOf"]]
     assert "string" in types
     assert "array" in types
 
     # Check required fields
     assert "required" in schema["input"]
-    assert "message" in schema["input"]["required"]
+    assert "messages" in schema["input"]["required"]
 
 
 def test_agent_as_tools_output_schema():
@@ -86,7 +86,7 @@ def test_agent_as_tools_schema_metadata():
     # Input uses default messages format (no custom title/description)
     assert "input" in schema
     assert "properties" in schema["input"]
-    assert "message" in schema["input"]["properties"]
+    assert "messages" in schema["input"]["properties"]
 
     # Check output metadata from agent's output_type
     assert "title" in schema["output"]

packages/uipath-openai-agents/tests/test_schema_inference.py

@@ -35,9 +35,9 @@ def test_schema_inference_from_agent_output_type():
     # Check input schema - should be default messages format
     assert "input" in schema
     assert "properties" in schema["input"]
-    assert "message" in schema["input"]["properties"]
+    assert "messages" in schema["input"]["properties"]
     assert "required" in schema["input"]
-    assert "message" in schema["input"]["required"]
+    assert "messages" in schema["input"]["required"]
 
     # Check output schema - extracted from agent's output_type
     assert "output" in schema
@@ -60,9 +60,9 @@ def test_schema_fallback_without_types():
     """Test that schemas fall back to defaults when no types are provided."""
     schema = get_entrypoints_schema(test_agent)
 
-    # Should use default message-based input schema
+    # Should use default messages-based input schema
     assert "input" in schema
-    assert "message" in schema["input"]["properties"]
+    assert "messages" in schema["input"]["properties"]
 
     # Should fall back to default result-based output
     assert "output" in schema
@@ -73,9 +73,9 @@ def test_schema_with_plain_agent():
     """Test schema extraction with a plain agent."""
     schema = get_entrypoints_schema(test_agent)
 
-    # Should use default message input
+    # Should use default messages input
     assert "input" in schema
-    assert "message" in schema["input"]["properties"]
+    assert "messages" in schema["input"]["properties"]
 
     # Should use default result output
     assert "output" in schema