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_by_id_versions_body_config_data_demonstrations.py

@@ -0,0 +1,106 @@
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.post_api_prompts_by_id_versions_body_config_data_demonstrations_columns_item import (
+        PostApiPromptsByIdVersionsBodyConfigDataDemonstrationsColumnsItem,
+    )
+    from ..models.post_api_prompts_by_id_versions_body_config_data_demonstrations_rows_item import (
+        PostApiPromptsByIdVersionsBodyConfigDataDemonstrationsRowsItem,
+    )
+
+
+T = TypeVar("T", bound="PostApiPromptsByIdVersionsBodyConfigDataDemonstrations")
+
+
+@_attrs_define
+class PostApiPromptsByIdVersionsBodyConfigDataDemonstrations:
+    """
+    Attributes:
+        columns (list['PostApiPromptsByIdVersionsBodyConfigDataDemonstrationsColumnsItem']):
+        rows (Union[Unset, list['PostApiPromptsByIdVersionsBodyConfigDataDemonstrationsRowsItem']]):
+    """
+
+    columns: list["PostApiPromptsByIdVersionsBodyConfigDataDemonstrationsColumnsItem"]
+    rows: Union[Unset, list["PostApiPromptsByIdVersionsBodyConfigDataDemonstrationsRowsItem"]] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        columns = []
+        for columns_item_data in self.columns:
+            columns_item = columns_item_data.to_dict()
+            columns.append(columns_item)
+
+        rows: Union[Unset, list[dict[str, Any]]] = UNSET
+        if not isinstance(self.rows, Unset):
+            rows = []
+            for rows_item_data in self.rows:
+                rows_item = rows_item_data.to_dict()
+                rows.append(rows_item)
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "columns": columns,
+            }
+        )
+        if rows is not UNSET:
+            field_dict["rows"] = rows
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.post_api_prompts_by_id_versions_body_config_data_demonstrations_columns_item import (
+            PostApiPromptsByIdVersionsBodyConfigDataDemonstrationsColumnsItem,
+        )
+        from ..models.post_api_prompts_by_id_versions_body_config_data_demonstrations_rows_item import (
+            PostApiPromptsByIdVersionsBodyConfigDataDemonstrationsRowsItem,
+        )
+
+        d = dict(src_dict)
+        columns = []
+        _columns = d.pop("columns")
+        for columns_item_data in _columns:
+            columns_item = PostApiPromptsByIdVersionsBodyConfigDataDemonstrationsColumnsItem.from_dict(
+                columns_item_data
+            )
+
+            columns.append(columns_item)
+
+        rows = []
+        _rows = d.pop("rows", UNSET)
+        for rows_item_data in _rows or []:
+            rows_item = PostApiPromptsByIdVersionsBodyConfigDataDemonstrationsRowsItem.from_dict(rows_item_data)
+
+            rows.append(rows_item)
+
+        post_api_prompts_by_id_versions_body_config_data_demonstrations = cls(
+            columns=columns,
+            rows=rows,
+        )
+
+        post_api_prompts_by_id_versions_body_config_data_demonstrations.additional_properties = d
+        return post_api_prompts_by_id_versions_body_config_data_demonstrations
+
+    @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_by_id_versions_body_config_data_demonstrations_columns_item.py

@@ -0,0 +1,79 @@
+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_by_id_versions_body_config_data_demonstrations_columns_item_type import (
+    PostApiPromptsByIdVersionsBodyConfigDataDemonstrationsColumnsItemType,
+)
+
+T = TypeVar("T", bound="PostApiPromptsByIdVersionsBodyConfigDataDemonstrationsColumnsItem")
+
+
+@_attrs_define
+class PostApiPromptsByIdVersionsBodyConfigDataDemonstrationsColumnsItem:
+    """
+    Attributes:
+        id (str):
+        name (str):
+        type_ (PostApiPromptsByIdVersionsBodyConfigDataDemonstrationsColumnsItemType):
+    """
+
+    id: str
+    name: str
+    type_: PostApiPromptsByIdVersionsBodyConfigDataDemonstrationsColumnsItemType
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        id = self.id
+
+        name = self.name
+
+        type_ = self.type_.value
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "id": id,
+                "name": name,
+                "type": type_,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        id = d.pop("id")
+
+        name = d.pop("name")
+
+        type_ = PostApiPromptsByIdVersionsBodyConfigDataDemonstrationsColumnsItemType(d.pop("type"))
+
+        post_api_prompts_by_id_versions_body_config_data_demonstrations_columns_item = cls(
+            id=id,
+            name=name,
+            type_=type_,
+        )
+
+        post_api_prompts_by_id_versions_body_config_data_demonstrations_columns_item.additional_properties = d
+        return post_api_prompts_by_id_versions_body_config_data_demonstrations_columns_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_by_id_versions_body_config_data_demonstrations_columns_item_type.py

@@ -0,0 +1,18 @@
+from enum import Enum
+
+
+class PostApiPromptsByIdVersionsBodyConfigDataDemonstrationsColumnsItemType(str, Enum):
+    ANNOTATIONS = "annotations"
+    BOOLEAN = "boolean"
+    CHAT_MESSAGES = "chat_messages"
+    DATE = "date"
+    EVALUATIONS = "evaluations"
+    JSON = "json"
+    LIST = "list"
+    NUMBER = "number"
+    RAG_CONTEXTS = "rag_contexts"
+    SPANS = "spans"
+    STRING = "string"
+
+    def __str__(self) -> str:
+        return str(self.value)

scenario/_generated/langwatch_api_client/lang_watch_api_client/models/post_api_prompts_by_id_versions_body_config_data_demonstrations_rows_item.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="PostApiPromptsByIdVersionsBodyConfigDataDemonstrationsRowsItem")
+
+
+@_attrs_define
+class PostApiPromptsByIdVersionsBodyConfigDataDemonstrationsRowsItem:
+    """
+    Attributes:
+        id (str):
+    """
+
+    id: str
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        id = self.id
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "id": id,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        id = d.pop("id")
+
+        post_api_prompts_by_id_versions_body_config_data_demonstrations_rows_item = cls(
+            id=id,
+        )
+
+        post_api_prompts_by_id_versions_body_config_data_demonstrations_rows_item.additional_properties = d
+        return post_api_prompts_by_id_versions_body_config_data_demonstrations_rows_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_by_id_versions_body_config_data_inputs_item.py

@@ -0,0 +1,71 @@
+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_by_id_versions_body_config_data_inputs_item_type import (
+    PostApiPromptsByIdVersionsBodyConfigDataInputsItemType,
+)
+
+T = TypeVar("T", bound="PostApiPromptsByIdVersionsBodyConfigDataInputsItem")
+
+
+@_attrs_define
+class PostApiPromptsByIdVersionsBodyConfigDataInputsItem:
+    """
+    Attributes:
+        identifier (str):
+        type_ (PostApiPromptsByIdVersionsBodyConfigDataInputsItemType):
+    """
+
+    identifier: str
+    type_: PostApiPromptsByIdVersionsBodyConfigDataInputsItemType
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        identifier = self.identifier
+
+        type_ = self.type_.value
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "identifier": identifier,
+                "type": type_,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        identifier = d.pop("identifier")
+
+        type_ = PostApiPromptsByIdVersionsBodyConfigDataInputsItemType(d.pop("type"))
+
+        post_api_prompts_by_id_versions_body_config_data_inputs_item = cls(
+            identifier=identifier,
+            type_=type_,
+        )
+
+        post_api_prompts_by_id_versions_body_config_data_inputs_item.additional_properties = d
+        return post_api_prompts_by_id_versions_body_config_data_inputs_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_by_id_versions_body_config_data_inputs_item_type.py

@@ -0,0 +1,16 @@
+from enum import Enum
+
+
+class PostApiPromptsByIdVersionsBodyConfigDataInputsItemType(str, Enum):
+    BOOL = "bool"
+    DICT = "dict"
+    FLOAT = "float"
+    IMAGE = "image"
+    LISTBOOL = "list[bool]"
+    LISTFLOAT = "list[float]"
+    LISTINT = "list[int]"
+    LISTSTR = "list[str]"
+    STR = "str"
+
+    def __str__(self) -> str:
+        return str(self.value)

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

@@ -0,0 +1,71 @@
+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_by_id_versions_body_config_data_messages_item_role import (
+    PostApiPromptsByIdVersionsBodyConfigDataMessagesItemRole,
+)
+
+T = TypeVar("T", bound="PostApiPromptsByIdVersionsBodyConfigDataMessagesItem")
+
+
+@_attrs_define
+class PostApiPromptsByIdVersionsBodyConfigDataMessagesItem:
+    """
+    Attributes:
+        role (PostApiPromptsByIdVersionsBodyConfigDataMessagesItemRole):
+        content (str):
+    """
+
+    role: PostApiPromptsByIdVersionsBodyConfigDataMessagesItemRole
+    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 = PostApiPromptsByIdVersionsBodyConfigDataMessagesItemRole(d.pop("role"))
+
+        content = d.pop("content")
+
+        post_api_prompts_by_id_versions_body_config_data_messages_item = cls(
+            role=role,
+            content=content,
+        )
+
+        post_api_prompts_by_id_versions_body_config_data_messages_item.additional_properties = d
+        return post_api_prompts_by_id_versions_body_config_data_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_by_id_versions_body_config_data_messages_item_role.py

@@ -0,0 +1,10 @@
+from enum import Enum
+
+
+class PostApiPromptsByIdVersionsBodyConfigDataMessagesItemRole(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_by_id_versions_body_config_data_outputs_item.py

@@ -0,0 +1,98 @@
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.post_api_prompts_by_id_versions_body_config_data_outputs_item_type import (
+    PostApiPromptsByIdVersionsBodyConfigDataOutputsItemType,
+)
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.post_api_prompts_by_id_versions_body_config_data_outputs_item_json_schema import (
+        PostApiPromptsByIdVersionsBodyConfigDataOutputsItemJsonSchema,
+    )
+
+
+T = TypeVar("T", bound="PostApiPromptsByIdVersionsBodyConfigDataOutputsItem")
+
+
+@_attrs_define
+class PostApiPromptsByIdVersionsBodyConfigDataOutputsItem:
+    """
+    Attributes:
+        identifier (str):
+        type_ (PostApiPromptsByIdVersionsBodyConfigDataOutputsItemType):
+        json_schema (Union[Unset, PostApiPromptsByIdVersionsBodyConfigDataOutputsItemJsonSchema]):
+    """
+
+    identifier: str
+    type_: PostApiPromptsByIdVersionsBodyConfigDataOutputsItemType
+    json_schema: Union[Unset, "PostApiPromptsByIdVersionsBodyConfigDataOutputsItemJsonSchema"] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        identifier = self.identifier
+
+        type_ = self.type_.value
+
+        json_schema: Union[Unset, dict[str, Any]] = UNSET
+        if not isinstance(self.json_schema, Unset):
+            json_schema = self.json_schema.to_dict()
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "identifier": identifier,
+                "type": type_,
+            }
+        )
+        if json_schema is not UNSET:
+            field_dict["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_by_id_versions_body_config_data_outputs_item_json_schema import (
+            PostApiPromptsByIdVersionsBodyConfigDataOutputsItemJsonSchema,
+        )
+
+        d = dict(src_dict)
+        identifier = d.pop("identifier")
+
+        type_ = PostApiPromptsByIdVersionsBodyConfigDataOutputsItemType(d.pop("type"))
+
+        _json_schema = d.pop("json_schema", UNSET)
+        json_schema: Union[Unset, PostApiPromptsByIdVersionsBodyConfigDataOutputsItemJsonSchema]
+        if isinstance(_json_schema, Unset):
+            json_schema = UNSET
+        else:
+            json_schema = PostApiPromptsByIdVersionsBodyConfigDataOutputsItemJsonSchema.from_dict(_json_schema)
+
+        post_api_prompts_by_id_versions_body_config_data_outputs_item = cls(
+            identifier=identifier,
+            type_=type_,
+            json_schema=json_schema,
+        )
+
+        post_api_prompts_by_id_versions_body_config_data_outputs_item.additional_properties = d
+        return post_api_prompts_by_id_versions_body_config_data_outputs_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_by_id_versions_body_config_data_outputs_item_json_schema.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="PostApiPromptsByIdVersionsBodyConfigDataOutputsItemJsonSchema")
+
+
+@_attrs_define
+class PostApiPromptsByIdVersionsBodyConfigDataOutputsItemJsonSchema:
+    """
+    Attributes:
+        type_ (str):
+    """
+
+    type_: str
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        type_ = self.type_
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "type": type_,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        type_ = d.pop("type")
+
+        post_api_prompts_by_id_versions_body_config_data_outputs_item_json_schema = cls(
+            type_=type_,
+        )
+
+        post_api_prompts_by_id_versions_body_config_data_outputs_item_json_schema.additional_properties = d
+        return post_api_prompts_by_id_versions_body_config_data_outputs_item_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_by_id_versions_body_config_data_outputs_item_type.py

@@ -0,0 +1,11 @@
+from enum import Enum
+
+
+class PostApiPromptsByIdVersionsBodyConfigDataOutputsItemType(str, Enum):
+    BOOL = "bool"
+    FLOAT = "float"
+    JSON_SCHEMA = "json_schema"
+    STR = "str"
+
+    def __str__(self) -> str:
+        return str(self.value)

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

@@ -0,0 +1,59 @@
+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="PostApiPromptsByIdVersionsBodyConfigDataPromptingTechnique")
+
+
+@_attrs_define
+class PostApiPromptsByIdVersionsBodyConfigDataPromptingTechnique:
+    """
+    Attributes:
+        ref (Union[Unset, str]):
+    """
+
+    ref: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        ref = self.ref
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if ref is not UNSET:
+            field_dict["ref"] = ref
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        ref = d.pop("ref", UNSET)
+
+        post_api_prompts_by_id_versions_body_config_data_prompting_technique = cls(
+            ref=ref,
+        )
+
+        post_api_prompts_by_id_versions_body_config_data_prompting_technique.additional_properties = d
+        return post_api_prompts_by_id_versions_body_config_data_prompting_technique
+
+    @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