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/patch_api_annotations_id_body.py

@@ -0,0 +1,77 @@
+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="PatchApiAnnotationsIdBody")
+
+
+@_attrs_define
+class PatchApiAnnotationsIdBody:
+    """
+    Attributes:
+        comment (Union[Unset, str]):
+        is_thumbs_up (Union[Unset, bool]):
+        email (Union[Unset, str]):
+    """
+
+    comment: Union[Unset, str] = UNSET
+    is_thumbs_up: Union[Unset, bool] = UNSET
+    email: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        comment = self.comment
+
+        is_thumbs_up = self.is_thumbs_up
+
+        email = self.email
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if comment is not UNSET:
+            field_dict["comment"] = comment
+        if is_thumbs_up is not UNSET:
+            field_dict["isThumbsUp"] = is_thumbs_up
+        if email is not UNSET:
+            field_dict["email"] = email
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        comment = d.pop("comment", UNSET)
+
+        is_thumbs_up = d.pop("isThumbsUp", UNSET)
+
+        email = d.pop("email", UNSET)
+
+        patch_api_annotations_id_body = cls(
+            comment=comment,
+            is_thumbs_up=is_thumbs_up,
+            email=email,
+        )
+
+        patch_api_annotations_id_body.additional_properties = d
+        return patch_api_annotations_id_body
+
+    @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/patch_api_annotations_id_response_200.py

@@ -0,0 +1,68 @@
+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="PatchApiAnnotationsIdResponse200")
+
+
+@_attrs_define
+class PatchApiAnnotationsIdResponse200:
+    """
+    Attributes:
+        status (Union[Unset, str]):
+        message (Union[Unset, str]):
+    """
+
+    status: Union[Unset, str] = UNSET
+    message: 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
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if status is not UNSET:
+            field_dict["status"] = status
+        if message is not UNSET:
+            field_dict["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", UNSET)
+
+        message = d.pop("message", UNSET)
+
+        patch_api_annotations_id_response_200 = cls(
+            status=status,
+            message=message,
+        )
+
+        patch_api_annotations_id_response_200.additional_properties = d
+        return patch_api_annotations_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/post_api_annotations_trace_id_body.py

@@ -0,0 +1,77 @@
+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="PostApiAnnotationsTraceIdBody")
+
+
+@_attrs_define
+class PostApiAnnotationsTraceIdBody:
+    """
+    Attributes:
+        comment (Union[Unset, str]):
+        is_thumbs_up (Union[Unset, bool]):
+        email (Union[Unset, str]):
+    """
+
+    comment: Union[Unset, str] = UNSET
+    is_thumbs_up: Union[Unset, bool] = UNSET
+    email: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        comment = self.comment
+
+        is_thumbs_up = self.is_thumbs_up
+
+        email = self.email
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if comment is not UNSET:
+            field_dict["comment"] = comment
+        if is_thumbs_up is not UNSET:
+            field_dict["isThumbsUp"] = is_thumbs_up
+        if email is not UNSET:
+            field_dict["email"] = email
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        comment = d.pop("comment", UNSET)
+
+        is_thumbs_up = d.pop("isThumbsUp", UNSET)
+
+        email = d.pop("email", UNSET)
+
+        post_api_annotations_trace_id_body = cls(
+            comment=comment,
+            is_thumbs_up=is_thumbs_up,
+            email=email,
+        )
+
+        post_api_annotations_trace_id_body.additional_properties = d
+        return post_api_annotations_trace_id_body
+
+    @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_body.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="PostApiPromptsBody")
+
+
+@_attrs_define
+class PostApiPromptsBody:
+    """
+    Attributes:
+        name (str):
+    """
+
+    name: str
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        name = self.name
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "name": name,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        name = d.pop("name")
+
+        post_api_prompts_body = cls(
+            name=name,
+        )
+
+        post_api_prompts_body.additional_properties = d
+        return post_api_prompts_body
+
+    @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.py

@@ -0,0 +1,147 @@
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, Literal, TypeVar, Union, cast
+
+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 import PostApiPromptsByIdVersionsBodyConfigData
+
+
+T = TypeVar("T", bound="PostApiPromptsByIdVersionsBody")
+
+
+@_attrs_define
+class PostApiPromptsByIdVersionsBody:
+    """
+    Attributes:
+        project_id (str):
+        config_id (str):
+        schema_version (Literal['1.0']):
+        commit_message (str):
+        version (float):
+        config_data (PostApiPromptsByIdVersionsBodyConfigData):
+        id (Union[Unset, str]):
+        author_id (Union[None, Unset, str]):
+        created_at (Union[Unset, str]):
+    """
+
+    project_id: str
+    config_id: str
+    schema_version: Literal["1.0"]
+    commit_message: str
+    version: float
+    config_data: "PostApiPromptsByIdVersionsBodyConfigData"
+    id: Union[Unset, str] = UNSET
+    author_id: Union[None, Unset, str] = UNSET
+    created_at: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        project_id = self.project_id
+
+        config_id = self.config_id
+
+        schema_version = self.schema_version
+
+        commit_message = self.commit_message
+
+        version = self.version
+
+        config_data = self.config_data.to_dict()
+
+        id = self.id
+
+        author_id: Union[None, Unset, str]
+        if isinstance(self.author_id, Unset):
+            author_id = UNSET
+        else:
+            author_id = self.author_id
+
+        created_at = self.created_at
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "projectId": project_id,
+                "configId": config_id,
+                "schemaVersion": schema_version,
+                "commitMessage": commit_message,
+                "version": version,
+                "configData": config_data,
+            }
+        )
+        if id is not UNSET:
+            field_dict["id"] = id
+        if author_id is not UNSET:
+            field_dict["authorId"] = author_id
+        if created_at is not UNSET:
+            field_dict["createdAt"] = created_at
+
+        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 import PostApiPromptsByIdVersionsBodyConfigData
+
+        d = dict(src_dict)
+        project_id = d.pop("projectId")
+
+        config_id = d.pop("configId")
+
+        schema_version = cast(Literal["1.0"], d.pop("schemaVersion"))
+        if schema_version != "1.0":
+            raise ValueError(f"schemaVersion must match const '1.0', got '{schema_version}'")
+
+        commit_message = d.pop("commitMessage")
+
+        version = d.pop("version")
+
+        config_data = PostApiPromptsByIdVersionsBodyConfigData.from_dict(d.pop("configData"))
+
+        id = d.pop("id", UNSET)
+
+        def _parse_author_id(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        author_id = _parse_author_id(d.pop("authorId", UNSET))
+
+        created_at = d.pop("createdAt", UNSET)
+
+        post_api_prompts_by_id_versions_body = cls(
+            project_id=project_id,
+            config_id=config_id,
+            schema_version=schema_version,
+            commit_message=commit_message,
+            version=version,
+            config_data=config_data,
+            id=id,
+            author_id=author_id,
+            created_at=created_at,
+        )
+
+        post_api_prompts_by_id_versions_body.additional_properties = d
+        return post_api_prompts_by_id_versions_body
+
+    @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.py

@@ -0,0 +1,207 @@
+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 import (
+        PostApiPromptsByIdVersionsBodyConfigDataDemonstrations,
+    )
+    from ..models.post_api_prompts_by_id_versions_body_config_data_inputs_item import (
+        PostApiPromptsByIdVersionsBodyConfigDataInputsItem,
+    )
+    from ..models.post_api_prompts_by_id_versions_body_config_data_messages_item import (
+        PostApiPromptsByIdVersionsBodyConfigDataMessagesItem,
+    )
+    from ..models.post_api_prompts_by_id_versions_body_config_data_outputs_item import (
+        PostApiPromptsByIdVersionsBodyConfigDataOutputsItem,
+    )
+    from ..models.post_api_prompts_by_id_versions_body_config_data_prompting_technique import (
+        PostApiPromptsByIdVersionsBodyConfigDataPromptingTechnique,
+    )
+
+
+T = TypeVar("T", bound="PostApiPromptsByIdVersionsBodyConfigData")
+
+
+@_attrs_define
+class PostApiPromptsByIdVersionsBodyConfigData:
+    """
+    Attributes:
+        prompt (str):
+        inputs (list['PostApiPromptsByIdVersionsBodyConfigDataInputsItem']):
+        outputs (list['PostApiPromptsByIdVersionsBodyConfigDataOutputsItem']):
+        model (str):
+        demonstrations (PostApiPromptsByIdVersionsBodyConfigDataDemonstrations):
+        version (Union[Unset, float]):
+        messages (Union[Unset, list['PostApiPromptsByIdVersionsBodyConfigDataMessagesItem']]):
+        temperature (Union[Unset, float]):
+        max_tokens (Union[Unset, float]):
+        prompting_technique (Union[Unset, PostApiPromptsByIdVersionsBodyConfigDataPromptingTechnique]):
+    """
+
+    prompt: str
+    inputs: list["PostApiPromptsByIdVersionsBodyConfigDataInputsItem"]
+    outputs: list["PostApiPromptsByIdVersionsBodyConfigDataOutputsItem"]
+    model: str
+    demonstrations: "PostApiPromptsByIdVersionsBodyConfigDataDemonstrations"
+    version: Union[Unset, float] = UNSET
+    messages: Union[Unset, list["PostApiPromptsByIdVersionsBodyConfigDataMessagesItem"]] = UNSET
+    temperature: Union[Unset, float] = UNSET
+    max_tokens: Union[Unset, float] = UNSET
+    prompting_technique: Union[Unset, "PostApiPromptsByIdVersionsBodyConfigDataPromptingTechnique"] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        prompt = self.prompt
+
+        inputs = []
+        for inputs_item_data in self.inputs:
+            inputs_item = inputs_item_data.to_dict()
+            inputs.append(inputs_item)
+
+        outputs = []
+        for outputs_item_data in self.outputs:
+            outputs_item = outputs_item_data.to_dict()
+            outputs.append(outputs_item)
+
+        model = self.model
+
+        demonstrations = self.demonstrations.to_dict()
+
+        version = self.version
+
+        messages: Union[Unset, list[dict[str, Any]]] = UNSET
+        if not isinstance(self.messages, Unset):
+            messages = []
+            for messages_item_data in self.messages:
+                messages_item = messages_item_data.to_dict()
+                messages.append(messages_item)
+
+        temperature = self.temperature
+
+        max_tokens = self.max_tokens
+
+        prompting_technique: Union[Unset, dict[str, Any]] = UNSET
+        if not isinstance(self.prompting_technique, Unset):
+            prompting_technique = self.prompting_technique.to_dict()
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "prompt": prompt,
+                "inputs": inputs,
+                "outputs": outputs,
+                "model": model,
+                "demonstrations": demonstrations,
+            }
+        )
+        if version is not UNSET:
+            field_dict["version"] = version
+        if messages is not UNSET:
+            field_dict["messages"] = messages
+        if temperature is not UNSET:
+            field_dict["temperature"] = temperature
+        if max_tokens is not UNSET:
+            field_dict["max_tokens"] = max_tokens
+        if prompting_technique is not UNSET:
+            field_dict["prompting_technique"] = prompting_technique
+
+        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 import (
+            PostApiPromptsByIdVersionsBodyConfigDataDemonstrations,
+        )
+        from ..models.post_api_prompts_by_id_versions_body_config_data_inputs_item import (
+            PostApiPromptsByIdVersionsBodyConfigDataInputsItem,
+        )
+        from ..models.post_api_prompts_by_id_versions_body_config_data_messages_item import (
+            PostApiPromptsByIdVersionsBodyConfigDataMessagesItem,
+        )
+        from ..models.post_api_prompts_by_id_versions_body_config_data_outputs_item import (
+            PostApiPromptsByIdVersionsBodyConfigDataOutputsItem,
+        )
+        from ..models.post_api_prompts_by_id_versions_body_config_data_prompting_technique import (
+            PostApiPromptsByIdVersionsBodyConfigDataPromptingTechnique,
+        )
+
+        d = dict(src_dict)
+        prompt = d.pop("prompt")
+
+        inputs = []
+        _inputs = d.pop("inputs")
+        for inputs_item_data in _inputs:
+            inputs_item = PostApiPromptsByIdVersionsBodyConfigDataInputsItem.from_dict(inputs_item_data)
+
+            inputs.append(inputs_item)
+
+        outputs = []
+        _outputs = d.pop("outputs")
+        for outputs_item_data in _outputs:
+            outputs_item = PostApiPromptsByIdVersionsBodyConfigDataOutputsItem.from_dict(outputs_item_data)
+
+            outputs.append(outputs_item)
+
+        model = d.pop("model")
+
+        demonstrations = PostApiPromptsByIdVersionsBodyConfigDataDemonstrations.from_dict(d.pop("demonstrations"))
+
+        version = d.pop("version", UNSET)
+
+        messages = []
+        _messages = d.pop("messages", UNSET)
+        for messages_item_data in _messages or []:
+            messages_item = PostApiPromptsByIdVersionsBodyConfigDataMessagesItem.from_dict(messages_item_data)
+
+            messages.append(messages_item)
+
+        temperature = d.pop("temperature", UNSET)
+
+        max_tokens = d.pop("max_tokens", UNSET)
+
+        _prompting_technique = d.pop("prompting_technique", UNSET)
+        prompting_technique: Union[Unset, PostApiPromptsByIdVersionsBodyConfigDataPromptingTechnique]
+        if isinstance(_prompting_technique, Unset):
+            prompting_technique = UNSET
+        else:
+            prompting_technique = PostApiPromptsByIdVersionsBodyConfigDataPromptingTechnique.from_dict(
+                _prompting_technique
+            )
+
+        post_api_prompts_by_id_versions_body_config_data = cls(
+            prompt=prompt,
+            inputs=inputs,
+            outputs=outputs,
+            model=model,
+            demonstrations=demonstrations,
+            version=version,
+            messages=messages,
+            temperature=temperature,
+            max_tokens=max_tokens,
+            prompting_technique=prompting_technique,
+        )
+
+        post_api_prompts_by_id_versions_body_config_data.additional_properties = d
+        return post_api_prompts_by_id_versions_body_config_data
+
+    @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