Update manually maintained files to ensure backward compatibility

Author: fern-support

.fernignore

@@ -11,6 +11,7 @@ src/cohere/utils.py
 src/cohere/overrides.py
 src/cohere/config.py
 src/cohere/manually_maintained
+src/cohere/manually_maintained/__init__.py
 src/cohere/bedrock_client.py
 src/cohere/aws_client.py
 src/cohere/sagemaker_client.py

src/cohere/aliases.py

@@ -1,3 +1,7 @@
+# Import overrides early to ensure they're applied before types are used
+# This is necessary for backwards compatibility patches like ToolCallV2.id being optional
+from . import overrides  # noqa: F401
+
 from .v2 import (
     ContentDeltaV2ChatStreamResponse,
     ContentEndV2ChatStreamResponse,

src/cohere/manually_maintained/__init__.py

@@ -0,0 +1,3 @@
+# This module ensures overrides are applied early in the import process
+# Import overrides to trigger backwards compatibility patches
+from .. import overrides  # noqa: F401

src/cohere/overrides.py

@@ -1,7 +1,8 @@
 import typing
+import uuid
 
 from . import EmbedByTypeResponseEmbeddings
-from .core.pydantic_utilities import _get_model_fields, Model
+from .core.pydantic_utilities import _get_model_fields, Model, IS_PYDANTIC_V2
 
 from pprint import pprint
 
@@ -33,6 +34,31 @@ def allow_access_to_aliases(self: typing.Type["Model"], name):
         f"'{type(self).__name__}' object has no attribute '{name}'")
 
 
+def make_tool_call_v2_id_optional(cls):
+    """
+    Override ToolCallV2 to make the 'id' field optional with a default UUID.
+    This ensures backward compatibility with code that doesn't provide an id.
+
+    We wrap the __init__ method to inject a default id before Pydantic validation runs.
+    """
+    # Store the original __init__ method
+    original_init = cls.__init__
+
+    def patched_init(self, /, **data):
+        """Patched __init__ that injects default id if not provided."""
+        # Inject default UUID if 'id' is not in the data
+        if 'id' not in data:
+            data['id'] = str(uuid.uuid4())
+
+        # Call the original __init__ with modified data
+        original_init(self, **data)
+
+    # Replace the __init__ method
+    cls.__init__ = patched_init
+
+    return cls
+
+
 def run_overrides():
     """
         These are overrides to allow us to make changes to generated code without touching the generated files themselves.
@@ -41,3 +67,14 @@ def run_overrides():
 
     # Override to allow access to aliases in EmbedByTypeResponseEmbeddings eg embeddings.float rather than embeddings.float_
     setattr(EmbedByTypeResponseEmbeddings, "__getattr__", allow_access_to_aliases)
+
+    # Import ToolCallV2 lazily to avoid circular dependency issues
+    from . import ToolCallV2
+
+    # Override ToolCallV2 to make id field optional with default UUID
+    make_tool_call_v2_id_optional(ToolCallV2)
+
+
+# Run overrides immediately at module import time to ensure they're applied
+# before any code tries to use the modified classes
+run_overrides()

src/cohere/utils.py

@@ -13,6 +13,9 @@
 from .datasets import DatasetsCreateResponse, DatasetsGetResponse
 from .overrides import get_fields
 
+# Note: utils.py does NOT call run_overrides() itself - that's done in client.py
+# which imports utils.py. This ensures overrides are applied when client is used.
+
 
 def get_terminal_states():
     return get_success_states() | get_failed_states()