update generated sdk

Author: CalebCourier

guard-rails-api-client/guard_rails_api_client/__init__.py

@@ -1,4 +1,5 @@
-""" A client library for accessing GuardRails API """
+"""A client library for accessing GuardRails API"""
+
 from .client import AuthenticatedClient, Client
 
 __all__ = (

guard-rails-api-client/guard_rails_api_client/api/__init__.py

@@ -1 +1 @@
-""" Contains methods for accessing the API """
+"""Contains methods for accessing the API"""

guard-rails-api-client/guard_rails_api_client/errors.py

@@ -1,4 +1,4 @@
-""" Contains shared errors types that can be raised from API functions """
+"""Contains shared errors types that can be raised from API functions"""
 
 
 class UnexpectedStatus(Exception):
@@ -8,7 +8,9 @@ def __init__(self, status_code: int, content: bytes):
         self.status_code = status_code
         self.content = content
 
-        super().__init__(f"Unexpected status code: {status_code}")
+        super().__init__(
+            f"Unexpected status code: {status_code}\n\nResponse content:\n{content.decode(errors='ignore')}"
+        )
 
 
 __all__ = ["UnexpectedStatus"]

guard-rails-api-client/guard_rails_api_client/models/__init__.py

@@ -1,4 +1,4 @@
-""" Contains all the data models used in inputs/outputs """
+"""Contains all the data models used in inputs/outputs"""
 
 from .any_object import AnyObject
 from .data_type import DataType
@@ -15,10 +15,8 @@
 from .health_check import HealthCheck
 from .history import History
 from .history_event import HistoryEvent
-from .history_event_parsed_output import HistoryEventParsedOutput
 from .history_event_prompt import HistoryEventPrompt
 from .history_event_reasks_item import HistoryEventReasksItem
-from .history_event_validated_output import HistoryEventValidatedOutput
 from .http_error import HttpError
 from .http_error_fields import HttpErrorFields
 from .ingestion import Ingestion
@@ -59,10 +57,8 @@
     "HealthCheck",
     "History",
     "HistoryEvent",
-    "HistoryEventParsedOutput",
     "HistoryEventPrompt",
     "HistoryEventReasksItem",
-    "HistoryEventValidatedOutput",
     "HttpError",
     "HttpErrorFields",
     "Ingestion",

guard-rails-api-client/guard_rails_api_client/models/history_event.py

@@ -1,15 +1,14 @@
-from typing import TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union
+from typing import TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
-    from ..models.history_event_parsed_output import HistoryEventParsedOutput
+    from ..models.any_object import AnyObject
     from ..models.history_event_prompt import HistoryEventPrompt
     from ..models.history_event_reasks_item import HistoryEventReasksItem
-    from ..models.history_event_validated_output import HistoryEventValidatedOutput
 
 
 T = TypeVar("T", bound="HistoryEvent")
@@ -21,28 +20,34 @@ class HistoryEvent:
     Attributes:
         instructions (Union[Unset, str]):
         output (Union[Unset, str]):
-        parsed_output (Union[Unset, HistoryEventParsedOutput]):
+        parsed_output (Union['AnyObject', Unset, str]):
         prompt (Union[Unset, HistoryEventPrompt]):
         reasks (Union[Unset, List['HistoryEventReasksItem']]):
-        validated_output (Union[Unset, HistoryEventValidatedOutput]):
+        validated_output (Union['AnyObject', Unset, str]):
     """
 
     instructions: Union[Unset, str] = UNSET
     output: Union[Unset, str] = UNSET
-    parsed_output: Union[Unset, "HistoryEventParsedOutput"] = UNSET
+    parsed_output: Union["AnyObject", Unset, str] = UNSET
     prompt: Union[Unset, "HistoryEventPrompt"] = UNSET
     reasks: Union[Unset, List["HistoryEventReasksItem"]] = UNSET
-    validated_output: Union[Unset, "HistoryEventValidatedOutput"] = UNSET
+    validated_output: Union["AnyObject", Unset, str] = UNSET
     additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> Dict[str, Any]:
+        from ..models.any_object import AnyObject
+
         instructions = self.instructions
 
         output = self.output
 
-        parsed_output: Union[Unset, Dict[str, Any]] = UNSET
-        if not isinstance(self.parsed_output, Unset):
+        parsed_output: Union[Dict[str, Any], Unset, str]
+        if isinstance(self.parsed_output, Unset):
+            parsed_output = UNSET
+        elif isinstance(self.parsed_output, AnyObject):
             parsed_output = self.parsed_output.to_dict()
+        else:
+            parsed_output = self.parsed_output
 
         prompt: Union[Unset, Dict[str, Any]] = UNSET
         if not isinstance(self.prompt, Unset):
@@ -55,9 +60,13 @@ def to_dict(self) -> Dict[str, Any]:
                 reasks_item = reasks_item_data.to_dict()
                 reasks.append(reasks_item)
 
-        validated_output: Union[Unset, Dict[str, Any]] = UNSET
-        if not isinstance(self.validated_output, Unset):
+        validated_output: Union[Dict[str, Any], Unset, str]
+        if isinstance(self.validated_output, Unset):
+            validated_output = UNSET
+        elif isinstance(self.validated_output, AnyObject):
             validated_output = self.validated_output.to_dict()
+        else:
+            validated_output = self.validated_output
 
         field_dict: Dict[str, Any] = {}
         field_dict.update(self.additional_properties)
@@ -79,22 +88,29 @@ def to_dict(self) -> Dict[str, Any]:
 
     @classmethod
     def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
-        from ..models.history_event_parsed_output import HistoryEventParsedOutput
+        from ..models.any_object import AnyObject
         from ..models.history_event_prompt import HistoryEventPrompt
         from ..models.history_event_reasks_item import HistoryEventReasksItem
-        from ..models.history_event_validated_output import HistoryEventValidatedOutput
 
         d = src_dict.copy()
         instructions = d.pop("instructions", UNSET)
 
         output = d.pop("output", UNSET)
 
-        _parsed_output = d.pop("parsedOutput", UNSET)
-        parsed_output: Union[Unset, HistoryEventParsedOutput]
-        if isinstance(_parsed_output, Unset):
-            parsed_output = UNSET
-        else:
-            parsed_output = HistoryEventParsedOutput.from_dict(_parsed_output)
+        def _parse_parsed_output(data: object) -> Union["AnyObject", Unset, str]:
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                parsed_output_type_0 = AnyObject.from_dict(data)
+
+                return parsed_output_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union["AnyObject", Unset, str], data)
+
+        parsed_output = _parse_parsed_output(d.pop("parsedOutput", UNSET))
 
         _prompt = d.pop("prompt", UNSET)
         prompt: Union[Unset, HistoryEventPrompt]
@@ -110,12 +126,20 @@ def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
 
             reasks.append(reasks_item)
 
-        _validated_output = d.pop("validatedOutput", UNSET)
-        validated_output: Union[Unset, HistoryEventValidatedOutput]
-        if isinstance(_validated_output, Unset):
-            validated_output = UNSET
-        else:
-            validated_output = HistoryEventValidatedOutput.from_dict(_validated_output)
+        def _parse_validated_output(data: object) -> Union["AnyObject", Unset, str]:
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                validated_output_type_0 = AnyObject.from_dict(data)
+
+                return validated_output_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union["AnyObject", Unset, str], data)
+
+        validated_output = _parse_validated_output(d.pop("validatedOutput", UNSET))
 
         history_event = cls(
             instructions=instructions,

guard-rails-api-client/guard_rails_api_client/models/history_event_parsed_output.py

@@ -1,4 +1,5 @@
-""" Contains some shared types for properties """
+"""Contains some shared types for properties"""
+
 from http import HTTPStatus
 from typing import BinaryIO, Generic, Literal, MutableMapping, Optional, Tuple, TypeVar
 

guard-rails-api-client/guard_rails_api_client/models/history_event_validated_output.py

@@ -7,7 +7,7 @@
 
 setup(
     name="guard-rails-api-client",
-    version="0.0.0",
+    version="0.0.2",
     description="A client library for accessing GuardRails API",
     long_description=long_description,
     long_description_content_type="text/markdown",

guard-rails-api-client/guard_rails_api_client/types.py

@@ -703,9 +703,9 @@ components:
         output:
           type: string
         parsedOutput:
-          type: object
-          properties: {}
-          additionalProperties: {}
+          oneOf:
+            - $ref: "#/components/schemas/AnyObject"
+            - type: string
         prompt:
           type: object
           properties:
@@ -718,9 +718,9 @@ components:
             properties: {}
             additionalProperties: {}
         validatedOutput:
-          type: object
-          properties: {}
-          additionalProperties: {}
+          oneOf:
+            - $ref: "#/components/schemas/AnyObject"
+            - type: string
       additionalProperties: {}
     Reask:
       type: object