vellum-ai 1.1.0__py3-none-any.whl → 1.1.2__py3-none-any.whl

vellum/client/core/__init__.py

@@ -5,6 +5,7 @@ from .client_wrapper import AsyncClientWrapper, BaseClientWrapper, SyncClientWra
 from .datetime_utils import serialize_datetime
 from .file import File, convert_file_dict_to_httpx_tuples, with_content_type
 from .http_client import AsyncHttpClient, HttpClient
+from .http_response import AsyncHttpResponse, HttpResponse
 from .jsonable_encoder import jsonable_encoder
 from .pydantic_utilities import (
     IS_PYDANTIC_V2,
@@ -24,10 +25,12 @@ __all__ = [
     "ApiError",
     "AsyncClientWrapper",
     "AsyncHttpClient",
+    "AsyncHttpResponse",
     "BaseClientWrapper",
     "FieldMetadata",
     "File",
     "HttpClient",
+    "HttpResponse",
     "IS_PYDANTIC_V2",
     "RequestOptions",
     "SyncClientWrapper",

vellum/client/core/client_wrapper.py

@@ -25,10 +25,10 @@ class BaseClientWrapper:
 
     def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
-            "User-Agent": "vellum-ai/1.1.0",
+            "User-Agent": "vellum-ai/1.1.2",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "1.1.0",
+            "X-Fern-SDK-Version": "1.1.2",
         }
         if self._api_version is not None:
             headers["X-API-Version"] = self._api_version

vellum/client/core/http_response.py

@@ -0,0 +1,47 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from typing import Dict, Generic, TypeVar
+
+import httpx
+
+T = TypeVar("T")
+
+
+class HttpResponse(Generic[T]):
+    _response: httpx.Response
+    _data: T
+
+    def __init__(self, response: httpx.Response, data: T):
+        self._response = response
+        self._data = data
+
+    @property
+    def headers(self) -> Dict[str, str]:
+        return dict(self._response.headers)
+
+    @property
+    def data(self) -> T:
+        return self._data
+
+    def close(self) -> None:
+        self._response.close()
+
+
+class AsyncHttpResponse(Generic[T]):
+    _response: httpx.Response
+    _data: T
+
+    def __init__(self, response: httpx.Response, data: T):
+        self._response = response
+        self._data = data
+
+    @property
+    def headers(self) -> Dict[str, str]:
+        return dict(self._response.headers)
+
+    @property
+    def data(self) -> T:
+        return self._data
+
+    async def close(self) -> None:
+        await self._response.aclose()

vellum/client/core/pydantic_utilities.py

@@ -55,12 +55,19 @@ else:
 
 T = typing.TypeVar("T")
 Model = typing.TypeVar("Model", bound=pydantic.BaseModel)
+type_adapter_cache: typing.Dict[typing.Type, pydantic.TypeAdapter] = {}
+annotated_types = {"ExecuteApiResponse"}
 
 
-def parse_obj_as(type_: typing.Type[T], object_: typing.Any) -> T:
-    dealiased_object = convert_and_respect_annotation_metadata(object_=object_, annotation=type_, direction="read")
+def parse_obj_as(type_: typing.Type[T], object_: typing.Any, convert_metadata: bool = False) -> T:
+    if hasattr(type_, "__name__") and type_.__name__ in annotated_types:
+        dealiased_object = convert_and_respect_annotation_metadata(object_=object_, annotation=type_, direction="read")
+    else:
+        dealiased_object = object_
     if IS_PYDANTIC_V2:
-        adapter = pydantic.TypeAdapter(type_)  # type: ignore # Pydantic v2
+        adapter = pydantic.TypeAdapter(type_) if type_ not in type_adapter_cache else type_adapter_cache[type_]  # type: ignore # Pydantic v2
+        if type_ not in type_adapter_cache:
+            type_adapter_cache[type_] = adapter
         return adapter.validate_python(dealiased_object)
     else:
         return pydantic.parse_obj_as(type_, dealiased_object)