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

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

@@ -0,0 +1,78 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="GetApiDatasetBySlugOrIdResponse404")
+
+
+@_attrs_define
+class GetApiDatasetBySlugOrIdResponse404:
+    """
+    Attributes:
+        status (str):  Default: 'error'.
+        message (str):
+        error (Union[Unset, str]):
+    """
+
+    message: str
+    status: str = "error"
+    error: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        status = self.status
+
+        message = self.message
+
+        error = self.error
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "status": status,
+                "message": message,
+            }
+        )
+        if error is not UNSET:
+            field_dict["error"] = error
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        status = d.pop("status")
+
+        message = d.pop("message")
+
+        error = d.pop("error", UNSET)
+
+        get_api_dataset_by_slug_or_id_response_404 = cls(
+            status=status,
+            message=message,
+            error=error,
+        )
+
+        get_api_dataset_by_slug_or_id_response_404.additional_properties = d
+        return get_api_dataset_by_slug_or_id_response_404
+
+    @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/get_api_dataset_by_slug_or_id_response_422.py

@@ -0,0 +1,67 @@
+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="GetApiDatasetBySlugOrIdResponse422")
+
+
+@_attrs_define
+class GetApiDatasetBySlugOrIdResponse422:
+    """
+    Attributes:
+        status (str):  Default: 'error'.
+        message (str):
+    """
+
+    message: str
+    status: str = "error"
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        status = self.status
+
+        message = self.message
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "status": status,
+                "message": message,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        status = d.pop("status")
+
+        message = d.pop("message")
+
+        get_api_dataset_by_slug_or_id_response_422 = cls(
+            status=status,
+            message=message,
+        )
+
+        get_api_dataset_by_slug_or_id_response_422.additional_properties = d
+        return get_api_dataset_by_slug_or_id_response_422
+
+    @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/get_api_dataset_by_slug_or_id_response_500.py

@@ -0,0 +1,78 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="GetApiDatasetBySlugOrIdResponse500")
+
+
+@_attrs_define
+class GetApiDatasetBySlugOrIdResponse500:
+    """
+    Attributes:
+        status (str):  Default: 'error'.
+        message (str):
+        error (Union[Unset, str]):
+    """
+
+    message: str
+    status: str = "error"
+    error: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        status = self.status
+
+        message = self.message
+
+        error = self.error
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "status": status,
+                "message": message,
+            }
+        )
+        if error is not UNSET:
+            field_dict["error"] = error
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        status = d.pop("status")
+
+        message = d.pop("message")
+
+        error = d.pop("error", UNSET)
+
+        get_api_dataset_by_slug_or_id_response_500 = cls(
+            status=status,
+            message=message,
+            error=error,
+        )
+
+        get_api_dataset_by_slug_or_id_response_500.additional_properties = d
+        return get_api_dataset_by_slug_or_id_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

scenario/_generated/langwatch_api_client/lang_watch_api_client/models/get_api_prompts_by_id_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.get_api_prompts_by_id_response_200_messages_item import GetApiPromptsByIdResponse200MessagesItem
+    from ..models.get_api_prompts_by_id_response_200_response_format_type_0 import (
+        GetApiPromptsByIdResponse200ResponseFormatType0,
+    )
+
+
+T = TypeVar("T", bound="GetApiPromptsByIdResponse200")
+
+
+@_attrs_define
+class GetApiPromptsByIdResponse200:
+    """
+    Attributes:
+        id (str):
+        name (str):
+        updated_at (str):
+        version (float):
+        version_id (str):
+        version_created_at (str):
+        model (str):
+        prompt (str):
+        messages (list['GetApiPromptsByIdResponse200MessagesItem']):
+        response_format (Union['GetApiPromptsByIdResponse200ResponseFormatType0', None]):
+    """
+
+    id: str
+    name: str
+    updated_at: str
+    version: float
+    version_id: str
+    version_created_at: str
+    model: str
+    prompt: str
+    messages: list["GetApiPromptsByIdResponse200MessagesItem"]
+    response_format: Union["GetApiPromptsByIdResponse200ResponseFormatType0", None]
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        from ..models.get_api_prompts_by_id_response_200_response_format_type_0 import (
+            GetApiPromptsByIdResponse200ResponseFormatType0,
+        )
+
+        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, GetApiPromptsByIdResponse200ResponseFormatType0):
+            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.get_api_prompts_by_id_response_200_messages_item import GetApiPromptsByIdResponse200MessagesItem
+        from ..models.get_api_prompts_by_id_response_200_response_format_type_0 import (
+            GetApiPromptsByIdResponse200ResponseFormatType0,
+        )
+
+        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 = GetApiPromptsByIdResponse200MessagesItem.from_dict(messages_item_data)
+
+            messages.append(messages_item)
+
+        def _parse_response_format(data: object) -> Union["GetApiPromptsByIdResponse200ResponseFormatType0", None]:
+            if data is None:
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                response_format_type_0 = GetApiPromptsByIdResponse200ResponseFormatType0.from_dict(data)
+
+                return response_format_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union["GetApiPromptsByIdResponse200ResponseFormatType0", None], data)
+
+        response_format = _parse_response_format(d.pop("response_format"))
+
+        get_api_prompts_by_id_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,
+        )
+
+        get_api_prompts_by_id_response_200.additional_properties = d
+        return get_api_prompts_by_id_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/get_api_prompts_by_id_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.get_api_prompts_by_id_response_200_messages_item_role import GetApiPromptsByIdResponse200MessagesItemRole
+
+T = TypeVar("T", bound="GetApiPromptsByIdResponse200MessagesItem")
+
+
+@_attrs_define
+class GetApiPromptsByIdResponse200MessagesItem:
+    """
+    Attributes:
+        role (GetApiPromptsByIdResponse200MessagesItemRole):
+        content (str):
+    """
+
+    role: GetApiPromptsByIdResponse200MessagesItemRole
+    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 = GetApiPromptsByIdResponse200MessagesItemRole(d.pop("role"))
+
+        content = d.pop("content")
+
+        get_api_prompts_by_id_response_200_messages_item = cls(
+            role=role,
+            content=content,
+        )
+
+        get_api_prompts_by_id_response_200_messages_item.additional_properties = d
+        return get_api_prompts_by_id_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/get_api_prompts_by_id_response_200_messages_item_role.py

@@ -0,0 +1,10 @@
+from enum import Enum
+
+
+class GetApiPromptsByIdResponse200MessagesItemRole(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/get_api_prompts_by_id_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.get_api_prompts_by_id_response_200_response_format_type_0_type import (
+    GetApiPromptsByIdResponse200ResponseFormatType0Type,
+)
+
+if TYPE_CHECKING:
+    from ..models.get_api_prompts_by_id_response_200_response_format_type_0_json_schema import (
+        GetApiPromptsByIdResponse200ResponseFormatType0JsonSchema,
+    )
+
+
+T = TypeVar("T", bound="GetApiPromptsByIdResponse200ResponseFormatType0")
+
+
+@_attrs_define
+class GetApiPromptsByIdResponse200ResponseFormatType0:
+    """
+    Attributes:
+        type_ (GetApiPromptsByIdResponse200ResponseFormatType0Type):
+        json_schema (GetApiPromptsByIdResponse200ResponseFormatType0JsonSchema):
+    """
+
+    type_: GetApiPromptsByIdResponse200ResponseFormatType0Type
+    json_schema: "GetApiPromptsByIdResponse200ResponseFormatType0JsonSchema"
+    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.get_api_prompts_by_id_response_200_response_format_type_0_json_schema import (
+            GetApiPromptsByIdResponse200ResponseFormatType0JsonSchema,
+        )
+
+        d = dict(src_dict)
+        type_ = GetApiPromptsByIdResponse200ResponseFormatType0Type(d.pop("type"))
+
+        json_schema = GetApiPromptsByIdResponse200ResponseFormatType0JsonSchema.from_dict(d.pop("json_schema"))
+
+        get_api_prompts_by_id_response_200_response_format_type_0 = cls(
+            type_=type_,
+            json_schema=json_schema,
+        )
+
+        get_api_prompts_by_id_response_200_response_format_type_0.additional_properties = d
+        return get_api_prompts_by_id_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/get_api_prompts_by_id_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.get_api_prompts_by_id_response_200_response_format_type_0_json_schema_schema import (
+        GetApiPromptsByIdResponse200ResponseFormatType0JsonSchemaSchema,
+    )
+
+
+T = TypeVar("T", bound="GetApiPromptsByIdResponse200ResponseFormatType0JsonSchema")
+
+
+@_attrs_define
+class GetApiPromptsByIdResponse200ResponseFormatType0JsonSchema:
+    """
+    Attributes:
+        name (str):
+        schema (GetApiPromptsByIdResponse200ResponseFormatType0JsonSchemaSchema):
+    """
+
+    name: str
+    schema: "GetApiPromptsByIdResponse200ResponseFormatType0JsonSchemaSchema"
+    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.get_api_prompts_by_id_response_200_response_format_type_0_json_schema_schema import (
+            GetApiPromptsByIdResponse200ResponseFormatType0JsonSchemaSchema,
+        )
+
+        d = dict(src_dict)
+        name = d.pop("name")
+
+        schema = GetApiPromptsByIdResponse200ResponseFormatType0JsonSchemaSchema.from_dict(d.pop("schema"))
+
+        get_api_prompts_by_id_response_200_response_format_type_0_json_schema = cls(
+            name=name,
+            schema=schema,
+        )
+
+        get_api_prompts_by_id_response_200_response_format_type_0_json_schema.additional_properties = d
+        return get_api_prompts_by_id_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/get_api_prompts_by_id_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="GetApiPromptsByIdResponse200ResponseFormatType0JsonSchemaSchema")
+
+
+@_attrs_define
+class GetApiPromptsByIdResponse200ResponseFormatType0JsonSchemaSchema:
+    """ """
+
+    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)
+        get_api_prompts_by_id_response_200_response_format_type_0_json_schema_schema = cls()
+
+        get_api_prompts_by_id_response_200_response_format_type_0_json_schema_schema.additional_properties = d
+        return get_api_prompts_by_id_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/get_api_prompts_by_id_response_200_response_format_type_0_type.py

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