letta-client 1.0.0a4__py3-none-any.whl → 1.0.0a6__py3-none-any.whl

letta_client/core/pydantic_utilities.py

@@ -1,255 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-# nopycln: file
-import datetime as dt
-from collections import defaultdict
-from typing import Any, Callable, ClassVar, Dict, List, Mapping, Optional, Set, Tuple, Type, TypeVar, Union, cast
-
-import pydantic
-
-IS_PYDANTIC_V2 = pydantic.VERSION.startswith("2.")
-
-if IS_PYDANTIC_V2:
-    from pydantic.v1.datetime_parse import parse_date as parse_date
-    from pydantic.v1.datetime_parse import parse_datetime as parse_datetime
-    from pydantic.v1.fields import ModelField as ModelField
-    from pydantic.v1.json import ENCODERS_BY_TYPE as encoders_by_type  # type: ignore[attr-defined]
-    from pydantic.v1.typing import get_args as get_args
-    from pydantic.v1.typing import get_origin as get_origin
-    from pydantic.v1.typing import is_literal_type as is_literal_type
-    from pydantic.v1.typing import is_union as is_union
-else:
-    from pydantic.datetime_parse import parse_date as parse_date  # type: ignore[no-redef]
-    from pydantic.datetime_parse import parse_datetime as parse_datetime  # type: ignore[no-redef]
-    from pydantic.fields import ModelField as ModelField  # type: ignore[attr-defined, no-redef]
-    from pydantic.json import ENCODERS_BY_TYPE as encoders_by_type  # type: ignore[no-redef]
-    from pydantic.typing import get_args as get_args  # type: ignore[no-redef]
-    from pydantic.typing import get_origin as get_origin  # type: ignore[no-redef]
-    from pydantic.typing import is_literal_type as is_literal_type  # type: ignore[no-redef]
-    from pydantic.typing import is_union as is_union  # type: ignore[no-redef]
-
-from .datetime_utils import serialize_datetime
-from .serialization import convert_and_respect_annotation_metadata
-from typing_extensions import TypeAlias
-
-T = TypeVar("T")
-Model = TypeVar("Model", bound=pydantic.BaseModel)
-
-
-def parse_obj_as(type_: Type[T], object_: Any) -> T:
-    dealiased_object = convert_and_respect_annotation_metadata(object_=object_, annotation=type_, direction="read")
-    if IS_PYDANTIC_V2:
-        adapter = pydantic.TypeAdapter(type_)  # type: ignore[attr-defined]
-        return adapter.validate_python(dealiased_object)
-    return pydantic.parse_obj_as(type_, dealiased_object)
-
-
-def to_jsonable_with_fallback(obj: Any, fallback_serializer: Callable[[Any], Any]) -> Any:
-    if IS_PYDANTIC_V2:
-        from pydantic_core import to_jsonable_python
-
-        return to_jsonable_python(obj, fallback=fallback_serializer)
-    return fallback_serializer(obj)
-
-
-class UniversalBaseModel(pydantic.BaseModel):
-    if IS_PYDANTIC_V2:
-        model_config: ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(  # type: ignore[typeddict-unknown-key]
-            # Allow fields beginning with `model_` to be used in the model
-            protected_namespaces=(),
-        )
-
-        @pydantic.model_serializer(mode="plain", when_used="json")  # type: ignore[attr-defined]
-        def serialize_model(self) -> Any:  # type: ignore[name-defined]
-            serialized = self.model_dump()
-            data = {k: serialize_datetime(v) if isinstance(v, dt.datetime) else v for k, v in serialized.items()}
-            return data
-
-    else:
-
-        class Config:
-            smart_union = True
-            json_encoders = {dt.datetime: serialize_datetime}
-
-    @classmethod
-    def model_construct(cls: Type["Model"], _fields_set: Optional[Set[str]] = None, **values: Any) -> "Model":
-        dealiased_object = convert_and_respect_annotation_metadata(object_=values, annotation=cls, direction="read")
-        return cls.construct(_fields_set, **dealiased_object)
-
-    @classmethod
-    def construct(cls: Type["Model"], _fields_set: Optional[Set[str]] = None, **values: Any) -> "Model":
-        dealiased_object = convert_and_respect_annotation_metadata(object_=values, annotation=cls, direction="read")
-        if IS_PYDANTIC_V2:
-            return super().model_construct(_fields_set, **dealiased_object)  # type: ignore[misc]
-        return super().construct(_fields_set, **dealiased_object)
-
-    def json(self, **kwargs: Any) -> str:
-        kwargs_with_defaults = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
-        if IS_PYDANTIC_V2:
-            return super().model_dump_json(**kwargs_with_defaults)  # type: ignore[misc]
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: Any) -> Dict[str, Any]:
-        """
-        Override the default dict method to `exclude_unset` by default. This function patches
-        `exclude_unset` to work include fields within non-None default values.
-        """
-        # Note: the logic here is multiplexed given the levers exposed in Pydantic V1 vs V2
-        # Pydantic V1's .dict can be extremely slow, so we do not want to call it twice.
-        #
-        # We'd ideally do the same for Pydantic V2, but it shells out to a library to serialize models
-        # that we have less control over, and this is less intrusive than custom serializers for now.
-        if IS_PYDANTIC_V2:
-            kwargs_with_defaults_exclude_unset = {
-                **kwargs,
-                "by_alias": True,
-                "exclude_unset": True,
-                "exclude_none": False,
-            }
-            kwargs_with_defaults_exclude_none = {
-                **kwargs,
-                "by_alias": True,
-                "exclude_none": True,
-                "exclude_unset": False,
-            }
-            dict_dump = deep_union_pydantic_dicts(
-                super().model_dump(**kwargs_with_defaults_exclude_unset),  # type: ignore[misc]
-                super().model_dump(**kwargs_with_defaults_exclude_none),  # type: ignore[misc]
-            )
-
-        else:
-            _fields_set = self.__fields_set__.copy()
-
-            fields = _get_model_fields(self.__class__)
-            for name, field in fields.items():
-                if name not in _fields_set:
-                    default = _get_field_default(field)
-
-                    # If the default values are non-null act like they've been set
-                    # This effectively allows exclude_unset to work like exclude_none where
-                    # the latter passes through intentionally set none values.
-                    if default is not None or ("exclude_unset" in kwargs and not kwargs["exclude_unset"]):
-                        _fields_set.add(name)
-
-                        if default is not None:
-                            self.__fields_set__.add(name)
-
-            kwargs_with_defaults_exclude_unset_include_fields = {
-                "by_alias": True,
-                "exclude_unset": True,
-                "include": _fields_set,
-                **kwargs,
-            }
-
-            dict_dump = super().dict(**kwargs_with_defaults_exclude_unset_include_fields)
-
-        return convert_and_respect_annotation_metadata(object_=dict_dump, annotation=self.__class__, direction="write")
-
-
-def _union_list_of_pydantic_dicts(source: List[Any], destination: List[Any]) -> List[Any]:
-    converted_list: List[Any] = []
-    for i, item in enumerate(source):
-        destination_value = destination[i]
-        if isinstance(item, dict):
-            converted_list.append(deep_union_pydantic_dicts(item, destination_value))
-        elif isinstance(item, list):
-            converted_list.append(_union_list_of_pydantic_dicts(item, destination_value))
-        else:
-            converted_list.append(item)
-    return converted_list
-
-
-def deep_union_pydantic_dicts(source: Dict[str, Any], destination: Dict[str, Any]) -> Dict[str, Any]:
-    for key, value in source.items():
-        node = destination.setdefault(key, {})
-        if isinstance(value, dict):
-            deep_union_pydantic_dicts(value, node)
-        # Note: we do not do this same processing for sets given we do not have sets of models
-        # and given the sets are unordered, the processing of the set and matching objects would
-        # be non-trivial.
-        elif isinstance(value, list):
-            destination[key] = _union_list_of_pydantic_dicts(value, node)
-        else:
-            destination[key] = value
-
-    return destination
-
-
-if IS_PYDANTIC_V2:
-
-    class V2RootModel(UniversalBaseModel, pydantic.RootModel):  # type: ignore[misc, name-defined, type-arg]
-        pass
-
-    UniversalRootModel: TypeAlias = V2RootModel  # type: ignore[misc]
-else:
-    UniversalRootModel: TypeAlias = UniversalBaseModel  # type: ignore[misc, no-redef]
-
-
-def encode_by_type(o: Any) -> Any:
-    encoders_by_class_tuples: Dict[Callable[[Any], Any], Tuple[Any, ...]] = defaultdict(tuple)
-    for type_, encoder in encoders_by_type.items():
-        encoders_by_class_tuples[encoder] += (type_,)
-
-    if type(o) in encoders_by_type:
-        return encoders_by_type[type(o)](o)
-    for encoder, classes_tuple in encoders_by_class_tuples.items():
-        if isinstance(o, classes_tuple):
-            return encoder(o)
-
-
-def update_forward_refs(model: Type["Model"], **localns: Any) -> None:
-    if IS_PYDANTIC_V2:
-        model.model_rebuild(raise_errors=False)  # type: ignore[attr-defined]
-    else:
-        model.update_forward_refs(**localns)
-
-
-# Mirrors Pydantic's internal typing
-AnyCallable = Callable[..., Any]
-
-
-def universal_root_validator(
-    pre: bool = False,
-) -> Callable[[AnyCallable], AnyCallable]:
-    def decorator(func: AnyCallable) -> AnyCallable:
-        if IS_PYDANTIC_V2:
-            return cast(AnyCallable, pydantic.model_validator(mode="before" if pre else "after")(func))  # type: ignore[attr-defined]
-        return cast(AnyCallable, pydantic.root_validator(pre=pre)(func))  # type: ignore[call-overload]
-
-    return decorator
-
-
-def universal_field_validator(field_name: str, pre: bool = False) -> Callable[[AnyCallable], AnyCallable]:
-    def decorator(func: AnyCallable) -> AnyCallable:
-        if IS_PYDANTIC_V2:
-            return cast(AnyCallable, pydantic.field_validator(field_name, mode="before" if pre else "after")(func))  # type: ignore[attr-defined]
-        return cast(AnyCallable, pydantic.validator(field_name, pre=pre)(func))
-
-    return decorator
-
-
-PydanticField = Union[ModelField, pydantic.fields.FieldInfo]
-
-
-def _get_model_fields(model: Type["Model"]) -> Mapping[str, PydanticField]:
-    if IS_PYDANTIC_V2:
-        return cast(Mapping[str, PydanticField], model.model_fields)  # type: ignore[attr-defined]
-    return cast(Mapping[str, PydanticField], model.__fields__)
-
-
-def _get_field_default(field: PydanticField) -> Any:
-    try:
-        value = field.get_default()  # type: ignore[union-attr]
-    except:
-        value = field.default
-    if IS_PYDANTIC_V2:
-        from pydantic_core import PydanticUndefined
-
-        if value == PydanticUndefined:
-            return None
-        return value
-    return value

letta_client/core/query_encoder.py

@@ -1,58 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from typing import Any, Dict, List, Optional, Tuple
-
-import pydantic
-
-
-# Flattens dicts to be of the form {"key[subkey][subkey2]": value} where value is not a dict
-def traverse_query_dict(dict_flat: Dict[str, Any], key_prefix: Optional[str] = None) -> List[Tuple[str, Any]]:
-    result = []
-    for k, v in dict_flat.items():
-        key = f"{key_prefix}[{k}]" if key_prefix is not None else k
-        if isinstance(v, dict):
-            result.extend(traverse_query_dict(v, key))
-        elif isinstance(v, list):
-            for arr_v in v:
-                if isinstance(arr_v, dict):
-                    result.extend(traverse_query_dict(arr_v, key))
-                else:
-                    result.append((key, arr_v))
-        else:
-            result.append((key, v))
-    return result
-
-
-def single_query_encoder(query_key: str, query_value: Any) -> List[Tuple[str, Any]]:
-    if isinstance(query_value, pydantic.BaseModel) or isinstance(query_value, dict):
-        if isinstance(query_value, pydantic.BaseModel):
-            obj_dict = query_value.dict(by_alias=True)
-        else:
-            obj_dict = query_value
-        return traverse_query_dict(obj_dict, query_key)
-    elif isinstance(query_value, list):
-        encoded_values: List[Tuple[str, Any]] = []
-        for value in query_value:
-            if isinstance(value, pydantic.BaseModel) or isinstance(value, dict):
-                if isinstance(value, pydantic.BaseModel):
-                    obj_dict = value.dict(by_alias=True)
-                elif isinstance(value, dict):
-                    obj_dict = value
-
-                encoded_values.extend(single_query_encoder(query_key, obj_dict))
-            else:
-                encoded_values.append((query_key, value))
-
-        return encoded_values
-
-    return [(query_key, query_value)]
-
-
-def encode_query(query: Optional[Dict[str, Any]]) -> Optional[List[Tuple[str, Any]]]:
-    if query is None:
-        return None
-
-    encoded_query = []
-    for k, v in query.items():
-        encoded_query.extend(single_query_encoder(k, v))
-    return encoded_query

letta_client/core/remove_none_from_dict.py

@@ -1,11 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from typing import Any, Dict, Mapping, Optional
-
-
-def remove_none_from_dict(original: Mapping[str, Optional[Any]]) -> Dict[str, Any]:
-    new: Dict[str, Any] = {}
-    for key, value in original.items():
-        if value is not None:
-            new[key] = value
-    return new

letta_client/core/request_options.py

@@ -1,35 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-
-try:
-    from typing import NotRequired  # type: ignore
-except ImportError:
-    from typing_extensions import NotRequired
-
-
-class RequestOptions(typing.TypedDict, total=False):
-    """
-    Additional options for request-specific configuration when calling APIs via the SDK.
-    This is used primarily as an optional final parameter for service functions.
-
-    Attributes:
-        - timeout_in_seconds: int. The number of seconds to await an API call before timing out.
-
-        - max_retries: int. The max number of retries to attempt if the API call fails.
-
-        - additional_headers: typing.Dict[str, typing.Any]. A dictionary containing additional parameters to spread into the request's header dict
-
-        - additional_query_parameters: typing.Dict[str, typing.Any]. A dictionary containing additional parameters to spread into the request's query parameters dict
-
-        - additional_body_parameters: typing.Dict[str, typing.Any]. A dictionary containing additional parameters to spread into the request's body parameters dict
-
-        - chunk_size: int. The size, in bytes, to process each chunk of data being streamed back within the response. This equates to leveraging `chunk_size` within `requests` or `httpx`, and is only leveraged for file downloads.
-    """
-
-    timeout_in_seconds: NotRequired[int]
-    max_retries: NotRequired[int]
-    additional_headers: NotRequired[typing.Dict[str, typing.Any]]
-    additional_query_parameters: NotRequired[typing.Dict[str, typing.Any]]
-    additional_body_parameters: NotRequired[typing.Dict[str, typing.Any]]
-    chunk_size: NotRequired[int]

letta_client/core/serialization.py

@@ -1,276 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import collections
-import inspect
-import typing
-
-import pydantic
-import typing_extensions
-
-
-class FieldMetadata:
-    """
-    Metadata class used to annotate fields to provide additional information.
-
-    Example:
-    class MyDict(TypedDict):
-        field: typing.Annotated[str, FieldMetadata(alias="field_name")]
-
-    Will serialize: `{"field": "value"}`
-    To: `{"field_name": "value"}`
-    """
-
-    alias: str
-
-    def __init__(self, *, alias: str) -> None:
-        self.alias = alias
-
-
-def convert_and_respect_annotation_metadata(
-    *,
-    object_: typing.Any,
-    annotation: typing.Any,
-    inner_type: typing.Optional[typing.Any] = None,
-    direction: typing.Literal["read", "write"],
-) -> typing.Any:
-    """
-    Respect the metadata annotations on a field, such as aliasing. This function effectively
-    manipulates the dict-form of an object to respect the metadata annotations. This is primarily used for
-    TypedDicts, which cannot support aliasing out of the box, and can be extended for additional
-    utilities, such as defaults.
-
-    Parameters
-    ----------
-    object_ : typing.Any
-
-    annotation : type
-        The type we're looking to apply typing annotations from
-
-    inner_type : typing.Optional[type]
-
-    Returns
-    -------
-    typing.Any
-    """
-
-    if object_ is None:
-        return None
-    if inner_type is None:
-        inner_type = annotation
-
-    clean_type = _remove_annotations(inner_type)
-    # Pydantic models
-    if (
-        inspect.isclass(clean_type)
-        and issubclass(clean_type, pydantic.BaseModel)
-        and isinstance(object_, typing.Mapping)
-    ):
-        return _convert_mapping(object_, clean_type, direction)
-    # TypedDicts
-    if typing_extensions.is_typeddict(clean_type) and isinstance(object_, typing.Mapping):
-        return _convert_mapping(object_, clean_type, direction)
-
-    if (
-        typing_extensions.get_origin(clean_type) == typing.Dict
-        or typing_extensions.get_origin(clean_type) == dict
-        or clean_type == typing.Dict
-    ) and isinstance(object_, typing.Dict):
-        key_type = typing_extensions.get_args(clean_type)[0]
-        value_type = typing_extensions.get_args(clean_type)[1]
-
-        return {
-            key: convert_and_respect_annotation_metadata(
-                object_=value,
-                annotation=annotation,
-                inner_type=value_type,
-                direction=direction,
-            )
-            for key, value in object_.items()
-        }
-
-    # If you're iterating on a string, do not bother to coerce it to a sequence.
-    if not isinstance(object_, str):
-        if (
-            typing_extensions.get_origin(clean_type) == typing.Set
-            or typing_extensions.get_origin(clean_type) == set
-            or clean_type == typing.Set
-        ) and isinstance(object_, typing.Set):
-            inner_type = typing_extensions.get_args(clean_type)[0]
-            return {
-                convert_and_respect_annotation_metadata(
-                    object_=item,
-                    annotation=annotation,
-                    inner_type=inner_type,
-                    direction=direction,
-                )
-                for item in object_
-            }
-        elif (
-            (
-                typing_extensions.get_origin(clean_type) == typing.List
-                or typing_extensions.get_origin(clean_type) == list
-                or clean_type == typing.List
-            )
-            and isinstance(object_, typing.List)
-        ) or (
-            (
-                typing_extensions.get_origin(clean_type) == typing.Sequence
-                or typing_extensions.get_origin(clean_type) == collections.abc.Sequence
-                or clean_type == typing.Sequence
-            )
-            and isinstance(object_, typing.Sequence)
-        ):
-            inner_type = typing_extensions.get_args(clean_type)[0]
-            return [
-                convert_and_respect_annotation_metadata(
-                    object_=item,
-                    annotation=annotation,
-                    inner_type=inner_type,
-                    direction=direction,
-                )
-                for item in object_
-            ]
-
-    if typing_extensions.get_origin(clean_type) == typing.Union:
-        # We should be able to ~relatively~ safely try to convert keys against all
-        # member types in the union, the edge case here is if one member aliases a field
-        # of the same name to a different name from another member
-        # Or if another member aliases a field of the same name that another member does not.
-        for member in typing_extensions.get_args(clean_type):
-            object_ = convert_and_respect_annotation_metadata(
-                object_=object_,
-                annotation=annotation,
-                inner_type=member,
-                direction=direction,
-            )
-        return object_
-
-    annotated_type = _get_annotation(annotation)
-    if annotated_type is None:
-        return object_
-
-    # If the object is not a TypedDict, a Union, or other container (list, set, sequence, etc.)
-    # Then we can safely call it on the recursive conversion.
-    return object_
-
-
-def _convert_mapping(
-    object_: typing.Mapping[str, object],
-    expected_type: typing.Any,
-    direction: typing.Literal["read", "write"],
-) -> typing.Mapping[str, object]:
-    converted_object: typing.Dict[str, object] = {}
-    try:
-        annotations = typing_extensions.get_type_hints(expected_type, include_extras=True)
-    except NameError:
-        # The TypedDict contains a circular reference, so
-        # we use the __annotations__ attribute directly.
-        annotations = getattr(expected_type, "__annotations__", {})
-    aliases_to_field_names = _get_alias_to_field_name(annotations)
-    for key, value in object_.items():
-        if direction == "read" and key in aliases_to_field_names:
-            dealiased_key = aliases_to_field_names.get(key)
-            if dealiased_key is not None:
-                type_ = annotations.get(dealiased_key)
-        else:
-            type_ = annotations.get(key)
-        # Note you can't get the annotation by the field name if you're in read mode, so you must check the aliases map
-        #
-        # So this is effectively saying if we're in write mode, and we don't have a type, or if we're in read mode and we don't have an alias
-        # then we can just pass the value through as is
-        if type_ is None:
-            converted_object[key] = value
-        elif direction == "read" and key not in aliases_to_field_names:
-            converted_object[key] = convert_and_respect_annotation_metadata(
-                object_=value, annotation=type_, direction=direction
-            )
-        else:
-            converted_object[_alias_key(key, type_, direction, aliases_to_field_names)] = (
-                convert_and_respect_annotation_metadata(object_=value, annotation=type_, direction=direction)
-            )
-    return converted_object
-
-
-def _get_annotation(type_: typing.Any) -> typing.Optional[typing.Any]:
-    maybe_annotated_type = typing_extensions.get_origin(type_)
-    if maybe_annotated_type is None:
-        return None
-
-    if maybe_annotated_type == typing_extensions.NotRequired:
-        type_ = typing_extensions.get_args(type_)[0]
-        maybe_annotated_type = typing_extensions.get_origin(type_)
-
-    if maybe_annotated_type == typing_extensions.Annotated:
-        return type_
-
-    return None
-
-
-def _remove_annotations(type_: typing.Any) -> typing.Any:
-    maybe_annotated_type = typing_extensions.get_origin(type_)
-    if maybe_annotated_type is None:
-        return type_
-
-    if maybe_annotated_type == typing_extensions.NotRequired:
-        return _remove_annotations(typing_extensions.get_args(type_)[0])
-
-    if maybe_annotated_type == typing_extensions.Annotated:
-        return _remove_annotations(typing_extensions.get_args(type_)[0])
-
-    return type_
-
-
-def get_alias_to_field_mapping(type_: typing.Any) -> typing.Dict[str, str]:
-    annotations = typing_extensions.get_type_hints(type_, include_extras=True)
-    return _get_alias_to_field_name(annotations)
-
-
-def get_field_to_alias_mapping(type_: typing.Any) -> typing.Dict[str, str]:
-    annotations = typing_extensions.get_type_hints(type_, include_extras=True)
-    return _get_field_to_alias_name(annotations)
-
-
-def _get_alias_to_field_name(
-    field_to_hint: typing.Dict[str, typing.Any],
-) -> typing.Dict[str, str]:
-    aliases = {}
-    for field, hint in field_to_hint.items():
-        maybe_alias = _get_alias_from_type(hint)
-        if maybe_alias is not None:
-            aliases[maybe_alias] = field
-    return aliases
-
-
-def _get_field_to_alias_name(
-    field_to_hint: typing.Dict[str, typing.Any],
-) -> typing.Dict[str, str]:
-    aliases = {}
-    for field, hint in field_to_hint.items():
-        maybe_alias = _get_alias_from_type(hint)
-        if maybe_alias is not None:
-            aliases[field] = maybe_alias
-    return aliases
-
-
-def _get_alias_from_type(type_: typing.Any) -> typing.Optional[str]:
-    maybe_annotated_type = _get_annotation(type_)
-
-    if maybe_annotated_type is not None:
-        # The actual annotations are 1 onward, the first is the annotated type
-        annotations = typing_extensions.get_args(maybe_annotated_type)[1:]
-
-        for annotation in annotations:
-            if isinstance(annotation, FieldMetadata) and annotation.alias is not None:
-                return annotation.alias
-    return None
-
-
-def _alias_key(
-    key: str,
-    type_: typing.Any,
-    direction: typing.Literal["read", "write"],
-    aliases_to_field_names: typing.Dict[str, str],
-) -> str:
-    if direction == "read":
-        return aliases_to_field_names.get(key, key)
-    return _get_alias_from_type(type_=type_) or key