orq-ai-sdk 4.2.0rc28__py3-none-any.whl → 4.2.6__py3-none-any.whl

orq_ai_sdk/models/getevalsop.py

@@ -4,10 +4,10 @@ from __future__ import annotations
 from dataclasses import dataclass, field
 import httpx
 from orq_ai_sdk.models import OrqError
-from orq_ai_sdk.types import BaseModel
+from orq_ai_sdk.types import BaseModel, UNSET_SENTINEL
 from orq_ai_sdk.utils import FieldMetadata, QueryParamMetadata, get_discriminator
 import pydantic
-from pydantic import Discriminator, Tag
+from pydantic import Discriminator, Tag, model_serializer
 from typing import Any, Dict, List, Literal, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
@@ -40,6 +40,22 @@ class GetEvalsRequest(BaseModel):
     ] = None
     r"""A cursor for use in pagination. `ending_before` is an object ID that defines your place in the list. For instance, if you make a list request and receive 20 objects, starting with `01JJ1HDHN79XAS7A01WB3HYSDB`, your subsequent call can include `before=01JJ1HDHN79XAS7A01WB3HYSDB` in order to fetch the previous page of the list."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["limit", "starting_after", "ending_before"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 class GetEvalsEvalsResponseBodyData(BaseModel):
     message: str
@@ -63,7 +79,7 @@ class GetEvalsEvalsResponseBody(OrqError):
         object.__setattr__(self, "data", data)
 
 
-GetEvalsObject = Literal["list",]
+Object = Literal["list",]
 
 
 GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyData72Type = Literal[
@@ -100,6 +116,22 @@ class GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONNumber(BaseModel):
 
     alert_on_failure: Optional[bool] = False
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["alert_on_failure"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyData7Type = Literal[
     "boolean",
@@ -122,6 +154,22 @@ class GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONBoolean(BaseModel):
 
     alert_on_failure: Optional[bool] = False
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["alert_on_failure"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 GetEvalsDataEvalsResponse200ApplicationJSONGuardrailConfigTypedDict = TypeAliasType(
     "GetEvalsDataEvalsResponse200ApplicationJSONGuardrailConfigTypedDict",
@@ -173,14 +221,30 @@ class DataTypescript(BaseModel):
 
     key: str
 
-    created: Optional[str] = "2026-01-14T08:04:34.218Z"
+    created: Optional[str] = "2026-01-25T15:02:08.775Z"
 
-    updated: Optional[str] = "2026-01-14T08:04:34.218Z"
+    updated: Optional[str] = "2026-01-25T15:02:08.775Z"
 
     guardrail_config: Optional[
         GetEvalsDataEvalsResponse200ApplicationJSONGuardrailConfig
     ] = None
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["created", "updated", "guardrail_config"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyData62Type = Literal[
     "number",
@@ -216,6 +280,22 @@ class GetEvalsGuardrailConfigEvalsResponse200Number(BaseModel):
 
     alert_on_failure: Optional[bool] = False
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["alert_on_failure"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyData6Type = Literal[
     "boolean",
@@ -238,6 +318,22 @@ class GetEvalsGuardrailConfigEvalsResponse200Boolean(BaseModel):
 
     alert_on_failure: Optional[bool] = False
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["alert_on_failure"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 GetEvalsDataEvalsResponse200GuardrailConfigTypedDict = TypeAliasType(
     "GetEvalsDataEvalsResponse200GuardrailConfigTypedDict",
@@ -301,12 +397,28 @@ class DataRagas(BaseModel):
 
     model: str
 
-    created: Optional[str] = "2026-01-14T08:04:34.218Z"
+    created: Optional[str] = "2026-01-25T15:02:08.775Z"
 
-    updated: Optional[str] = "2026-01-14T08:04:34.218Z"
+    updated: Optional[str] = "2026-01-25T15:02:08.775Z"
 
     guardrail_config: Optional[GetEvalsDataEvalsResponse200GuardrailConfig] = None
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["created", "updated", "guardrail_config"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyData52Type = Literal[
     "number",
@@ -342,6 +454,22 @@ class GetEvalsGuardrailConfigEvalsResponseNumber(BaseModel):
 
     alert_on_failure: Optional[bool] = False
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["alert_on_failure"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyData5Type = Literal[
     "boolean",
@@ -364,6 +492,22 @@ class GetEvalsGuardrailConfigEvalsResponseBoolean(BaseModel):
 
     alert_on_failure: Optional[bool] = False
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["alert_on_failure"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 GetEvalsDataEvalsResponseGuardrailConfigTypedDict = TypeAliasType(
     "GetEvalsDataEvalsResponseGuardrailConfigTypedDict",
@@ -928,12 +1072,28 @@ class DataFunction(BaseModel):
 
     key: str
 
-    created: Optional[str] = "2026-01-14T08:04:34.218Z"
+    created: Optional[str] = "2026-01-25T15:02:08.775Z"
 
-    updated: Optional[str] = "2026-01-14T08:04:34.218Z"
+    updated: Optional[str] = "2026-01-25T15:02:08.775Z"
 
     guardrail_config: Optional[GetEvalsDataEvalsResponseGuardrailConfig] = None
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["created", "updated", "guardrail_config"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyDataType = Literal[
     "number",
@@ -969,6 +1129,22 @@ class GetEvalsGuardrailConfigEvalsNumber(BaseModel):
 
     alert_on_failure: Optional[bool] = False
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["alert_on_failure"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyType = Literal[
     "boolean",
@@ -991,6 +1167,22 @@ class GetEvalsGuardrailConfigEvalsBoolean(BaseModel):
 
     alert_on_failure: Optional[bool] = False
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["alert_on_failure"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 GetEvalsDataEvalsGuardrailConfigTypedDict = TypeAliasType(
     "GetEvalsDataEvalsGuardrailConfigTypedDict",
@@ -1035,12 +1227,28 @@ class DataPython(BaseModel):
 
     key: str
 
-    created: Optional[str] = "2026-01-14T08:04:34.218Z"
+    created: Optional[str] = "2026-01-25T15:02:08.775Z"
 
-    updated: Optional[str] = "2026-01-14T08:04:34.218Z"
+    updated: Optional[str] = "2026-01-25T15:02:08.775Z"
 
     guardrail_config: Optional[GetEvalsDataEvalsGuardrailConfig] = None
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["created", "updated", "guardrail_config"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONType = Literal["number",]
 
@@ -1074,6 +1282,22 @@ class GetEvalsGuardrailConfigNumber(BaseModel):
 
     alert_on_failure: Optional[bool] = False
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["alert_on_failure"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 GetEvalsGuardrailConfigEvalsResponse200Type = Literal["boolean",]
 
@@ -1094,6 +1318,22 @@ class GetEvalsGuardrailConfigBoolean(BaseModel):
 
     alert_on_failure: Optional[bool] = False
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["alert_on_failure"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 GetEvalsDataGuardrailConfigTypedDict = TypeAliasType(
     "GetEvalsDataGuardrailConfigTypedDict",
@@ -1152,12 +1392,28 @@ class DataHTTP(BaseModel):
 
     key: str
 
-    created: Optional[str] = "2026-01-14T08:04:34.218Z"
+    created: Optional[str] = "2026-01-25T15:02:08.775Z"
 
-    updated: Optional[str] = "2026-01-14T08:04:34.218Z"
+    updated: Optional[str] = "2026-01-25T15:02:08.775Z"
 
     guardrail_config: Optional[GetEvalsDataGuardrailConfig] = None
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["created", "updated", "guardrail_config"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 GetEvalsGuardrailConfigEvalsResponseType = Literal["number",]
 
@@ -1201,6 +1457,22 @@ class GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyDataNumb
 
     alert_on_failure: Optional[bool] = False
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["alert_on_failure"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 GetEvalsGuardrailConfigEvalsType = Literal["boolean",]
 
@@ -1225,6 +1497,62 @@ class GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyDataBool
 
     alert_on_failure: Optional[bool] = False
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["alert_on_failure"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
+
+class GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyNumberTypedDict(
+    TypedDict
+):
+    enabled: bool
+    type: GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyData1Type
+    value: float
+    operator: GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyOperator
+    alert_on_failure: NotRequired[bool]
+
+
+class GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyNumber(
+    BaseModel
+):
+    enabled: bool
+
+    type: GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyData1Type
+
+    value: float
+
+    operator: GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyOperator
+
+    alert_on_failure: Optional[bool] = False
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["alert_on_failure"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 DataGuardrailConfigTypedDict = TypeAliasType(
     "DataGuardrailConfigTypedDict",
@@ -1275,12 +1603,28 @@ class DataJSON(BaseModel):
 
     key: str
 
-    created: Optional[str] = "2026-01-14T08:04:34.218Z"
+    created: Optional[str] = "2026-01-25T15:02:08.775Z"
 
-    updated: Optional[str] = "2026-01-14T08:04:34.218Z"
+    updated: Optional[str] = "2026-01-25T15:02:08.775Z"
 
     guardrail_config: Optional[DataGuardrailConfig] = None
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["created", "updated", "guardrail_config"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyData1Type = Literal[
     "number",
@@ -1297,30 +1641,6 @@ GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyOperator = Lit
 ]
 
 
-class GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyNumberTypedDict(
-    TypedDict
-):
-    enabled: bool
-    type: GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyData1Type
-    value: float
-    operator: GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyOperator
-    alert_on_failure: NotRequired[bool]
-
-
-class GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyNumber(
-    BaseModel
-):
-    enabled: bool
-
-    type: GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyData1Type
-
-    value: float
-
-    operator: GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyOperator
-
-    alert_on_failure: Optional[bool] = False
-
-
 GetEvalsGuardrailConfigType = Literal["boolean",]
 
 
@@ -1344,6 +1664,22 @@ class GetEvalsGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyBoolean(
 
     alert_on_failure: Optional[bool] = False
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["alert_on_failure"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 GetEvalsDataEvalsResponse200ApplicationJSONResponseBodyGuardrailConfigTypedDict = TypeAliasType(
     "GetEvalsDataEvalsResponse200ApplicationJSONResponseBodyGuardrailConfigTypedDict",
@@ -1399,14 +1735,30 @@ class DataLLM(BaseModel):
 
     model: str
 
-    created: Optional[str] = "2026-01-14T08:04:34.218Z"
+    created: Optional[str] = "2026-01-25T15:02:08.775Z"
 
-    updated: Optional[str] = "2026-01-14T08:04:34.218Z"
+    updated: Optional[str] = "2026-01-25T15:02:08.775Z"
 
     guardrail_config: Optional[
         GetEvalsDataEvalsResponse200ApplicationJSONResponseBodyGuardrailConfig
     ] = None
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["created", "updated", "guardrail_config"])
+        serialized = handler(self)
+        m = {}
+
+        for n, f in type(self).model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+
+            if val != UNSET_SENTINEL:
+                if val is not None or k not in optional_fields:
+                    m[k] = val
+
+        return m
+
 
 GetEvalsDataTypedDict = TypeAliasType(
     "GetEvalsDataTypedDict",
@@ -1439,7 +1791,7 @@ GetEvalsData = Annotated[
 class GetEvalsResponseBodyTypedDict(TypedDict):
     r"""Returns a list of evals"""
 
-    object: GetEvalsObject
+    object: Object
     data: List[GetEvalsDataTypedDict]
     has_more: bool
 
@@ -1447,7 +1799,7 @@ class GetEvalsResponseBodyTypedDict(TypedDict):
 class GetEvalsResponseBody(BaseModel):
     r"""Returns a list of evals"""
 
-    object: GetEvalsObject
+    object: Object
 
     data: List[GetEvalsData]
 

orq_ai_sdk/models/getonechunkop.py

@@ -114,30 +114,25 @@ class GetOneChunkResponseBody(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["metadata", "created_by_id", "update_by_id"]
-        nullable_fields = ["created_by_id", "update_by_id"]
-        null_default_fields = []
-
+        optional_fields = set(["metadata", "created_by_id", "update_by_id"])
+        nullable_fields = set(["created_by_id", "update_by_id"])
         serialized = handler(self)
-
         m = {}
 
         for n, f in type(self).model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
-            serialized.pop(k, None)
-
-            optional_nullable = k in optional_fields and k in nullable_fields
-            is_set = (
-                self.__pydantic_fields_set__.intersection({n})
-                or k in null_default_fields
-            )  # pylint: disable=no-member
-
-            if val is not None and val != UNSET_SENTINEL:
-                m[k] = val
-            elif val != UNSET_SENTINEL and (
-                not k in optional_fields or (optional_nullable and is_set)
-            ):
-                m[k] = val
+            is_nullable_and_explicitly_set = (
+                k in nullable_fields
+                and (self.__pydantic_fields_set__.intersection({n}))  # pylint: disable=no-member
+            )
+
+            if val != UNSET_SENTINEL:
+                if (
+                    val is not None
+                    or k not in optional_fields
+                    or is_nullable_and_explicitly_set
+                ):
+                    m[k] = val
 
         return m