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

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

@@ -0,0 +1,127 @@
+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_scenario_events_body_type_0_metadata import PostApiScenarioEventsBodyType0Metadata
+
+
+T = TypeVar("T", bound="PostApiScenarioEventsBodyType0")
+
+
+@_attrs_define
+class PostApiScenarioEventsBodyType0:
+    """
+    Attributes:
+        type_ (Literal['SCENARIO_RUN_STARTED']):
+        timestamp (float):
+        batch_run_id (str):
+        scenario_id (str):
+        scenario_run_id (str):
+        metadata (PostApiScenarioEventsBodyType0Metadata):
+        raw_event (Union[Unset, Any]):
+        scenario_set_id (Union[Unset, str]):  Default: 'default'.
+    """
+
+    type_: Literal["SCENARIO_RUN_STARTED"]
+    timestamp: float
+    batch_run_id: str
+    scenario_id: str
+    scenario_run_id: str
+    metadata: "PostApiScenarioEventsBodyType0Metadata"
+    raw_event: Union[Unset, Any] = UNSET
+    scenario_set_id: Union[Unset, str] = "default"
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        type_ = self.type_
+
+        timestamp = self.timestamp
+
+        batch_run_id = self.batch_run_id
+
+        scenario_id = self.scenario_id
+
+        scenario_run_id = self.scenario_run_id
+
+        metadata = self.metadata.to_dict()
+
+        raw_event = self.raw_event
+
+        scenario_set_id = self.scenario_set_id
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "type": type_,
+                "timestamp": timestamp,
+                "batchRunId": batch_run_id,
+                "scenarioId": scenario_id,
+                "scenarioRunId": scenario_run_id,
+                "metadata": metadata,
+            }
+        )
+        if raw_event is not UNSET:
+            field_dict["rawEvent"] = raw_event
+        if scenario_set_id is not UNSET:
+            field_dict["scenarioSetId"] = scenario_set_id
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.post_api_scenario_events_body_type_0_metadata import PostApiScenarioEventsBodyType0Metadata
+
+        d = dict(src_dict)
+        type_ = cast(Literal["SCENARIO_RUN_STARTED"], d.pop("type"))
+        if type_ != "SCENARIO_RUN_STARTED":
+            raise ValueError(f"type must match const 'SCENARIO_RUN_STARTED', got '{type_}'")
+
+        timestamp = d.pop("timestamp")
+
+        batch_run_id = d.pop("batchRunId")
+
+        scenario_id = d.pop("scenarioId")
+
+        scenario_run_id = d.pop("scenarioRunId")
+
+        metadata = PostApiScenarioEventsBodyType0Metadata.from_dict(d.pop("metadata"))
+
+        raw_event = d.pop("rawEvent", UNSET)
+
+        scenario_set_id = d.pop("scenarioSetId", UNSET)
+
+        post_api_scenario_events_body_type_0 = cls(
+            type_=type_,
+            timestamp=timestamp,
+            batch_run_id=batch_run_id,
+            scenario_id=scenario_id,
+            scenario_run_id=scenario_run_id,
+            metadata=metadata,
+            raw_event=raw_event,
+            scenario_set_id=scenario_set_id,
+        )
+
+        post_api_scenario_events_body_type_0.additional_properties = d
+        return post_api_scenario_events_body_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_scenario_events_body_type_0_metadata.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="PostApiScenarioEventsBodyType0Metadata")
+
+
+@_attrs_define
+class PostApiScenarioEventsBodyType0Metadata:
+    """
+    Attributes:
+        name (Union[Unset, str]):
+        description (Union[Unset, str]):
+    """
+
+    name: Union[Unset, str] = UNSET
+    description: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        name = self.name
+
+        description = self.description
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if name is not UNSET:
+            field_dict["name"] = name
+        if description is not UNSET:
+            field_dict["description"] = description
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        name = d.pop("name", UNSET)
+
+        description = d.pop("description", UNSET)
+
+        post_api_scenario_events_body_type_0_metadata = cls(
+            name=name,
+            description=description,
+        )
+
+        post_api_scenario_events_body_type_0_metadata.additional_properties = d
+        return post_api_scenario_events_body_type_0_metadata
+
+    @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_scenario_events_body_type_1.py

@@ -0,0 +1,164 @@
+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 ..models.post_api_scenario_events_body_type_1_status import PostApiScenarioEventsBodyType1Status
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.post_api_scenario_events_body_type_1_results_type_0 import PostApiScenarioEventsBodyType1ResultsType0
+
+
+T = TypeVar("T", bound="PostApiScenarioEventsBodyType1")
+
+
+@_attrs_define
+class PostApiScenarioEventsBodyType1:
+    """
+    Attributes:
+        type_ (Literal['SCENARIO_RUN_FINISHED']):
+        timestamp (float):
+        batch_run_id (str):
+        scenario_id (str):
+        scenario_run_id (str):
+        status (PostApiScenarioEventsBodyType1Status):
+        raw_event (Union[Unset, Any]):
+        scenario_set_id (Union[Unset, str]):  Default: 'default'.
+        results (Union['PostApiScenarioEventsBodyType1ResultsType0', None, Unset]):
+    """
+
+    type_: Literal["SCENARIO_RUN_FINISHED"]
+    timestamp: float
+    batch_run_id: str
+    scenario_id: str
+    scenario_run_id: str
+    status: PostApiScenarioEventsBodyType1Status
+    raw_event: Union[Unset, Any] = UNSET
+    scenario_set_id: Union[Unset, str] = "default"
+    results: Union["PostApiScenarioEventsBodyType1ResultsType0", None, Unset] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        from ..models.post_api_scenario_events_body_type_1_results_type_0 import (
+            PostApiScenarioEventsBodyType1ResultsType0,
+        )
+
+        type_ = self.type_
+
+        timestamp = self.timestamp
+
+        batch_run_id = self.batch_run_id
+
+        scenario_id = self.scenario_id
+
+        scenario_run_id = self.scenario_run_id
+
+        status = self.status.value
+
+        raw_event = self.raw_event
+
+        scenario_set_id = self.scenario_set_id
+
+        results: Union[None, Unset, dict[str, Any]]
+        if isinstance(self.results, Unset):
+            results = UNSET
+        elif isinstance(self.results, PostApiScenarioEventsBodyType1ResultsType0):
+            results = self.results.to_dict()
+        else:
+            results = self.results
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "type": type_,
+                "timestamp": timestamp,
+                "batchRunId": batch_run_id,
+                "scenarioId": scenario_id,
+                "scenarioRunId": scenario_run_id,
+                "status": status,
+            }
+        )
+        if raw_event is not UNSET:
+            field_dict["rawEvent"] = raw_event
+        if scenario_set_id is not UNSET:
+            field_dict["scenarioSetId"] = scenario_set_id
+        if results is not UNSET:
+            field_dict["results"] = results
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.post_api_scenario_events_body_type_1_results_type_0 import (
+            PostApiScenarioEventsBodyType1ResultsType0,
+        )
+
+        d = dict(src_dict)
+        type_ = cast(Literal["SCENARIO_RUN_FINISHED"], d.pop("type"))
+        if type_ != "SCENARIO_RUN_FINISHED":
+            raise ValueError(f"type must match const 'SCENARIO_RUN_FINISHED', got '{type_}'")
+
+        timestamp = d.pop("timestamp")
+
+        batch_run_id = d.pop("batchRunId")
+
+        scenario_id = d.pop("scenarioId")
+
+        scenario_run_id = d.pop("scenarioRunId")
+
+        status = PostApiScenarioEventsBodyType1Status(d.pop("status"))
+
+        raw_event = d.pop("rawEvent", UNSET)
+
+        scenario_set_id = d.pop("scenarioSetId", UNSET)
+
+        def _parse_results(data: object) -> Union["PostApiScenarioEventsBodyType1ResultsType0", None, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                results_type_0 = PostApiScenarioEventsBodyType1ResultsType0.from_dict(data)
+
+                return results_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union["PostApiScenarioEventsBodyType1ResultsType0", None, Unset], data)
+
+        results = _parse_results(d.pop("results", UNSET))
+
+        post_api_scenario_events_body_type_1 = cls(
+            type_=type_,
+            timestamp=timestamp,
+            batch_run_id=batch_run_id,
+            scenario_id=scenario_id,
+            scenario_run_id=scenario_run_id,
+            status=status,
+            raw_event=raw_event,
+            scenario_set_id=scenario_set_id,
+            results=results,
+        )
+
+        post_api_scenario_events_body_type_1.additional_properties = d
+        return post_api_scenario_events_body_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

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

@@ -0,0 +1,98 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.post_api_scenario_events_body_type_1_results_type_0_verdict import (
+    PostApiScenarioEventsBodyType1ResultsType0Verdict,
+)
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="PostApiScenarioEventsBodyType1ResultsType0")
+
+
+@_attrs_define
+class PostApiScenarioEventsBodyType1ResultsType0:
+    """
+    Attributes:
+        verdict (PostApiScenarioEventsBodyType1ResultsType0Verdict):
+        met_criteria (list[str]):
+        unmet_criteria (list[str]):
+        reasoning (Union[Unset, str]):
+        error (Union[Unset, str]):
+    """
+
+    verdict: PostApiScenarioEventsBodyType1ResultsType0Verdict
+    met_criteria: list[str]
+    unmet_criteria: list[str]
+    reasoning: Union[Unset, str] = UNSET
+    error: Union[Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        verdict = self.verdict.value
+
+        met_criteria = self.met_criteria
+
+        unmet_criteria = self.unmet_criteria
+
+        reasoning = self.reasoning
+
+        error = self.error
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "verdict": verdict,
+                "metCriteria": met_criteria,
+                "unmetCriteria": unmet_criteria,
+            }
+        )
+        if reasoning is not UNSET:
+            field_dict["reasoning"] = reasoning
+        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)
+        verdict = PostApiScenarioEventsBodyType1ResultsType0Verdict(d.pop("verdict"))
+
+        met_criteria = cast(list[str], d.pop("metCriteria"))
+
+        unmet_criteria = cast(list[str], d.pop("unmetCriteria"))
+
+        reasoning = d.pop("reasoning", UNSET)
+
+        error = d.pop("error", UNSET)
+
+        post_api_scenario_events_body_type_1_results_type_0 = cls(
+            verdict=verdict,
+            met_criteria=met_criteria,
+            unmet_criteria=unmet_criteria,
+            reasoning=reasoning,
+            error=error,
+        )
+
+        post_api_scenario_events_body_type_1_results_type_0.additional_properties = d
+        return post_api_scenario_events_body_type_1_results_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_scenario_events_body_type_1_results_type_0_verdict.py

@@ -0,0 +1,10 @@
+from enum import Enum
+
+
+class PostApiScenarioEventsBodyType1ResultsType0Verdict(str, Enum):
+    FAILURE = "failure"
+    INCONCLUSIVE = "inconclusive"
+    SUCCESS = "success"
+
+    def __str__(self) -> str:
+        return str(self.value)

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

@@ -0,0 +1,13 @@
+from enum import Enum
+
+
+class PostApiScenarioEventsBodyType1Status(str, Enum):
+    CANCELLED = "CANCELLED"
+    ERROR = "ERROR"
+    FAILED = "FAILED"
+    IN_PROGRESS = "IN_PROGRESS"
+    PENDING = "PENDING"
+    SUCCESS = "SUCCESS"
+
+    def __str__(self) -> str:
+        return str(self.value)