vellum-ai 0.0.30__py3-none-any.whl → 0.0.31__py3-none-any.whl

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/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/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/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/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}

{vellum_ai-0.0.30.dist-info → vellum_ai-0.0.31.dist-info}/METADATA

@@ -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.dist-info → vellum_ai-0.0.31.dist-info}/RECORD

@@ -1,8 +1,8 @@
 vellum/__init__.py,sha256=HrSVslaxHhe2HO6epFwF49_XWDS4HhbfNRrSNVqZdss,9127
-vellum/client.py,sha256=Kxyp8ZazvSbjIEEoBAVCDJeqIHCfW4OMa5vZl84cRYQ,32325
+vellum/client.py,sha256=FXZCbMfR3jgPMXAdobRkPpInZN1dz4kOjKPgRsyvSEQ,32613
 vellum/core/__init__.py,sha256=QJS3CJ2TYP2E1Tge0CS6Z7r8LTNzJHQVX1hD3558eP0,519
 vellum/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-vellum/core/client_wrapper.py,sha256=wNp6eZVx2wfl0CBL2eRja6VzGPjd-VH0F27tkE1YSGk,758
+vellum/core/client_wrapper.py,sha256=svrxCUFTlYC-vgNq0OcVMPv2gWSuK7bbuGWyWJ2Y7f8,897
 vellum/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 vellum/core/jsonable_encoder.py,sha256=yHrx0C19n1H77G-GanO-HsFyBPVMlsJz7WffsHwXEVI,3710
 vellum/core/remove_none_from_dict.py,sha256=8m91FC3YuVem0Gm9_sXhJ2tGvP33owJJdrqCLEdowGw,330
@@ -123,12 +123,12 @@ vellum/types/search_weights_request.py,sha256=EdAbOOXOWXCq0C3lziyRHTuNIlODw0BECQ
 vellum/types/slim_document.py,sha256=BnS2YA9JLj6dBm21Z-GXnPSgi1T-RWWTaIjx2RKpk84,2820
 vellum/types/submit_completion_actual_request.py,sha256=k5BHx1JUfq8XcLbBZOsbSNzVA3xNVNVFsJkM7tMtXpo,1656
 vellum/types/submit_completion_actuals_error_response.py,sha256=8ZNcGD7l4crYQoKn4LTdMKbRVGHs3v-M3ER9lTkKzmM,772
-vellum/types/terminal_node_chat_history_result.py,sha256=SvlvoLfQFxUNRpaBNAbE6X72X54GrM1ERmaZ373efto,939
-vellum/types/terminal_node_json_result.py,sha256=X9Dzwy25RcBLU_8s2pK5bc-nHWRfoOUslJyKYAj0SsU,898
+vellum/types/terminal_node_chat_history_result.py,sha256=RzqPzyUswQ86pXHgjCPMhmkBhqjQVcTcLipwygXNQqY,956
+vellum/types/terminal_node_json_result.py,sha256=d85-Q-1NJLk8cVq6mWHUgQssYABe0sNHqkAOpV0l7Dk,915
 vellum/types/terminal_node_result.py,sha256=dNdckayDfIQ_LR9IKfklxyFfJErPgtxA9tzsDUXjiPA,833
 vellum/types/terminal_node_result_data.py,sha256=4fsLNH_jkTOIKViXfHavkpE2GhB5NBUJdnVni82tEpk,845
 vellum/types/terminal_node_result_output.py,sha256=QkdMPv5SH8gvtfVd2md9KoAe2inIBMINGul94ZZ-YXE,1107
-vellum/types/terminal_node_string_result.py,sha256=puo54vwX0G-3zDZVRlz5deVA8K-lOQQt0Lu50g8zWmA,875
+vellum/types/terminal_node_string_result.py,sha256=xl19fA2GR705Qx0MFQsz0vgl1XApcJL0njue85gMeCo,892
 vellum/types/test_suite_test_case.py,sha256=_zk_wcESjEWG_fePNPLXsmvpFJmB_bPueacVLVeihNg,1439
 vellum/types/upload_document_error_response.py,sha256=VNSsUbd0TSCKnlBohoKv1h3AJDJeicH-q3kjkXpxKG4,763
 vellum/types/upload_document_response.py,sha256=wi6kn2SWkE5ZlbSQvGXKuQUZInz-2ik1ib7A4kqvkGo,833
@@ -150,6 +150,6 @@ vellum/types/workflow_result_event_output_data_chat_history.py,sha256=viyNe9VF8-
 vellum/types/workflow_result_event_output_data_json.py,sha256=-zLIOCCOurui_vXE_WJlaDO5xfTQx81EGyWdgSurItk,1125
 vellum/types/workflow_result_event_output_data_string.py,sha256=EqTpT1SJxagtuFCxFkJuze64MgUjGuZlzYEOqnR1UGM,1253
 vellum/types/workflow_stream_event.py,sha256=sVpX2OBZq-vVn8A74opHnV_pD83eoICS5_aJ1H9WjxM,819
-vellum_ai-0.0.30.dist-info/METADATA,sha256=FOI3xQkEP5P61_nFQvadknFxElHIlCOPVy6470SVcus,3487
-vellum_ai-0.0.30.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-vellum_ai-0.0.30.dist-info/RECORD,,
+vellum_ai-0.0.31.dist-info/METADATA,sha256=mM6eMuIL0jR9oUFxNFfOVt7SAygB6X9mkiwbGVe6UbU,3487
+vellum_ai-0.0.31.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+vellum_ai-0.0.31.dist-info/RECORD,,