chore(types): rebuild Pydantic models after all types are defined

Author: stainless-app[bot]

src/do_gradientai/types/__init__.py

@@ -2,6 +2,18 @@
 
 from __future__ import annotations
 
+from . import (
+    agents,
+    models,
+    api_agent,
+    api_workspace,
+    agent_create_response,
+    agent_delete_response,
+    agent_update_response,
+    agent_retrieve_response,
+    agent_update_status_response,
+)
+from .. import _compat
 from .shared import (
     Size as Size,
     Image as Image,
@@ -80,3 +92,56 @@
 from .gpu_droplet_list_firewalls_response import GPUDropletListFirewallsResponse as GPUDropletListFirewallsResponse
 from .gpu_droplet_list_neighbors_response import GPUDropletListNeighborsResponse as GPUDropletListNeighborsResponse
 from .gpu_droplet_list_snapshots_response import GPUDropletListSnapshotsResponse as GPUDropletListSnapshotsResponse
+
+# Rebuild cyclical models only after all modules are imported.
+# This ensures that, when building the deferred (due to cyclical references) model schema,
+# Pydantic can resolve the necessary references.
+# See: https://github.com/pydantic/pydantic/issues/11250 for more context.
+if _compat.PYDANTIC_V2:
+    api_agent.APIAgent.model_rebuild(_parent_namespace_depth=0)
+    api_workspace.APIWorkspace.model_rebuild(_parent_namespace_depth=0)
+    agent_create_response.AgentCreateResponse.model_rebuild(_parent_namespace_depth=0)
+    agent_retrieve_response.AgentRetrieveResponse.model_rebuild(_parent_namespace_depth=0)
+    agent_update_response.AgentUpdateResponse.model_rebuild(_parent_namespace_depth=0)
+    agent_delete_response.AgentDeleteResponse.model_rebuild(_parent_namespace_depth=0)
+    agent_update_status_response.AgentUpdateStatusResponse.model_rebuild(_parent_namespace_depth=0)
+    agents.evaluation_metrics.workspace_create_response.WorkspaceCreateResponse.model_rebuild(_parent_namespace_depth=0)
+    agents.evaluation_metrics.workspace_retrieve_response.WorkspaceRetrieveResponse.model_rebuild(
+        _parent_namespace_depth=0
+    )
+    agents.evaluation_metrics.workspace_update_response.WorkspaceUpdateResponse.model_rebuild(_parent_namespace_depth=0)
+    agents.evaluation_metrics.workspace_list_response.WorkspaceListResponse.model_rebuild(_parent_namespace_depth=0)
+    agents.evaluation_metrics.workspaces.agent_list_response.AgentListResponse.model_rebuild(_parent_namespace_depth=0)
+    agents.evaluation_metrics.workspaces.agent_move_response.AgentMoveResponse.model_rebuild(_parent_namespace_depth=0)
+    agents.function_create_response.FunctionCreateResponse.model_rebuild(_parent_namespace_depth=0)
+    agents.function_update_response.FunctionUpdateResponse.model_rebuild(_parent_namespace_depth=0)
+    agents.function_delete_response.FunctionDeleteResponse.model_rebuild(_parent_namespace_depth=0)
+    agents.api_link_knowledge_base_output.APILinkKnowledgeBaseOutput.model_rebuild(_parent_namespace_depth=0)
+    agents.knowledge_base_detach_response.KnowledgeBaseDetachResponse.model_rebuild(_parent_namespace_depth=0)
+    agents.route_view_response.RouteViewResponse.model_rebuild(_parent_namespace_depth=0)
+    models.providers.anthropic_list_agents_response.AnthropicListAgentsResponse.model_rebuild(_parent_namespace_depth=0)
+    models.providers.openai_retrieve_agents_response.OpenAIRetrieveAgentsResponse.model_rebuild(
+        _parent_namespace_depth=0
+    )
+else:
+    api_agent.APIAgent.update_forward_refs()  # type: ignore
+    api_workspace.APIWorkspace.update_forward_refs()  # type: ignore
+    agent_create_response.AgentCreateResponse.update_forward_refs()  # type: ignore
+    agent_retrieve_response.AgentRetrieveResponse.update_forward_refs()  # type: ignore
+    agent_update_response.AgentUpdateResponse.update_forward_refs()  # type: ignore
+    agent_delete_response.AgentDeleteResponse.update_forward_refs()  # type: ignore
+    agent_update_status_response.AgentUpdateStatusResponse.update_forward_refs()  # type: ignore
+    agents.evaluation_metrics.workspace_create_response.WorkspaceCreateResponse.update_forward_refs()  # type: ignore
+    agents.evaluation_metrics.workspace_retrieve_response.WorkspaceRetrieveResponse.update_forward_refs()  # type: ignore
+    agents.evaluation_metrics.workspace_update_response.WorkspaceUpdateResponse.update_forward_refs()  # type: ignore
+    agents.evaluation_metrics.workspace_list_response.WorkspaceListResponse.update_forward_refs()  # type: ignore
+    agents.evaluation_metrics.workspaces.agent_list_response.AgentListResponse.update_forward_refs()  # type: ignore
+    agents.evaluation_metrics.workspaces.agent_move_response.AgentMoveResponse.update_forward_refs()  # type: ignore
+    agents.function_create_response.FunctionCreateResponse.update_forward_refs()  # type: ignore
+    agents.function_update_response.FunctionUpdateResponse.update_forward_refs()  # type: ignore
+    agents.function_delete_response.FunctionDeleteResponse.update_forward_refs()  # type: ignore
+    agents.api_link_knowledge_base_output.APILinkKnowledgeBaseOutput.update_forward_refs()  # type: ignore
+    agents.knowledge_base_detach_response.KnowledgeBaseDetachResponse.update_forward_refs()  # type: ignore
+    agents.route_view_response.RouteViewResponse.update_forward_refs()  # type: ignore
+    models.providers.anthropic_list_agents_response.AnthropicListAgentsResponse.update_forward_refs()  # type: ignore
+    models.providers.openai_retrieve_agents_response.OpenAIRetrieveAgentsResponse.update_forward_refs()  # type: ignore