orq-ai-sdk 4.2.0rc28__py3-none-any.whl → 4.3.0rc7__py3-none-any.whl

orq_ai_sdk/models/retrievedatapointop.py

@@ -125,6 +125,22 @@ class RetrieveDatapointMessagesCacheControl(BaseModel):
     Defaults to `5m`. Only supported by `Anthropic` Claude models.
     """
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["ttl"])
+        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 RetrieveDatapointMessagesToolMessageTypedDict(TypedDict):
     role: RetrieveDatapointMessagesDatasetsResponse200ApplicationJSONRole
@@ -150,31 +166,26 @@ class RetrieveDatapointMessagesToolMessage(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["cache_control"]
-        nullable_fields = ["tool_call_id"]
-        null_default_fields = []
-
+        optional_fields = set(["cache_control"])
+        nullable_fields = set(["tool_call_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
 
@@ -251,6 +262,22 @@ class RetrieveDatapointMessagesFunction(BaseModel):
     arguments: Optional[str] = None
     r"""The arguments to call the function with, as generated by the model in JSON format. Note that the model does not always generate valid JSON, and may hallucinate parameters not defined by your function schema. Validate the arguments in your code before calling your function."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["name", "arguments"])
+        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 RetrieveDatapointMessagesToolCallsTypedDict(TypedDict):
     id: str
@@ -274,6 +301,22 @@ class RetrieveDatapointMessagesToolCalls(BaseModel):
     thought_signature: Optional[str] = None
     r"""Encrypted representation of the model internal reasoning state during function calling. Required by Gemini 3 models when continuing a conversation after a tool call."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["thought_signature"])
+        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 RetrieveDatapointMessagesAssistantMessageTypedDict(TypedDict):
     role: RetrieveDatapointMessagesDatasetsResponse200Role
@@ -315,31 +358,26 @@ class RetrieveDatapointMessagesAssistantMessage(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["content", "refusal", "name", "audio", "tool_calls"]
-        nullable_fields = ["content", "refusal", "audio"]
-        null_default_fields = []
-
+        optional_fields = set(["content", "refusal", "name", "audio", "tool_calls"])
+        nullable_fields = set(["content", "refusal", "audio"])
         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
 
@@ -395,6 +433,22 @@ class RetrieveDatapoint2CacheControl(BaseModel):
     Defaults to `5m`. Only supported by `Anthropic` Claude models.
     """
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["ttl"])
+        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 RetrieveDatapoint24TypedDict(TypedDict):
     type: RetrieveDatapoint2Type
@@ -413,6 +467,22 @@ class RetrieveDatapoint24(BaseModel):
 
     cache_control: Optional[RetrieveDatapoint2CacheControl] = None
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["cache_control"])
+        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
+
 
 RetrieveDatapointContent2TypedDict = TypeAliasType(
     "RetrieveDatapointContent2TypedDict",
@@ -469,6 +539,22 @@ class RetrieveDatapointMessagesUserMessage(BaseModel):
     name: Optional[str] = None
     r"""An optional name for the participant. Provides the model information to differentiate between participants of the same role."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["name"])
+        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
+
 
 RetrieveDatapointMessagesDatasetsRole = Literal["developer",]
 r"""The role of the messages author, in this case  `developer`."""
@@ -506,6 +592,22 @@ class RetrieveDatapointMessagesDeveloperMessage(BaseModel):
     name: Optional[str] = None
     r"""An optional name for the participant. Provides the model information to differentiate between participants of the same role."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["name"])
+        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
+
 
 RetrieveDatapointMessagesRole = Literal["system",]
 r"""The role of the messages author, in this case `system`."""
@@ -547,6 +649,22 @@ class RetrieveDatapointMessagesSystemMessage(BaseModel):
     name: Optional[str] = None
     r"""An optional name for the participant. Provides the model information to differentiate between participants of the same role."""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["name"])
+        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
+
 
 RetrieveDatapointMessagesTypedDict = TypeAliasType(
     "RetrieveDatapointMessagesTypedDict",
@@ -620,9 +738,25 @@ class RetrieveDatapointEvaluations3(BaseModel):
 
     source: Optional[RetrieveDatapointEvaluationsDatasetsResponseSource] = "orq"
 
-    reviewed_at: Optional[datetime] = parse_datetime("2026-01-14T08:04:45.088Z")
+    reviewed_at: Optional[datetime] = parse_datetime("2026-01-28T17:04:01.044Z")
     r"""The date and time the item was reviewed"""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["source", "reviewed_at"])
+        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
+
 
 RetrieveDatapointEvaluationsDatasetsEvaluationType = Literal["human_review",]
 r"""The type of evaluation"""
@@ -672,9 +806,25 @@ class RetrieveDatapointEvaluations2(BaseModel):
 
     source: Optional[RetrieveDatapointEvaluationsDatasetsSource] = "orq"
 
-    reviewed_at: Optional[datetime] = parse_datetime("2026-01-14T08:04:45.087Z")
+    reviewed_at: Optional[datetime] = parse_datetime("2026-01-28T17:04:01.043Z")
     r"""The date and time the item was reviewed"""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["source", "reviewed_at"])
+        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
+
 
 RetrieveDatapointEvaluationsEvaluationType = Literal["human_review",]
 r"""The type of evaluation"""
@@ -724,9 +874,25 @@ class RetrieveDatapointEvaluations1(BaseModel):
 
     source: Optional[RetrieveDatapointEvaluationsSource] = "orq"
 
-    reviewed_at: Optional[datetime] = parse_datetime("2026-01-14T08:04:45.086Z")
+    reviewed_at: Optional[datetime] = parse_datetime("2026-01-28T17:04:01.043Z")
     r"""The date and time the item was reviewed"""
 
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["source", "reviewed_at"])
+        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
+
 
 RetrieveDatapointEvaluationsTypedDict = TypeAliasType(
     "RetrieveDatapointEvaluationsTypedDict",
@@ -811,5 +977,33 @@ class RetrieveDatapointResponseBody(BaseModel):
     created: Optional[datetime] = None
     r"""The date and time the resource was created"""
 
-    updated: Optional[datetime] = parse_datetime("2026-01-14T08:04:31.906Z")
+    updated: Optional[datetime] = parse_datetime("2026-01-28T17:03:47.548Z")
     r"""The date and time the resource was last updated"""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(
+            [
+                "inputs",
+                "messages",
+                "expected_output",
+                "evaluations",
+                "snapshot_version",
+                "created_by_id",
+                "updated_by_id",
+                "created",
+                "updated",
+            ]
+        )
+        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

orq_ai_sdk/models/retrievedatasetop.py

@@ -2,9 +2,10 @@
 
 from __future__ import annotations
 from datetime import datetime
-from orq_ai_sdk.types import BaseModel
+from orq_ai_sdk.types import BaseModel, UNSET_SENTINEL
 from orq_ai_sdk.utils import FieldMetadata, PathParamMetadata, parse_datetime
 import pydantic
+from pydantic import model_serializer
 from typing import Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
@@ -80,5 +81,21 @@ class RetrieveDatasetResponseBody(BaseModel):
     created: Optional[datetime] = None
     r"""The date and time the resource was created"""
 
-    updated: Optional[datetime] = parse_datetime("2026-01-14T08:04:31.906Z")
+    updated: Optional[datetime] = parse_datetime("2026-01-28T17:03:47.548Z")
     r"""The date and time the resource was last updated"""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["created_by_id", "updated_by_id", "created", "updated"])
+        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

orq_ai_sdk/models/retrievedatasourceop.py

@@ -90,7 +90,7 @@ class RetrieveDatasourceResponseBody(BaseModel):
     r"""The number of chunks in the datasource"""
 
     id: Annotated[Optional[str], pydantic.Field(alias="_id")] = (
-        "01KEXRJ89TABD51P50EYVSJ8CW"
+        "01KG2RZQXV7AW8K26RHSVSGPJ2"
     )
     r"""The unique identifier of the data source"""
 
@@ -108,36 +108,27 @@ class RetrieveDatasourceResponseBody(BaseModel):
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = [
-            "_id",
-            "description",
-            "file_id",
-            "created_by_id",
-            "update_by_id",
-        ]
-        nullable_fields = ["file_id", "created_by_id", "update_by_id"]
-        null_default_fields = []
-
+        optional_fields = set(
+            ["_id", "description", "file_id", "created_by_id", "update_by_id"]
+        )
+        nullable_fields = set(["file_id", "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

orq_ai_sdk/models/{retrievecontactop.py → retrieveidentityop.py}

@@ -19,32 +19,32 @@ from typing import Any, Dict, List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-class RetrieveContactRequestTypedDict(TypedDict):
+class RetrieveIdentityRequestTypedDict(TypedDict):
     id: str
-    r"""Unique contact id or external id"""
+    r"""Unique identity id or external id"""
 
 
-class RetrieveContactRequest(BaseModel):
+class RetrieveIdentityRequest(BaseModel):
     id: Annotated[
         str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
     ]
-    r"""Unique contact id or external id"""
+    r"""Unique identity id or external id"""
 
 
-class RetrieveContactContactsResponseBodyData(BaseModel):
+class RetrieveIdentityIdentitiesResponseBodyData(BaseModel):
     error: str
     r"""Error message"""
 
 
 @dataclass(unsafe_hash=True)
-class RetrieveContactContactsResponseBody(OrqError):
-    r"""Contact not found"""
+class RetrieveIdentityIdentitiesResponseBody(OrqError):
+    r"""Identity not found"""
 
-    data: RetrieveContactContactsResponseBodyData = field(hash=False)
+    data: RetrieveIdentityIdentitiesResponseBodyData = field(hash=False)
 
     def __init__(
         self,
-        data: RetrieveContactContactsResponseBodyData,
+        data: RetrieveIdentityIdentitiesResponseBodyData,
         raw_response: httpx.Response,
         body: Optional[str] = None,
     ):
@@ -53,8 +53,8 @@ class RetrieveContactContactsResponseBody(OrqError):
         object.__setattr__(self, "data", data)
 
 
-class RetrieveContactResponseBodyTypedDict(TypedDict):
-    r"""Contact details"""
+class RetrieveIdentityResponseBodyTypedDict(TypedDict):
+    r"""Identity details"""
 
     id: str
     r"""Unique ULID (Universally Unique Lexicographically Sortable Identifier) for the contact"""
@@ -76,8 +76,8 @@ class RetrieveContactResponseBodyTypedDict(TypedDict):
     r"""The date and time the resource was last updated"""
 
 
-class RetrieveContactResponseBody(BaseModel):
-    r"""Contact details"""
+class RetrieveIdentityResponseBody(BaseModel):
+    r"""Identity details"""
 
     id: Annotated[str, pydantic.Field(alias="_id")]
     r"""Unique ULID (Universally Unique Lexicographically Sortable Identifier) for the contact"""
@@ -103,43 +103,40 @@ class RetrieveContactResponseBody(BaseModel):
     created: Optional[datetime] = None
     r"""The date and time the resource was created"""
 
-    updated: Optional[datetime] = parse_datetime("2026-01-14T08:04:31.906Z")
+    updated: Optional[datetime] = parse_datetime("2026-01-28T17:03:47.548Z")
     r"""The date and time the resource was last updated"""
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = [
-            "display_name",
-            "email",
-            "avatar_url",
-            "tags",
-            "metadata",
-            "created",
-            "updated",
-        ]
-        nullable_fields = ["display_name", "email", "avatar_url"]
-        null_default_fields = []
-
+        optional_fields = set(
+            [
+                "display_name",
+                "email",
+                "avatar_url",
+                "tags",
+                "metadata",
+                "created",
+                "updated",
+            ]
+        )
+        nullable_fields = set(["display_name", "email", "avatar_url"])
         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

orq_ai_sdk/models/retrievememorydocumentop.py

@@ -1,9 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from orq_ai_sdk.types import BaseModel
+from orq_ai_sdk.types import BaseModel, UNSET_SENTINEL
 from orq_ai_sdk.utils import FieldMetadata, PathParamMetadata
 import pydantic
+from pydantic import model_serializer
 from typing import Dict, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
@@ -75,3 +76,19 @@ class RetrieveMemoryDocumentResponseBody(BaseModel):
 
     metadata: Optional[Dict[str, str]] = None
     r"""Flexible key-value pairs for custom filtering and categorization. Clients can add arbitrary string metadata to enable future filtering of memory documents based on their specific needs (e.g., document type, source, topic, relevance score, or any custom taxonomy)."""
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = set(["created_by_id", "updated_by_id", "metadata"])
+        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