casual-llm 0.4.2__tar.gz → 0.4.3__tar.gz

{casual_llm-0.4.2/src/casual_llm.egg-info → casual_llm-0.4.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: casual-llm
-Version: 0.4.2
+Version: 0.4.3
 Summary: Lightweight LLM provider abstraction with standardized message models
 Author-email: Alex Stansfield <alex@casualgenius.com>
 License: MIT

{casual_llm-0.4.2 → casual_llm-0.4.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "casual-llm"
-version = "0.4.2"
+version = "0.4.3"
 description = "Lightweight LLM provider abstraction with standardized message models"
 readme = "README.md"
 requires-python = ">=3.10"

{casual_llm-0.4.2 → casual_llm-0.4.3}/src/casual_llm/__init__.py

@@ -7,7 +7,7 @@ A simple, protocol-based library for working with different LLM providers
 Part of the casual-* ecosystem of lightweight AI tools.
 """
 
-__version__ = "0.4.2"
+__version__ = "0.4.3"
 
 # Model configuration
 from casual_llm.config import ModelConfig, Provider

{casual_llm-0.4.2 → casual_llm-0.4.3}/src/casual_llm/tools.py

@@ -7,6 +7,7 @@ Provides unified tool models compatible with Ollama, OpenAI, and MCP.
 from __future__ import annotations
 
 from typing import Any
+
 from pydantic import BaseModel, Field
 
 
@@ -14,10 +15,13 @@ class ToolParameter(BaseModel):
     """
     A single tool parameter definition following JSON Schema.
 
-    Supports common JSON Schema fields for describing function parameters.
+    Supports common JSON Schema fields for describing function parameters,
+    including union types via anyOf/oneOf schemas.
     """
 
-    type: str = Field(..., description="JSON Schema type (string, number, object, array, etc.)")
+    type: str | None = Field(
+        None, description="JSON Schema type (string, number, object, array, etc.)"
+    )
     description: str | None = Field(None, description="Parameter description")
     enum: list[Any] | None = Field(None, description="Allowed values for enum types")
     items: dict[str, Any] | None = Field(None, description="Array item schema")
@@ -26,6 +30,8 @@ class ToolParameter(BaseModel):
     )
     required: list[str] | None = Field(None, description="Required properties for nested objects")
     default: Any | None = Field(None, description="Default value")
+    anyOf: list[dict[str, Any]] | None = Field(None, description="Union type schemas")
+    oneOf: list[dict[str, Any]] | None = Field(None, description="Exclusive union type schemas")
 
     model_config = {"extra": "allow"}  # Allow additional JSON Schema fields
 

{casual_llm-0.4.2 → casual_llm-0.4.3/src/casual_llm.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: casual-llm
-Version: 0.4.2
+Version: 0.4.3
 Summary: Lightweight LLM provider abstraction with standardized message models
 Author-email: Alex Stansfield <alex@casualgenius.com>
 License: MIT

{casual_llm-0.4.2 → casual_llm-0.4.3}/tests/test_tools.py

@@ -71,6 +71,35 @@ class TestToolParameter:
         assert dumped["maxLength"] == 100
         assert dumped["pattern"] == "^[a-z]+$"
 
+    def test_anyof_parameter(self):
+        """Test parameter with anyOf union type."""
+        param = ToolParameter(
+            anyOf=[{"type": "string"}, {"type": "null"}],
+            description="Optional string",
+        )
+        assert param.type is None
+        assert param.anyOf is not None
+
+    def test_oneof_parameter(self):
+        """Test parameter with oneOf exclusive union type."""
+        param = ToolParameter(
+            oneOf=[{"type": "string"}, {"type": "number"}],
+            description="String or number",
+        )
+        assert param.type is None
+        assert param.oneOf is not None
+
+    def test_anyof_serialization(self):
+        """Test that anyOf is properly serialized."""
+        param = ToolParameter(
+            anyOf=[{"type": "string"}, {"type": "null"}],
+            description="Optional",
+        )
+        dumped = param.model_dump(exclude_none=True)
+        assert "anyOf" in dumped
+        assert len(dumped["anyOf"]) == 2
+        assert "type" not in dumped  # type is None, excluded
+
 
 class TestTool:
     """Tests for Tool model."""
@@ -188,6 +217,41 @@ class TestTool:
         assert reconstructed.required == original_tool.required
         assert len(reconstructed.parameters) == len(original_tool.parameters)
 
+    def test_input_schema_with_anyof(self):
+        """Test that input_schema preserves anyOf without injecting type."""
+        tool = Tool(
+            name="test",
+            description="Test",
+            parameters={
+                "optional_field": ToolParameter(
+                    anyOf=[{"type": "string"}, {"type": "null"}],
+                    description="An optional string",
+                )
+            },
+        )
+        schema = tool.input_schema
+        # anyOf should be preserved as-is
+        assert "anyOf" in schema["properties"]["optional_field"]
+        # type should NOT be injected (anyOf is valid JSON Schema)
+        assert "type" not in schema["properties"]["optional_field"]
+
+    def test_input_schema_with_oneof(self):
+        """Test that input_schema preserves oneOf without injecting type."""
+        tool = Tool(
+            name="test",
+            description="Test",
+            parameters={
+                "mixed_field": ToolParameter(
+                    oneOf=[{"type": "number"}, {"type": "string"}],
+                )
+            },
+        )
+        schema = tool.input_schema
+        # oneOf should be preserved as-is
+        assert "oneOf" in schema["properties"]["mixed_field"]
+        # type should NOT be injected
+        assert "type" not in schema["properties"]["mixed_field"]
+
 
 class TestOllamaConverters:
     """Tests for Ollama format converters."""