diff --git a/guardrails-api-client/guardrails_api_client/models/rail_spec.py b/guardrails-api-client/guardrails_api_client/models/rail_spec.py index 7903190..423217f 100644 --- a/guardrails-api-client/guardrails_api_client/models/rail_spec.py +++ b/guardrails-api-client/guardrails_api_client/models/rail_spec.py @@ -19,24 +19,22 @@ class RailSpec: Attributes: output_schema (Schema): - prompt (str): version (RailSpecVersion): Default: RailSpecVersion.VALUE_0. input_schema (Union[Unset, Schema]): instructions (Union[Unset, str]): + prompt (Union[Unset, str]): """ output_schema: "Schema" - prompt: str version: RailSpecVersion = RailSpecVersion.VALUE_0 input_schema: Union[Unset, "Schema"] = UNSET instructions: Union[Unset, str] = UNSET + prompt: Union[Unset, str] = UNSET additional_properties: Dict[str, Any] = _attrs_field(init=False, factory=dict) def to_dict(self) -> Dict[str, Any]: output_schema = self.output_schema.to_dict() - prompt = self.prompt - version = self.version.value input_schema: Union[Unset, Dict[str, Any]] = UNSET @@ -45,12 +43,13 @@ def to_dict(self) -> Dict[str, Any]: instructions = self.instructions + prompt = self.prompt + field_dict: Dict[str, Any] = {} field_dict.update(self.additional_properties) field_dict.update( { "outputSchema": output_schema, - "prompt": prompt, "version": version, } ) @@ -58,6 +57,8 @@ def to_dict(self) -> Dict[str, Any]: field_dict["inputSchema"] = input_schema if instructions is not UNSET: field_dict["instructions"] = instructions + if prompt is not UNSET: + field_dict["prompt"] = prompt return field_dict @@ -68,8 +69,6 @@ def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T: d = src_dict.copy() output_schema = Schema.from_dict(d.pop("outputSchema")) - prompt = d.pop("prompt") - version = RailSpecVersion(d.pop("version")) _input_schema = d.pop("inputSchema", UNSET) @@ -81,12 +80,14 @@ def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T: instructions = d.pop("instructions", UNSET) + prompt = d.pop("prompt", UNSET) + rail_spec = cls( output_schema=output_schema, - prompt=prompt, version=version, input_schema=input_schema, instructions=instructions, + prompt=prompt, ) rail_spec.additional_properties = d diff --git a/open-api-spec.yml b/open-api-spec.yml index db07370..aa3b899 100644 --- a/open-api-spec.yml +++ b/open-api-spec.yml @@ -599,7 +599,6 @@ components: - "0.1" required: - outputSchema - - prompt - version Schema: type: object diff --git a/service-specs/guardrails-service-spec.yml b/service-specs/guardrails-service-spec.yml index 73ef2fb..132077b 100644 --- a/service-specs/guardrails-service-spec.yml +++ b/service-specs/guardrails-service-spec.yml @@ -301,7 +301,6 @@ components: - "0.1" required: - outputSchema - - prompt - version Schema: type: object