update generated sdk

Author: CalebCourier

guardrails-api-client/guardrails_api_client/models/__init__.py

@@ -33,6 +33,7 @@
 from .schema_element import SchemaElement
 from .status import Status
 from .validate_payload import ValidatePayload
+from .validate_payload_llm_api import ValidatePayloadLlmApi
 from .validate_payload_prompt_params import ValidatePayloadPromptParams
 from .validation_output import ValidationOutput
 from .validator_metrics import ValidatorMetrics
@@ -74,6 +75,7 @@
     "SchemaElement",
     "Status",
     "ValidatePayload",
+    "ValidatePayloadLlmApi",
     "ValidatePayloadPromptParams",
     "ValidationOutput",
     "ValidatorMetrics",

guardrails-api-client/guardrails_api_client/models/validate_payload.py

@@ -1,8 +1,9 @@
-from typing import TYPE_CHECKING, Any, Dict, List, Literal, Type, TypeVar, Union, cast
+from typing import TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
+from ..models.validate_payload_llm_api import ValidatePayloadLlmApi
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
@@ -20,26 +21,13 @@ class ValidatePayload:
             output.
         num_reasks (Union[Unset, int]): An override for the number of re-asks to perform
         prompt_params (Union[Unset, ValidatePayloadPromptParams]): additional params for llm prompts
-        llm_api (Union[Literal['litellm.acompletion'], Literal['litellm.completion'],
-            Literal['openai.ChatCompletion.acreate'], Literal['openai.ChatCompletion.create'],
-            Literal['openai.Completion.acreate'], Literal['openai.Completion.create'],
-            Literal['openai.chat.completions.create'], Literal['openai.completions.create'], Unset]):
+        llm_api (Union[Unset, ValidatePayloadLlmApi]):
     """
 
     llm_output: Union[Unset, str] = UNSET
     num_reasks: Union[Unset, int] = UNSET
     prompt_params: Union[Unset, "ValidatePayloadPromptParams"] = UNSET
-    llm_api: Union[
-        Literal["litellm.acompletion"],
-        Literal["litellm.completion"],
-        Literal["openai.ChatCompletion.acreate"],
-        Literal["openai.ChatCompletion.create"],
-        Literal["openai.Completion.acreate"],
-        Literal["openai.Completion.create"],
-        Literal["openai.chat.completions.create"],
-        Literal["openai.completions.create"],
-        Unset,
-    ] = UNSET
+    llm_api: Union[Unset, ValidatePayloadLlmApi] = UNSET
     additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> Dict[str, Any]:
@@ -51,21 +39,9 @@ def to_dict(self) -> Dict[str, Any]:
         if not isinstance(self.prompt_params, Unset):
             prompt_params = self.prompt_params.to_dict()
 
-        llm_api: Union[
-            Literal["litellm.acompletion"],
-            Literal["litellm.completion"],
-            Literal["openai.ChatCompletion.acreate"],
-            Literal["openai.ChatCompletion.create"],
-            Literal["openai.Completion.acreate"],
-            Literal["openai.Completion.create"],
-            Literal["openai.chat.completions.create"],
-            Literal["openai.completions.create"],
-            Unset,
-        ]
-        if isinstance(self.llm_api, Unset):
-            llm_api = UNSET
-        else:
-            llm_api = self.llm_api
+        llm_api: Union[Unset, str] = UNSET
+        if not isinstance(self.llm_api, Unset):
+            llm_api = self.llm_api.value
 
         field_dict: Dict[str, Any] = {}
         field_dict.update(self.additional_properties)
@@ -97,37 +73,12 @@ def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
         else:
             prompt_params = ValidatePayloadPromptParams.from_dict(_prompt_params)
 
-        def _parse_llm_api(
-            data: object,
-        ) -> Union[
-            Literal["litellm.acompletion"],
-            Literal["litellm.completion"],
-            Literal["openai.ChatCompletion.acreate"],
-            Literal["openai.ChatCompletion.create"],
-            Literal["openai.Completion.acreate"],
-            Literal["openai.Completion.create"],
-            Literal["openai.chat.completions.create"],
-            Literal["openai.completions.create"],
-            Unset,
-        ]:
-            if isinstance(data, Unset):
-                return data
-            return cast(
-                Union[
-                    Literal["litellm.acompletion"],
-                    Literal["litellm.completion"],
-                    Literal["openai.ChatCompletion.acreate"],
-                    Literal["openai.ChatCompletion.create"],
-                    Literal["openai.Completion.acreate"],
-                    Literal["openai.Completion.create"],
-                    Literal["openai.chat.completions.create"],
-                    Literal["openai.completions.create"],
-                    Unset,
-                ],
-                data,
-            )
-
-        llm_api = _parse_llm_api(d.pop("llmApi", UNSET))
+        _llm_api = d.pop("llmApi", UNSET)
+        llm_api: Union[Unset, ValidatePayloadLlmApi]
+        if isinstance(_llm_api, Unset):
+            llm_api = UNSET
+        else:
+            llm_api = ValidatePayloadLlmApi(_llm_api)
 
         validate_payload = cls(
             llm_output=llm_output,

guardrails-api-client/guardrails_api_client/models/validate_payload_llm_api.py

@@ -0,0 +1,15 @@
+from enum import Enum
+
+
+class ValidatePayloadLlmApi(str, Enum):
+    LITELLM_ACOMPLETION = "litellm.acompletion"
+    LITELLM_COMPLETION = "litellm.completion"
+    OPENAI_CHATCOMPLETION_ACREATE = "openai.ChatCompletion.acreate"
+    OPENAI_CHATCOMPLETION_CREATE = "openai.ChatCompletion.create"
+    OPENAI_CHAT_COMPLETIONS_CREATE = "openai.chat.completions.create"
+    OPENAI_COMPLETIONS_CREATE = "openai.completions.create"
+    OPENAI_COMPLETION_ACREATE = "openai.Completion.acreate"
+    OPENAI_COMPLETION_CREATE = "openai.Completion.create"
+
+    def __str__(self) -> str:
+        return str(self.value)

open-api-spec.yml

@@ -763,33 +763,6 @@ components:
               oneOf:
                 - $ref: "#/components/schemas/AnyObject"
                 - $ref: "#/components/schemas/AnyPrimitive"
-    LlmApi:
-      type: string
-      oneOf:
-        - title: OPENAI_V0_COMPLETION_CREATE
-          const: openai.Completion.create
-          description: OpenAI V0 Completion Create
-        - title: OPENAI_V1_COMPLETIONS_CREATE
-          const: openai.completions.create
-          description: OpenAI V1 Completions Create
-        - title: OPENAI_V0_CHAT_COMPLETION_CREATE
-          const: openai.ChatCompletion.create
-          description: OpenAI V0 Chat Completion Create
-        - title: OPENAI_V1_CHAT_COMPLETION_CREATE
-          const: openai.chat.completions.create
-          description: OpenAI V1 Chat Completions Create
-        - title: OPENAI_V0_COMPLETION_ASYNC_CREATE
-          const: openai.Completion.acreate
-          description: OpenAI V0 Completion Async Create
-        - title: OPENAI_V0_CHAT_COMPLETION_ASYNC_CREATE
-          const: openai.ChatCompletion.acreate
-          description: OpenAI V0 Chat Completion Async Create
-        - title: LITELLM_COMPLETION
-          const: litellm.completion
-          description: LiteLLM Completion
-        - title: LITELLM_ASYNC_COMPLETION
-          const: litellm.acompletion
-          description: LiteLLM Async Completion
     ValidatePayload:
       type: object
       properties:
@@ -806,7 +779,8 @@ components:
           properties: {}
           additionalProperties: {}
         llmApi:
-          $ref: "#/components/schemas/LlmApi"
+          type: string
+          enum: ["openai.Completion.create", "openai.completions.create", "openai.ChatCompletion.create", "openai.chat.completions.create", "openai.Completion.acreate", "openai.ChatCompletion.acreate", "litellm.completion", "litellm.acompletion"]
       additionalProperties: {}
     Ingestion:
       type: object