vellum-ai 0.0.30__tar.gz → 0.0.31__tar.gz

{vellum_ai-0.0.30 → vellum_ai-0.0.31}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.0.30
+Version: 0.0.31
 Summary: 
 Requires-Python: >=3.7,<4.0
 Classifier: Programming Language :: Python :: 3

{vellum_ai-0.0.30 → vellum_ai-0.0.31}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "vellum-ai"
-version = "v0.0.30"
+version = "v0.0.31"
 description = ""
 readme = "README.md"
 authors = []

{vellum_ai-0.0.30 → vellum_ai-0.0.31}/src/vellum/client.py

@@ -109,12 +109,14 @@ class Vellum:
                         continue
                     yield pydantic.parse_obj_as(WorkflowStreamEvent, json.loads(_text))  # type: ignore
                 return
+            _response.read()
+            if _response.status_code == 400:
+                raise BadRequestError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
             if _response.status_code == 404:
                 raise NotFoundError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
             if _response.status_code == 500:
                 raise InternalServerError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
             try:
-                _response.read()
                 _response_json = _response.json()
             except JSONDecodeError:
                 raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -218,6 +220,7 @@ class Vellum:
                         continue
                     yield pydantic.parse_obj_as(GenerateStreamResponse, json.loads(_text))  # type: ignore
                 return
+            _response.read()
             if _response.status_code == 400:
                 raise BadRequestError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
             if _response.status_code == 403:
@@ -227,7 +230,6 @@ class Vellum:
             if _response.status_code == 500:
                 raise InternalServerError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
             try:
-                _response.read()
                 _response_json = _response.json()
             except JSONDecodeError:
                 raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -405,12 +407,14 @@ class AsyncVellum:
                         continue
                     yield pydantic.parse_obj_as(WorkflowStreamEvent, json.loads(_text))  # type: ignore
                 return
+            await _response.aread()
+            if _response.status_code == 400:
+                raise BadRequestError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
             if _response.status_code == 404:
                 raise NotFoundError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
             if _response.status_code == 500:
                 raise InternalServerError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
             try:
-                await _response.aread()
                 _response_json = _response.json()
             except JSONDecodeError:
                 raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -514,6 +518,7 @@ class AsyncVellum:
                         continue
                     yield pydantic.parse_obj_as(GenerateStreamResponse, json.loads(_text))  # type: ignore
                 return
+            await _response.aread()
             if _response.status_code == 400:
                 raise BadRequestError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
             if _response.status_code == 403:
@@ -523,7 +528,6 @@ class AsyncVellum:
             if _response.status_code == 500:
                 raise InternalServerError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
             try:
-                await _response.aread()
                 _response_json = _response.json()
             except JSONDecodeError:
                 raise ApiError(status_code=_response.status_code, body=_response.text)

{vellum_ai-0.0.30 → vellum_ai-0.0.31}/src/vellum/core/client_wrapper.py

@@ -10,7 +10,11 @@ class BaseClientWrapper:
         self.api_key = api_key
 
     def get_headers(self) -> typing.Dict[str, str]:
-        headers: typing.Dict[str, str] = {}
+        headers: typing.Dict[str, str] = {
+            "X-Fern-Language": "Python",
+            "X-Fern-SDK-Name": "vellum-ai",
+            "X-Fern-SDK-Version": "v0.0.31",
+        }
         headers["X_API_KEY"] = self.api_key
         return headers
 

{vellum_ai-0.0.30 → vellum_ai-0.0.31}/src/vellum/types/terminal_node_chat_history_result.py

@@ -11,7 +11,7 @@ from .chat_message import ChatMessage
 
 class TerminalNodeChatHistoryResult(pydantic.BaseModel):
     name: str = pydantic.Field(description="The unique name given to the terminal node that produced this output.")
-    value: typing.List[ChatMessage]
+    value: typing.Optional[typing.List[ChatMessage]]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

{vellum_ai-0.0.30 → vellum_ai-0.0.31}/src/vellum/types/terminal_node_json_result.py

@@ -10,7 +10,7 @@ from ..core.datetime_utils import serialize_datetime
 
 class TerminalNodeJsonResult(pydantic.BaseModel):
     name: str = pydantic.Field(description="The unique name given to the terminal node that produced this output.")
-    value: typing.Dict[str, typing.Any]
+    value: typing.Optional[typing.Dict[str, typing.Any]]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

{vellum_ai-0.0.30 → vellum_ai-0.0.31}/src/vellum/types/terminal_node_string_result.py

@@ -10,7 +10,7 @@ from ..core.datetime_utils import serialize_datetime
 
 class TerminalNodeStringResult(pydantic.BaseModel):
     name: str = pydantic.Field(description="The unique name given to the terminal node that produced this output.")
-    value: str
+    value: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}