|
2 | 2 |
|
3 | 3 | from __future__ import annotations
|
4 | 4 |
|
| 5 | +from . import ( |
| 6 | + agents, |
| 7 | + models, |
| 8 | + api_agent, |
| 9 | + api_workspace, |
| 10 | + agent_create_response, |
| 11 | + agent_delete_response, |
| 12 | + agent_update_response, |
| 13 | + agent_retrieve_response, |
| 14 | + agent_update_status_response, |
| 15 | +) |
| 16 | +from .. import _compat |
| 17 | +from .agents import evaluation_metrics # type: ignore # noqa: F401 |
| 18 | +from .models import providers # type: ignore # noqa: F401 |
5 | 19 | from .shared import (
|
6 | 20 | Size as Size,
|
7 | 21 | Image as Image,
|
|
55 | 69 | from .api_openai_api_key_info import APIOpenAIAPIKeyInfo as APIOpenAIAPIKeyInfo
|
56 | 70 | from .gpu_droplet_list_params import GPUDropletListParams as GPUDropletListParams
|
57 | 71 | from .model_retrieve_response import ModelRetrieveResponse as ModelRetrieveResponse
|
58 |
| -from .api_deployment_visibility import APIDeploymentVisibility as APIDeploymentVisibility |
| 72 | +from .agents.evaluation_metrics import workspaces # type: ignore # noqa: F401 |
| 73 | +from .api_deployment_visibility import ( |
| 74 | + APIDeploymentVisibility as APIDeploymentVisibility, |
| 75 | +) |
59 | 76 | from .gpu_droplet_create_params import GPUDropletCreateParams as GPUDropletCreateParams
|
60 | 77 | from .gpu_droplet_list_response import GPUDropletListResponse as GPUDropletListResponse
|
61 |
| -from .agent_update_status_params import AgentUpdateStatusParams as AgentUpdateStatusParams |
| 78 | +from .agent_update_status_params import ( |
| 79 | + AgentUpdateStatusParams as AgentUpdateStatusParams, |
| 80 | +) |
62 | 81 | from .api_anthropic_api_key_info import APIAnthropicAPIKeyInfo as APIAnthropicAPIKeyInfo
|
63 |
| -from .knowledge_base_list_params import KnowledgeBaseListParams as KnowledgeBaseListParams |
64 |
| -from .droplet_backup_policy_param import DropletBackupPolicyParam as DropletBackupPolicyParam |
65 |
| -from .gpu_droplet_create_response import GPUDropletCreateResponse as GPUDropletCreateResponse |
66 |
| -from .agent_update_status_response import AgentUpdateStatusResponse as AgentUpdateStatusResponse |
67 |
| -from .knowledge_base_create_params import KnowledgeBaseCreateParams as KnowledgeBaseCreateParams |
68 |
| -from .knowledge_base_list_response import KnowledgeBaseListResponse as KnowledgeBaseListResponse |
69 |
| -from .knowledge_base_update_params import KnowledgeBaseUpdateParams as KnowledgeBaseUpdateParams |
70 |
| -from .gpu_droplet_retrieve_response import GPUDropletRetrieveResponse as GPUDropletRetrieveResponse |
71 |
| -from .knowledge_base_create_response import KnowledgeBaseCreateResponse as KnowledgeBaseCreateResponse |
72 |
| -from .knowledge_base_delete_response import KnowledgeBaseDeleteResponse as KnowledgeBaseDeleteResponse |
73 |
| -from .knowledge_base_update_response import KnowledgeBaseUpdateResponse as KnowledgeBaseUpdateResponse |
74 |
| -from .gpu_droplet_list_kernels_params import GPUDropletListKernelsParams as GPUDropletListKernelsParams |
75 |
| -from .gpu_droplet_delete_by_tag_params import GPUDropletDeleteByTagParams as GPUDropletDeleteByTagParams |
76 |
| -from .knowledge_base_retrieve_response import KnowledgeBaseRetrieveResponse as KnowledgeBaseRetrieveResponse |
77 |
| -from .gpu_droplet_list_firewalls_params import GPUDropletListFirewallsParams as GPUDropletListFirewallsParams |
78 |
| -from .gpu_droplet_list_kernels_response import GPUDropletListKernelsResponse as GPUDropletListKernelsResponse |
79 |
| -from .gpu_droplet_list_snapshots_params import GPUDropletListSnapshotsParams as GPUDropletListSnapshotsParams |
80 |
| -from .gpu_droplet_list_firewalls_response import GPUDropletListFirewallsResponse as GPUDropletListFirewallsResponse |
81 |
| -from .gpu_droplet_list_neighbors_response import GPUDropletListNeighborsResponse as GPUDropletListNeighborsResponse |
82 |
| -from .gpu_droplet_list_snapshots_response import GPUDropletListSnapshotsResponse as GPUDropletListSnapshotsResponse |
| 82 | +from .knowledge_base_list_params import ( |
| 83 | + KnowledgeBaseListParams as KnowledgeBaseListParams, |
| 84 | +) |
| 85 | +from .droplet_backup_policy_param import ( |
| 86 | + DropletBackupPolicyParam as DropletBackupPolicyParam, |
| 87 | +) |
| 88 | +from .gpu_droplet_create_response import ( |
| 89 | + GPUDropletCreateResponse as GPUDropletCreateResponse, |
| 90 | +) |
| 91 | +from .agent_update_status_response import ( |
| 92 | + AgentUpdateStatusResponse as AgentUpdateStatusResponse, |
| 93 | +) |
| 94 | +from .knowledge_base_create_params import ( |
| 95 | + KnowledgeBaseCreateParams as KnowledgeBaseCreateParams, |
| 96 | +) |
| 97 | +from .knowledge_base_list_response import ( |
| 98 | + KnowledgeBaseListResponse as KnowledgeBaseListResponse, |
| 99 | +) |
| 100 | +from .knowledge_base_update_params import ( |
| 101 | + KnowledgeBaseUpdateParams as KnowledgeBaseUpdateParams, |
| 102 | +) |
| 103 | +from .gpu_droplet_retrieve_response import ( |
| 104 | + GPUDropletRetrieveResponse as GPUDropletRetrieveResponse, |
| 105 | +) |
| 106 | +from .knowledge_base_create_response import ( |
| 107 | + KnowledgeBaseCreateResponse as KnowledgeBaseCreateResponse, |
| 108 | +) |
| 109 | +from .knowledge_base_delete_response import ( |
| 110 | + KnowledgeBaseDeleteResponse as KnowledgeBaseDeleteResponse, |
| 111 | +) |
| 112 | +from .knowledge_base_update_response import ( |
| 113 | + KnowledgeBaseUpdateResponse as KnowledgeBaseUpdateResponse, |
| 114 | +) |
| 115 | +from .gpu_droplet_list_kernels_params import ( |
| 116 | + GPUDropletListKernelsParams as GPUDropletListKernelsParams, |
| 117 | +) |
| 118 | +from .gpu_droplet_delete_by_tag_params import ( |
| 119 | + GPUDropletDeleteByTagParams as GPUDropletDeleteByTagParams, |
| 120 | +) |
| 121 | +from .knowledge_base_retrieve_response import ( |
| 122 | + KnowledgeBaseRetrieveResponse as KnowledgeBaseRetrieveResponse, |
| 123 | +) |
| 124 | +from .gpu_droplet_list_firewalls_params import ( |
| 125 | + GPUDropletListFirewallsParams as GPUDropletListFirewallsParams, |
| 126 | +) |
| 127 | +from .gpu_droplet_list_kernels_response import ( |
| 128 | + GPUDropletListKernelsResponse as GPUDropletListKernelsResponse, |
| 129 | +) |
| 130 | +from .gpu_droplet_list_snapshots_params import ( |
| 131 | + GPUDropletListSnapshotsParams as GPUDropletListSnapshotsParams, |
| 132 | +) |
| 133 | +from .gpu_droplet_list_firewalls_response import ( |
| 134 | + GPUDropletListFirewallsResponse as GPUDropletListFirewallsResponse, |
| 135 | +) |
| 136 | +from .gpu_droplet_list_neighbors_response import ( |
| 137 | + GPUDropletListNeighborsResponse as GPUDropletListNeighborsResponse, |
| 138 | +) |
| 139 | +from .gpu_droplet_list_snapshots_response import ( |
| 140 | + GPUDropletListSnapshotsResponse as GPUDropletListSnapshotsResponse, |
| 141 | +) |
| 142 | +from .agents.evaluation_metrics.workspaces import ( |
| 143 | + agent_list_response, # type: ignore # noqa: F401 |
| 144 | + agent_move_response, # type: ignore # noqa: F401 |
| 145 | +) |
| 146 | + |
| 147 | +# Rebuild cyclical models only after all modules are imported. |
| 148 | +# This ensures that, when building the deferred (due to cyclical references) model schema, |
| 149 | +# Pydantic can resolve the necessary references. |
| 150 | +# See: https://github.com/pydantic/pydantic/issues/11250 for more context. |
| 151 | +if _compat.PYDANTIC_V2: |
| 152 | + api_agent.APIAgent.model_rebuild(_parent_namespace_depth=0) |
| 153 | + api_workspace.APIWorkspace.model_rebuild(_parent_namespace_depth=0) |
| 154 | + agent_create_response.AgentCreateResponse.model_rebuild(_parent_namespace_depth=0) |
| 155 | + agent_retrieve_response.AgentRetrieveResponse.model_rebuild( |
| 156 | + _parent_namespace_depth=0 |
| 157 | + ) |
| 158 | + agent_update_response.AgentUpdateResponse.model_rebuild(_parent_namespace_depth=0) |
| 159 | + agent_delete_response.AgentDeleteResponse.model_rebuild(_parent_namespace_depth=0) |
| 160 | + agent_update_status_response.AgentUpdateStatusResponse.model_rebuild( |
| 161 | + _parent_namespace_depth=0 |
| 162 | + ) |
| 163 | + agents.evaluation_metrics.workspace_create_response.WorkspaceCreateResponse.model_rebuild( |
| 164 | + _parent_namespace_depth=0 |
| 165 | + ) |
| 166 | + agents.evaluation_metrics.workspace_retrieve_response.WorkspaceRetrieveResponse.model_rebuild( |
| 167 | + _parent_namespace_depth=0 |
| 168 | + ) |
| 169 | + agents.evaluation_metrics.workspace_update_response.WorkspaceUpdateResponse.model_rebuild( |
| 170 | + _parent_namespace_depth=0 |
| 171 | + ) |
| 172 | + agents.evaluation_metrics.workspace_list_response.WorkspaceListResponse.model_rebuild( |
| 173 | + _parent_namespace_depth=0 |
| 174 | + ) |
| 175 | + agents.evaluation_metrics.workspaces.agent_list_response.AgentListResponse.model_rebuild( |
| 176 | + _parent_namespace_depth=0 |
| 177 | + ) |
| 178 | + agents.evaluation_metrics.workspaces.agent_move_response.AgentMoveResponse.model_rebuild( |
| 179 | + _parent_namespace_depth=0 |
| 180 | + ) |
| 181 | + agents.function_create_response.FunctionCreateResponse.model_rebuild( |
| 182 | + _parent_namespace_depth=0 |
| 183 | + ) |
| 184 | + agents.function_update_response.FunctionUpdateResponse.model_rebuild( |
| 185 | + _parent_namespace_depth=0 |
| 186 | + ) |
| 187 | + agents.function_delete_response.FunctionDeleteResponse.model_rebuild( |
| 188 | + _parent_namespace_depth=0 |
| 189 | + ) |
| 190 | + agents.api_link_knowledge_base_output.APILinkKnowledgeBaseOutput.model_rebuild( |
| 191 | + _parent_namespace_depth=0 |
| 192 | + ) |
| 193 | + agents.knowledge_base_detach_response.KnowledgeBaseDetachResponse.model_rebuild( |
| 194 | + _parent_namespace_depth=0 |
| 195 | + ) |
| 196 | + agents.route_view_response.RouteViewResponse.model_rebuild( |
| 197 | + _parent_namespace_depth=0 |
| 198 | + ) |
| 199 | + models.providers.anthropic_list_agents_response.AnthropicListAgentsResponse.model_rebuild( |
| 200 | + _parent_namespace_depth=0 |
| 201 | + ) |
| 202 | + models.providers.openai_retrieve_agents_response.OpenAIRetrieveAgentsResponse.model_rebuild( |
| 203 | + _parent_namespace_depth=0 |
| 204 | + ) |
| 205 | +else: |
| 206 | + api_agent.APIAgent.update_forward_refs() # type: ignore |
| 207 | + api_workspace.APIWorkspace.update_forward_refs() # type: ignore |
| 208 | + agent_create_response.AgentCreateResponse.update_forward_refs() # type: ignore |
| 209 | + agent_retrieve_response.AgentRetrieveResponse.update_forward_refs() # type: ignore |
| 210 | + agent_update_response.AgentUpdateResponse.update_forward_refs() # type: ignore |
| 211 | + agent_delete_response.AgentDeleteResponse.update_forward_refs() # type: ignore |
| 212 | + agent_update_status_response.AgentUpdateStatusResponse.update_forward_refs() # type: ignore |
| 213 | + agents.evaluation_metrics.workspace_create_response.WorkspaceCreateResponse.update_forward_refs() # type: ignore |
| 214 | + agents.evaluation_metrics.workspace_retrieve_response.WorkspaceRetrieveResponse.update_forward_refs() # type: ignore |
| 215 | + agents.evaluation_metrics.workspace_update_response.WorkspaceUpdateResponse.update_forward_refs() # type: ignore |
| 216 | + agents.evaluation_metrics.workspace_list_response.WorkspaceListResponse.update_forward_refs() # type: ignore |
| 217 | + agents.evaluation_metrics.workspaces.agent_list_response.AgentListResponse.update_forward_refs() # type: ignore |
| 218 | + agents.evaluation_metrics.workspaces.agent_move_response.AgentMoveResponse.update_forward_refs() # type: ignore |
| 219 | + agents.function_create_response.FunctionCreateResponse.update_forward_refs() # type: ignore |
| 220 | + agents.function_update_response.FunctionUpdateResponse.update_forward_refs() # type: ignore |
| 221 | + agents.function_delete_response.FunctionDeleteResponse.update_forward_refs() # type: ignore |
| 222 | + agents.api_link_knowledge_base_output.APILinkKnowledgeBaseOutput.update_forward_refs() # type: ignore |
| 223 | + agents.knowledge_base_detach_response.KnowledgeBaseDetachResponse.update_forward_refs() # type: ignore |
| 224 | + agents.route_view_response.RouteViewResponse.update_forward_refs() # type: ignore |
| 225 | + models.providers.anthropic_list_agents_response.AnthropicListAgentsResponse.update_forward_refs() # type: ignore |
| 226 | + models.providers.openai_retrieve_agents_response.OpenAIRetrieveAgentsResponse.update_forward_refs() # type: ignore |
0 commit comments