windmill-api 1.543.0__py3-none-any.whl → 1.544.1__py3-none-any.whl

windmill_api/models/dynamic_input_data_runnable_ref_type_0.py

@@ -0,0 +1,76 @@
+from typing import Any, Dict, List, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.dynamic_input_data_runnable_ref_type_0_runnable_kind import DynamicInputDataRunnableRefType0RunnableKind
+from ..models.dynamic_input_data_runnable_ref_type_0_source import DynamicInputDataRunnableRefType0Source
+
+T = TypeVar("T", bound="DynamicInputDataRunnableRefType0")
+
+
+@_attrs_define
+class DynamicInputDataRunnableRefType0:
+    """
+    Attributes:
+        source (DynamicInputDataRunnableRefType0Source):
+        path (str): Path to the deployed script or flow
+        runnable_kind (DynamicInputDataRunnableRefType0RunnableKind):
+    """
+
+    source: DynamicInputDataRunnableRefType0Source
+    path: str
+    runnable_kind: DynamicInputDataRunnableRefType0RunnableKind
+    additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        source = self.source.value
+
+        path = self.path
+        runnable_kind = self.runnable_kind.value
+
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "source": source,
+                "path": path,
+                "runnable_kind": runnable_kind,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        source = DynamicInputDataRunnableRefType0Source(d.pop("source"))
+
+        path = d.pop("path")
+
+        runnable_kind = DynamicInputDataRunnableRefType0RunnableKind(d.pop("runnable_kind"))
+
+        dynamic_input_data_runnable_ref_type_0 = cls(
+            source=source,
+            path=path,
+            runnable_kind=runnable_kind,
+        )
+
+        dynamic_input_data_runnable_ref_type_0.additional_properties = d
+        return dynamic_input_data_runnable_ref_type_0
+
+    @property
+    def additional_keys(self) -> List[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

windmill_api/models/dynamic_input_data_runnable_ref_type_0_runnable_kind.py

@@ -0,0 +1,9 @@
+from enum import Enum
+
+
+class DynamicInputDataRunnableRefType0RunnableKind(str, Enum):
+    FLOW = "flow"
+    SCRIPT = "script"
+
+    def __str__(self) -> str:
+        return str(self.value)

windmill_api/models/dynamic_input_data_runnable_ref_type_0_source.py

@@ -0,0 +1,8 @@
+from enum import Enum
+
+
+class DynamicInputDataRunnableRefType0Source(str, Enum):
+    DEPLOYED = "deployed"
+
+    def __str__(self) -> str:
+        return str(self.value)

windmill_api/models/dynamic_input_data_runnable_ref_type_1.py

@@ -0,0 +1,85 @@
+from typing import Any, Dict, List, Type, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.dynamic_input_data_runnable_ref_type_1_language import DynamicInputDataRunnableRefType1Language
+from ..models.dynamic_input_data_runnable_ref_type_1_source import DynamicInputDataRunnableRefType1Source
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="DynamicInputDataRunnableRefType1")
+
+
+@_attrs_define
+class DynamicInputDataRunnableRefType1:
+    """
+    Attributes:
+        source (DynamicInputDataRunnableRefType1Source):
+        code (str): Code content for inline execution
+        language (Union[Unset, DynamicInputDataRunnableRefType1Language]):
+    """
+
+    source: DynamicInputDataRunnableRefType1Source
+    code: str
+    language: Union[Unset, DynamicInputDataRunnableRefType1Language] = UNSET
+    additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        source = self.source.value
+
+        code = self.code
+        language: Union[Unset, str] = UNSET
+        if not isinstance(self.language, Unset):
+            language = self.language.value
+
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "source": source,
+                "code": code,
+            }
+        )
+        if language is not UNSET:
+            field_dict["language"] = language
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        source = DynamicInputDataRunnableRefType1Source(d.pop("source"))
+
+        code = d.pop("code")
+
+        _language = d.pop("language", UNSET)
+        language: Union[Unset, DynamicInputDataRunnableRefType1Language]
+        if isinstance(_language, Unset):
+            language = UNSET
+        else:
+            language = DynamicInputDataRunnableRefType1Language(_language)
+
+        dynamic_input_data_runnable_ref_type_1 = cls(
+            source=source,
+            code=code,
+            language=language,
+        )
+
+        dynamic_input_data_runnable_ref_type_1.additional_properties = d
+        return dynamic_input_data_runnable_ref_type_1
+
+    @property
+    def additional_keys(self) -> List[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

windmill_api/models/dynamic_input_data_runnable_ref_type_1_language.py

@@ -0,0 +1,29 @@
+from enum import Enum
+
+
+class DynamicInputDataRunnableRefType1Language(str, Enum):
+    ANSIBLE = "ansible"
+    BASH = "bash"
+    BIGQUERY = "bigquery"
+    BUN = "bun"
+    CSHARP = "csharp"
+    DENO = "deno"
+    DUCKDB = "duckdb"
+    GO = "go"
+    GRAPHQL = "graphql"
+    JAVA = "java"
+    MSSQL = "mssql"
+    MYSQL = "mysql"
+    NATIVETS = "nativets"
+    NU = "nu"
+    ORACLEDB = "oracledb"
+    PHP = "php"
+    POSTGRESQL = "postgresql"
+    POWERSHELL = "powershell"
+    PYTHON3 = "python3"
+    RUBY = "ruby"
+    RUST = "rust"
+    SNOWFLAKE = "snowflake"
+
+    def __str__(self) -> str:
+        return str(self.value)

windmill_api/models/dynamic_input_data_runnable_ref_type_1_source.py

@@ -0,0 +1,8 @@
+from enum import Enum
+
+
+class DynamicInputDataRunnableRefType1Source(str, Enum):
+    INLINE = "inline"
+
+    def __str__(self) -> str:
+        return str(self.value)

windmill_api/models/edit_copilot_config_json_body.py

@@ -11,6 +11,7 @@ if TYPE_CHECKING:
     )
     from ..models.edit_copilot_config_json_body_custom_prompts import EditCopilotConfigJsonBodyCustomPrompts
     from ..models.edit_copilot_config_json_body_default_model import EditCopilotConfigJsonBodyDefaultModel
+    from ..models.edit_copilot_config_json_body_max_tokens_per_model import EditCopilotConfigJsonBodyMaxTokensPerModel
     from ..models.edit_copilot_config_json_body_providers import EditCopilotConfigJsonBodyProviders
 
 
@@ -25,12 +26,14 @@ class EditCopilotConfigJsonBody:
         default_model (Union[Unset, EditCopilotConfigJsonBodyDefaultModel]):
         code_completion_model (Union[Unset, EditCopilotConfigJsonBodyCodeCompletionModel]):
         custom_prompts (Union[Unset, EditCopilotConfigJsonBodyCustomPrompts]):
+        max_tokens_per_model (Union[Unset, EditCopilotConfigJsonBodyMaxTokensPerModel]):
     """
 
     providers: Union[Unset, "EditCopilotConfigJsonBodyProviders"] = UNSET
     default_model: Union[Unset, "EditCopilotConfigJsonBodyDefaultModel"] = UNSET
     code_completion_model: Union[Unset, "EditCopilotConfigJsonBodyCodeCompletionModel"] = UNSET
     custom_prompts: Union[Unset, "EditCopilotConfigJsonBodyCustomPrompts"] = UNSET
+    max_tokens_per_model: Union[Unset, "EditCopilotConfigJsonBodyMaxTokensPerModel"] = UNSET
     additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> Dict[str, Any]:
@@ -50,6 +53,10 @@ class EditCopilotConfigJsonBody:
         if not isinstance(self.custom_prompts, Unset):
             custom_prompts = self.custom_prompts.to_dict()
 
+        max_tokens_per_model: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.max_tokens_per_model, Unset):
+            max_tokens_per_model = self.max_tokens_per_model.to_dict()
+
         field_dict: Dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
@@ -61,6 +68,8 @@ class EditCopilotConfigJsonBody:
             field_dict["code_completion_model"] = code_completion_model
         if custom_prompts is not UNSET:
             field_dict["custom_prompts"] = custom_prompts
+        if max_tokens_per_model is not UNSET:
+            field_dict["max_tokens_per_model"] = max_tokens_per_model
 
         return field_dict
 
@@ -71,6 +80,9 @@ class EditCopilotConfigJsonBody:
         )
         from ..models.edit_copilot_config_json_body_custom_prompts import EditCopilotConfigJsonBodyCustomPrompts
         from ..models.edit_copilot_config_json_body_default_model import EditCopilotConfigJsonBodyDefaultModel
+        from ..models.edit_copilot_config_json_body_max_tokens_per_model import (
+            EditCopilotConfigJsonBodyMaxTokensPerModel,
+        )
         from ..models.edit_copilot_config_json_body_providers import EditCopilotConfigJsonBodyProviders
 
         d = src_dict.copy()
@@ -102,11 +114,19 @@ class EditCopilotConfigJsonBody:
         else:
             custom_prompts = EditCopilotConfigJsonBodyCustomPrompts.from_dict(_custom_prompts)
 
+        _max_tokens_per_model = d.pop("max_tokens_per_model", UNSET)
+        max_tokens_per_model: Union[Unset, EditCopilotConfigJsonBodyMaxTokensPerModel]
+        if isinstance(_max_tokens_per_model, Unset):
+            max_tokens_per_model = UNSET
+        else:
+            max_tokens_per_model = EditCopilotConfigJsonBodyMaxTokensPerModel.from_dict(_max_tokens_per_model)
+
         edit_copilot_config_json_body = cls(
             providers=providers,
             default_model=default_model,
             code_completion_model=code_completion_model,
             custom_prompts=custom_prompts,
+            max_tokens_per_model=max_tokens_per_model,
         )
 
         edit_copilot_config_json_body.additional_properties = d

windmill_api/models/edit_copilot_config_json_body_max_tokens_per_model.py

@@ -0,0 +1,44 @@
+from typing import Any, Dict, List, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="EditCopilotConfigJsonBodyMaxTokensPerModel")
+
+
+@_attrs_define
+class EditCopilotConfigJsonBodyMaxTokensPerModel:
+    """ """
+
+    additional_properties: Dict[str, int] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        edit_copilot_config_json_body_max_tokens_per_model = cls()
+
+        edit_copilot_config_json_body_max_tokens_per_model.additional_properties = d
+        return edit_copilot_config_json_body_max_tokens_per_model
+
+    @property
+    def additional_keys(self) -> List[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> int:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: int) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

windmill_api/models/edit_large_file_storage_config_json_body_large_file_storage.py

@@ -9,6 +9,9 @@ from ..models.edit_large_file_storage_config_json_body_large_file_storage_type i
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
+    from ..models.edit_large_file_storage_config_json_body_large_file_storage_advanced_permissions_item import (
+        EditLargeFileStorageConfigJsonBodyLargeFileStorageAdvancedPermissionsItem,
+    )
     from ..models.edit_large_file_storage_config_json_body_large_file_storage_secondary_storage import (
         EditLargeFileStorageConfigJsonBodyLargeFileStorageSecondaryStorage,
     )
@@ -26,6 +29,8 @@ class EditLargeFileStorageConfigJsonBodyLargeFileStorage:
         azure_blob_resource_path (Union[Unset, str]):
         gcs_resource_path (Union[Unset, str]):
         public_resource (Union[Unset, bool]):
+        advanced_permissions (Union[Unset,
+            List['EditLargeFileStorageConfigJsonBodyLargeFileStorageAdvancedPermissionsItem']]):
         secondary_storage (Union[Unset, EditLargeFileStorageConfigJsonBodyLargeFileStorageSecondaryStorage]):
     """
 
@@ -34,6 +39,9 @@ class EditLargeFileStorageConfigJsonBodyLargeFileStorage:
     azure_blob_resource_path: Union[Unset, str] = UNSET
     gcs_resource_path: Union[Unset, str] = UNSET
     public_resource: Union[Unset, bool] = UNSET
+    advanced_permissions: Union[
+        Unset, List["EditLargeFileStorageConfigJsonBodyLargeFileStorageAdvancedPermissionsItem"]
+    ] = UNSET
     secondary_storage: Union[Unset, "EditLargeFileStorageConfigJsonBodyLargeFileStorageSecondaryStorage"] = UNSET
     additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
 
@@ -46,6 +54,14 @@ class EditLargeFileStorageConfigJsonBodyLargeFileStorage:
         azure_blob_resource_path = self.azure_blob_resource_path
         gcs_resource_path = self.gcs_resource_path
         public_resource = self.public_resource
+        advanced_permissions: Union[Unset, List[Dict[str, Any]]] = UNSET
+        if not isinstance(self.advanced_permissions, Unset):
+            advanced_permissions = []
+            for advanced_permissions_item_data in self.advanced_permissions:
+                advanced_permissions_item = advanced_permissions_item_data.to_dict()
+
+                advanced_permissions.append(advanced_permissions_item)
+
         secondary_storage: Union[Unset, Dict[str, Any]] = UNSET
         if not isinstance(self.secondary_storage, Unset):
             secondary_storage = self.secondary_storage.to_dict()
@@ -63,6 +79,8 @@ class EditLargeFileStorageConfigJsonBodyLargeFileStorage:
             field_dict["gcs_resource_path"] = gcs_resource_path
         if public_resource is not UNSET:
             field_dict["public_resource"] = public_resource
+        if advanced_permissions is not UNSET:
+            field_dict["advanced_permissions"] = advanced_permissions
         if secondary_storage is not UNSET:
             field_dict["secondary_storage"] = secondary_storage
 
@@ -70,6 +88,9 @@ class EditLargeFileStorageConfigJsonBodyLargeFileStorage:
 
     @classmethod
     def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        from ..models.edit_large_file_storage_config_json_body_large_file_storage_advanced_permissions_item import (
+            EditLargeFileStorageConfigJsonBodyLargeFileStorageAdvancedPermissionsItem,
+        )
         from ..models.edit_large_file_storage_config_json_body_large_file_storage_secondary_storage import (
             EditLargeFileStorageConfigJsonBodyLargeFileStorageSecondaryStorage,
         )
@@ -90,6 +111,17 @@ class EditLargeFileStorageConfigJsonBodyLargeFileStorage:
 
         public_resource = d.pop("public_resource", UNSET)
 
+        advanced_permissions = []
+        _advanced_permissions = d.pop("advanced_permissions", UNSET)
+        for advanced_permissions_item_data in _advanced_permissions or []:
+            advanced_permissions_item = (
+                EditLargeFileStorageConfigJsonBodyLargeFileStorageAdvancedPermissionsItem.from_dict(
+                    advanced_permissions_item_data
+                )
+            )
+
+            advanced_permissions.append(advanced_permissions_item)
+
         _secondary_storage = d.pop("secondary_storage", UNSET)
         secondary_storage: Union[Unset, EditLargeFileStorageConfigJsonBodyLargeFileStorageSecondaryStorage]
         if isinstance(_secondary_storage, Unset):
@@ -105,6 +137,7 @@ class EditLargeFileStorageConfigJsonBodyLargeFileStorage:
             azure_blob_resource_path=azure_blob_resource_path,
             gcs_resource_path=gcs_resource_path,
             public_resource=public_resource,
+            advanced_permissions=advanced_permissions,
             secondary_storage=secondary_storage,
         )
 

windmill_api/models/edit_large_file_storage_config_json_body_large_file_storage_advanced_permissions_item.py

@@ -0,0 +1,65 @@
+from typing import Any, Dict, List, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="EditLargeFileStorageConfigJsonBodyLargeFileStorageAdvancedPermissionsItem")
+
+
+@_attrs_define
+class EditLargeFileStorageConfigJsonBodyLargeFileStorageAdvancedPermissionsItem:
+    """
+    Attributes:
+        pattern (str):
+        allow (str):
+    """
+
+    pattern: str
+    allow: str
+    additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        pattern = self.pattern
+        allow = self.allow
+
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "pattern": pattern,
+                "allow": allow,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        pattern = d.pop("pattern")
+
+        allow = d.pop("allow")
+
+        edit_large_file_storage_config_json_body_large_file_storage_advanced_permissions_item = cls(
+            pattern=pattern,
+            allow=allow,
+        )
+
+        edit_large_file_storage_config_json_body_large_file_storage_advanced_permissions_item.additional_properties = d
+        return edit_large_file_storage_config_json_body_large_file_storage_advanced_permissions_item
+
+    @property
+    def additional_keys(self) -> List[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

windmill_api/models/gcp_trigger_data.py

@@ -33,6 +33,8 @@ class GcpTriggerData:
         delivery_config (Union[Unset, GcpTriggerDataDeliveryConfig]):
         enabled (Union[Unset, bool]):
         auto_acknowledge_msg (Union[Unset, bool]):
+        ack_deadline (Union[Unset, int]): Time in seconds within which the message must be acknowledged. If not
+            provided, defaults to the subscription's acknowledgment deadline (600 seconds).
         error_handler_path (Union[Unset, str]):
         error_handler_args (Union[Unset, GcpTriggerDataErrorHandlerArgs]): The arguments to pass to the script or flow
         retry (Union[Unset, GcpTriggerDataRetry]):
@@ -50,6 +52,7 @@ class GcpTriggerData:
     delivery_config: Union[Unset, "GcpTriggerDataDeliveryConfig"] = UNSET
     enabled: Union[Unset, bool] = UNSET
     auto_acknowledge_msg: Union[Unset, bool] = UNSET
+    ack_deadline: Union[Unset, int] = UNSET
     error_handler_path: Union[Unset, str] = UNSET
     error_handler_args: Union[Unset, "GcpTriggerDataErrorHandlerArgs"] = UNSET
     retry: Union[Unset, "GcpTriggerDataRetry"] = UNSET
@@ -75,6 +78,7 @@ class GcpTriggerData:
 
         enabled = self.enabled
         auto_acknowledge_msg = self.auto_acknowledge_msg
+        ack_deadline = self.ack_deadline
         error_handler_path = self.error_handler_path
         error_handler_args: Union[Unset, Dict[str, Any]] = UNSET
         if not isinstance(self.error_handler_args, Unset):
@@ -108,6 +112,8 @@ class GcpTriggerData:
             field_dict["enabled"] = enabled
         if auto_acknowledge_msg is not UNSET:
             field_dict["auto_acknowledge_msg"] = auto_acknowledge_msg
+        if ack_deadline is not UNSET:
+            field_dict["ack_deadline"] = ack_deadline
         if error_handler_path is not UNSET:
             field_dict["error_handler_path"] = error_handler_path
         if error_handler_args is not UNSET:
@@ -158,6 +164,8 @@ class GcpTriggerData:
 
         auto_acknowledge_msg = d.pop("auto_acknowledge_msg", UNSET)
 
+        ack_deadline = d.pop("ack_deadline", UNSET)
+
         error_handler_path = d.pop("error_handler_path", UNSET)
 
         _error_handler_args = d.pop("error_handler_args", UNSET)
@@ -187,6 +195,7 @@ class GcpTriggerData:
             delivery_config=delivery_config,
             enabled=enabled,
             auto_acknowledge_msg=auto_acknowledge_msg,
+            ack_deadline=ack_deadline,
             error_handler_path=error_handler_path,
             error_handler_args=error_handler_args,
             retry=retry,

windmill_api/models/get_copilot_info_response_200.py

@@ -11,6 +11,7 @@ if TYPE_CHECKING:
     )
     from ..models.get_copilot_info_response_200_custom_prompts import GetCopilotInfoResponse200CustomPrompts
     from ..models.get_copilot_info_response_200_default_model import GetCopilotInfoResponse200DefaultModel
+    from ..models.get_copilot_info_response_200_max_tokens_per_model import GetCopilotInfoResponse200MaxTokensPerModel
     from ..models.get_copilot_info_response_200_providers import GetCopilotInfoResponse200Providers
 
 
@@ -25,12 +26,14 @@ class GetCopilotInfoResponse200:
         default_model (Union[Unset, GetCopilotInfoResponse200DefaultModel]):
         code_completion_model (Union[Unset, GetCopilotInfoResponse200CodeCompletionModel]):
         custom_prompts (Union[Unset, GetCopilotInfoResponse200CustomPrompts]):
+        max_tokens_per_model (Union[Unset, GetCopilotInfoResponse200MaxTokensPerModel]):
     """
 
     providers: Union[Unset, "GetCopilotInfoResponse200Providers"] = UNSET
     default_model: Union[Unset, "GetCopilotInfoResponse200DefaultModel"] = UNSET
     code_completion_model: Union[Unset, "GetCopilotInfoResponse200CodeCompletionModel"] = UNSET
     custom_prompts: Union[Unset, "GetCopilotInfoResponse200CustomPrompts"] = UNSET
+    max_tokens_per_model: Union[Unset, "GetCopilotInfoResponse200MaxTokensPerModel"] = UNSET
     additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> Dict[str, Any]:
@@ -50,6 +53,10 @@ class GetCopilotInfoResponse200:
         if not isinstance(self.custom_prompts, Unset):
             custom_prompts = self.custom_prompts.to_dict()
 
+        max_tokens_per_model: Union[Unset, Dict[str, Any]] = UNSET
+        if not isinstance(self.max_tokens_per_model, Unset):
+            max_tokens_per_model = self.max_tokens_per_model.to_dict()
+
         field_dict: Dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
@@ -61,6 +68,8 @@ class GetCopilotInfoResponse200:
             field_dict["code_completion_model"] = code_completion_model
         if custom_prompts is not UNSET:
             field_dict["custom_prompts"] = custom_prompts
+        if max_tokens_per_model is not UNSET:
+            field_dict["max_tokens_per_model"] = max_tokens_per_model
 
         return field_dict
 
@@ -71,6 +80,9 @@ class GetCopilotInfoResponse200:
         )
         from ..models.get_copilot_info_response_200_custom_prompts import GetCopilotInfoResponse200CustomPrompts
         from ..models.get_copilot_info_response_200_default_model import GetCopilotInfoResponse200DefaultModel
+        from ..models.get_copilot_info_response_200_max_tokens_per_model import (
+            GetCopilotInfoResponse200MaxTokensPerModel,
+        )
         from ..models.get_copilot_info_response_200_providers import GetCopilotInfoResponse200Providers
 
         d = src_dict.copy()
@@ -102,11 +114,19 @@ class GetCopilotInfoResponse200:
         else:
             custom_prompts = GetCopilotInfoResponse200CustomPrompts.from_dict(_custom_prompts)
 
+        _max_tokens_per_model = d.pop("max_tokens_per_model", UNSET)
+        max_tokens_per_model: Union[Unset, GetCopilotInfoResponse200MaxTokensPerModel]
+        if isinstance(_max_tokens_per_model, Unset):
+            max_tokens_per_model = UNSET
+        else:
+            max_tokens_per_model = GetCopilotInfoResponse200MaxTokensPerModel.from_dict(_max_tokens_per_model)
+
         get_copilot_info_response_200 = cls(
             providers=providers,
             default_model=default_model,
             code_completion_model=code_completion_model,
             custom_prompts=custom_prompts,
+            max_tokens_per_model=max_tokens_per_model,
         )
 
         get_copilot_info_response_200.additional_properties = d

windmill_api/models/get_copilot_info_response_200_max_tokens_per_model.py

@@ -0,0 +1,44 @@
+from typing import Any, Dict, List, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="GetCopilotInfoResponse200MaxTokensPerModel")
+
+
+@_attrs_define
+class GetCopilotInfoResponse200MaxTokensPerModel:
+    """ """
+
+    additional_properties: Dict[str, int] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        get_copilot_info_response_200_max_tokens_per_model = cls()
+
+        get_copilot_info_response_200_max_tokens_per_model.additional_properties = d
+        return get_copilot_info_response_200_max_tokens_per_model
+
+    @property
+    def additional_keys(self) -> List[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> int:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: int) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties