langwatch-scenario 0.6.0__py3-none-any.whl → 0.7.1__py3-none-any.whl

scenario/_generated/langwatch_api_client/lang_watch_api_client/models/post_api_prompts_response_200.py

@@ -0,0 +1,172 @@
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+if TYPE_CHECKING:
+    from ..models.post_api_prompts_response_200_messages_item import PostApiPromptsResponse200MessagesItem
+    from ..models.post_api_prompts_response_200_response_format_type_0 import (
+        PostApiPromptsResponse200ResponseFormatType0,
+    )
+
+
+T = TypeVar("T", bound="PostApiPromptsResponse200")
+
+
+@_attrs_define
+class PostApiPromptsResponse200:
+    """
+    Attributes:
+        id (str):
+        name (str):
+        updated_at (str):
+        version (float):
+        version_id (str):
+        version_created_at (str):
+        model (str):
+        prompt (str):
+        messages (list['PostApiPromptsResponse200MessagesItem']):
+        response_format (Union['PostApiPromptsResponse200ResponseFormatType0', None]):
+    """
+
+    id: str
+    name: str
+    updated_at: str
+    version: float
+    version_id: str
+    version_created_at: str
+    model: str
+    prompt: str
+    messages: list["PostApiPromptsResponse200MessagesItem"]
+    response_format: Union["PostApiPromptsResponse200ResponseFormatType0", None]
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        from ..models.post_api_prompts_response_200_response_format_type_0 import (
+            PostApiPromptsResponse200ResponseFormatType0,
+        )
+
+        id = self.id
+
+        name = self.name
+
+        updated_at = self.updated_at
+
+        version = self.version
+
+        version_id = self.version_id
+
+        version_created_at = self.version_created_at
+
+        model = self.model
+
+        prompt = self.prompt
+
+        messages = []
+        for messages_item_data in self.messages:
+            messages_item = messages_item_data.to_dict()
+            messages.append(messages_item)
+
+        response_format: Union[None, dict[str, Any]]
+        if isinstance(self.response_format, PostApiPromptsResponse200ResponseFormatType0):
+            response_format = self.response_format.to_dict()
+        else:
+            response_format = self.response_format
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "id": id,
+                "name": name,
+                "updatedAt": updated_at,
+                "version": version,
+                "versionId": version_id,
+                "versionCreatedAt": version_created_at,
+                "model": model,
+                "prompt": prompt,
+                "messages": messages,
+                "response_format": response_format,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.post_api_prompts_response_200_messages_item import PostApiPromptsResponse200MessagesItem
+        from ..models.post_api_prompts_response_200_response_format_type_0 import (
+            PostApiPromptsResponse200ResponseFormatType0,
+        )
+
+        d = dict(src_dict)
+        id = d.pop("id")
+
+        name = d.pop("name")
+
+        updated_at = d.pop("updatedAt")
+
+        version = d.pop("version")
+
+        version_id = d.pop("versionId")
+
+        version_created_at = d.pop("versionCreatedAt")
+
+        model = d.pop("model")
+
+        prompt = d.pop("prompt")
+
+        messages = []
+        _messages = d.pop("messages")
+        for messages_item_data in _messages:
+            messages_item = PostApiPromptsResponse200MessagesItem.from_dict(messages_item_data)
+
+            messages.append(messages_item)
+
+        def _parse_response_format(data: object) -> Union["PostApiPromptsResponse200ResponseFormatType0", None]:
+            if data is None:
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                response_format_type_0 = PostApiPromptsResponse200ResponseFormatType0.from_dict(data)
+
+                return response_format_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union["PostApiPromptsResponse200ResponseFormatType0", None], data)
+
+        response_format = _parse_response_format(d.pop("response_format"))
+
+        post_api_prompts_response_200 = cls(
+            id=id,
+            name=name,
+            updated_at=updated_at,
+            version=version,
+            version_id=version_id,
+            version_created_at=version_created_at,
+            model=model,
+            prompt=prompt,
+            messages=messages,
+            response_format=response_format,
+        )
+
+        post_api_prompts_response_200.additional_properties = d
+        return post_api_prompts_response_200
+
+    @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

scenario/_generated/langwatch_api_client/lang_watch_api_client/models/post_api_prompts_response_200_messages_item.py

@@ -0,0 +1,69 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.post_api_prompts_response_200_messages_item_role import PostApiPromptsResponse200MessagesItemRole
+
+T = TypeVar("T", bound="PostApiPromptsResponse200MessagesItem")
+
+
+@_attrs_define
+class PostApiPromptsResponse200MessagesItem:
+    """
+    Attributes:
+        role (PostApiPromptsResponse200MessagesItemRole):
+        content (str):
+    """
+
+    role: PostApiPromptsResponse200MessagesItemRole
+    content: str
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        role = self.role.value
+
+        content = self.content
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "role": role,
+                "content": content,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        role = PostApiPromptsResponse200MessagesItemRole(d.pop("role"))
+
+        content = d.pop("content")
+
+        post_api_prompts_response_200_messages_item = cls(
+            role=role,
+            content=content,
+        )
+
+        post_api_prompts_response_200_messages_item.additional_properties = d
+        return post_api_prompts_response_200_messages_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

scenario/_generated/langwatch_api_client/lang_watch_api_client/models/post_api_prompts_response_200_messages_item_role.py

@@ -0,0 +1,10 @@
+from enum import Enum
+
+
+class PostApiPromptsResponse200MessagesItemRole(str, Enum):
+    ASSISTANT = "assistant"
+    SYSTEM = "system"
+    USER = "user"
+
+    def __str__(self) -> str:
+        return str(self.value)

scenario/_generated/langwatch_api_client/lang_watch_api_client/models/post_api_prompts_response_200_response_format_type_0.py

@@ -0,0 +1,81 @@
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.post_api_prompts_response_200_response_format_type_0_type import (
+    PostApiPromptsResponse200ResponseFormatType0Type,
+)
+
+if TYPE_CHECKING:
+    from ..models.post_api_prompts_response_200_response_format_type_0_json_schema import (
+        PostApiPromptsResponse200ResponseFormatType0JsonSchema,
+    )
+
+
+T = TypeVar("T", bound="PostApiPromptsResponse200ResponseFormatType0")
+
+
+@_attrs_define
+class PostApiPromptsResponse200ResponseFormatType0:
+    """
+    Attributes:
+        type_ (PostApiPromptsResponse200ResponseFormatType0Type):
+        json_schema (PostApiPromptsResponse200ResponseFormatType0JsonSchema):
+    """
+
+    type_: PostApiPromptsResponse200ResponseFormatType0Type
+    json_schema: "PostApiPromptsResponse200ResponseFormatType0JsonSchema"
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        type_ = self.type_.value
+
+        json_schema = self.json_schema.to_dict()
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "type": type_,
+                "json_schema": json_schema,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.post_api_prompts_response_200_response_format_type_0_json_schema import (
+            PostApiPromptsResponse200ResponseFormatType0JsonSchema,
+        )
+
+        d = dict(src_dict)
+        type_ = PostApiPromptsResponse200ResponseFormatType0Type(d.pop("type"))
+
+        json_schema = PostApiPromptsResponse200ResponseFormatType0JsonSchema.from_dict(d.pop("json_schema"))
+
+        post_api_prompts_response_200_response_format_type_0 = cls(
+            type_=type_,
+            json_schema=json_schema,
+        )
+
+        post_api_prompts_response_200_response_format_type_0.additional_properties = d
+        return post_api_prompts_response_200_response_format_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

scenario/_generated/langwatch_api_client/lang_watch_api_client/models/post_api_prompts_response_200_response_format_type_0_json_schema.py

@@ -0,0 +1,77 @@
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+if TYPE_CHECKING:
+    from ..models.post_api_prompts_response_200_response_format_type_0_json_schema_schema import (
+        PostApiPromptsResponse200ResponseFormatType0JsonSchemaSchema,
+    )
+
+
+T = TypeVar("T", bound="PostApiPromptsResponse200ResponseFormatType0JsonSchema")
+
+
+@_attrs_define
+class PostApiPromptsResponse200ResponseFormatType0JsonSchema:
+    """
+    Attributes:
+        name (str):
+        schema (PostApiPromptsResponse200ResponseFormatType0JsonSchemaSchema):
+    """
+
+    name: str
+    schema: "PostApiPromptsResponse200ResponseFormatType0JsonSchemaSchema"
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        name = self.name
+
+        schema = self.schema.to_dict()
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "name": name,
+                "schema": schema,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.post_api_prompts_response_200_response_format_type_0_json_schema_schema import (
+            PostApiPromptsResponse200ResponseFormatType0JsonSchemaSchema,
+        )
+
+        d = dict(src_dict)
+        name = d.pop("name")
+
+        schema = PostApiPromptsResponse200ResponseFormatType0JsonSchemaSchema.from_dict(d.pop("schema"))
+
+        post_api_prompts_response_200_response_format_type_0_json_schema = cls(
+            name=name,
+            schema=schema,
+        )
+
+        post_api_prompts_response_200_response_format_type_0_json_schema.additional_properties = d
+        return post_api_prompts_response_200_response_format_type_0_json_schema
+
+    @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

scenario/_generated/langwatch_api_client/lang_watch_api_client/models/post_api_prompts_response_200_response_format_type_0_json_schema_schema.py

@@ -0,0 +1,44 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="PostApiPromptsResponse200ResponseFormatType0JsonSchemaSchema")
+
+
+@_attrs_define
+class PostApiPromptsResponse200ResponseFormatType0JsonSchemaSchema:
+    """ """
+
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        post_api_prompts_response_200_response_format_type_0_json_schema_schema = cls()
+
+        post_api_prompts_response_200_response_format_type_0_json_schema_schema.additional_properties = d
+        return post_api_prompts_response_200_response_format_type_0_json_schema_schema
+
+    @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

scenario/_generated/langwatch_api_client/lang_watch_api_client/models/post_api_prompts_response_200_response_format_type_0_type.py

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

scenario/_generated/langwatch_api_client/lang_watch_api_client/models/post_api_prompts_response_400.py

@@ -0,0 +1,61 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.post_api_prompts_response_400_error import PostApiPromptsResponse400Error
+
+T = TypeVar("T", bound="PostApiPromptsResponse400")
+
+
+@_attrs_define
+class PostApiPromptsResponse400:
+    """
+    Attributes:
+        error (PostApiPromptsResponse400Error):
+    """
+
+    error: PostApiPromptsResponse400Error
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        error = self.error.value
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "error": error,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        error = PostApiPromptsResponse400Error(d.pop("error"))
+
+        post_api_prompts_response_400 = cls(
+            error=error,
+        )
+
+        post_api_prompts_response_400.additional_properties = d
+        return post_api_prompts_response_400
+
+    @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

scenario/_generated/langwatch_api_client/lang_watch_api_client/models/post_api_prompts_response_400_error.py

@@ -0,0 +1,8 @@
+from enum import Enum
+
+
+class PostApiPromptsResponse400Error(str, Enum):
+    BAD_REQUEST = "Bad Request"
+
+    def __str__(self) -> str:
+        return str(self.value)

scenario/_generated/langwatch_api_client/lang_watch_api_client/models/post_api_prompts_response_401.py

@@ -0,0 +1,61 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.post_api_prompts_response_401_error import PostApiPromptsResponse401Error
+
+T = TypeVar("T", bound="PostApiPromptsResponse401")
+
+
+@_attrs_define
+class PostApiPromptsResponse401:
+    """
+    Attributes:
+        error (PostApiPromptsResponse401Error):
+    """
+
+    error: PostApiPromptsResponse401Error
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        error = self.error.value
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "error": error,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        error = PostApiPromptsResponse401Error(d.pop("error"))
+
+        post_api_prompts_response_401 = cls(
+            error=error,
+        )
+
+        post_api_prompts_response_401.additional_properties = d
+        return post_api_prompts_response_401
+
+    @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

scenario/_generated/langwatch_api_client/lang_watch_api_client/models/post_api_prompts_response_401_error.py

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

scenario/_generated/langwatch_api_client/lang_watch_api_client/models/post_api_prompts_response_500.py

@@ -0,0 +1,59 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="PostApiPromptsResponse500")
+
+
+@_attrs_define
+class PostApiPromptsResponse500:
+    """
+    Attributes:
+        error (str):
+    """
+
+    error: str
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        error = self.error
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "error": error,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        error = d.pop("error")
+
+        post_api_prompts_response_500 = cls(
+            error=error,
+        )
+
+        post_api_prompts_response_500.additional_properties = d
+        return post_api_prompts_response_500
+
+    @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