SDK regeneration

Author: fern-api[bot]

src/truefoundry_sdk/__init__.py

@@ -65,6 +65,7 @@
     AwsAssumedRoleBasedAuth,
     AwsBedrockGuardrailConfig,
     AwsBedrockGuardrailConfigAuthData,
+    AwsBedrockGuardrailConfigOperation,
     AwsBedrockProviderAccount,
     AwsBedrockProviderAccountAuthData,
     AwsEcr,
@@ -164,6 +165,8 @@
     CustomBlobStorage,
     CustomGuardrailConfig,
     CustomGuardrailConfigAuthData,
+    CustomGuardrailConfigOperation,
+    CustomGuardrailConfigTarget,
     CustomHelmRepo,
     CustomIntegrations,
     CustomJwtAuthIntegration,
@@ -800,6 +803,7 @@
     "AwsAssumedRoleBasedAuth",
     "AwsBedrockGuardrailConfig",
     "AwsBedrockGuardrailConfigAuthData",
+    "AwsBedrockGuardrailConfigOperation",
     "AwsBedrockProviderAccount",
     "AwsBedrockProviderAccountAuthData",
     "AwsEcr",
@@ -902,6 +906,8 @@
     "CustomBlobStorage",
     "CustomGuardrailConfig",
     "CustomGuardrailConfigAuthData",
+    "CustomGuardrailConfigOperation",
+    "CustomGuardrailConfigTarget",
     "CustomHelmRepo",
     "CustomIntegrations",
     "CustomJwtAuthIntegration",

src/truefoundry_sdk/types/__init__.py

@@ -67,6 +67,7 @@
 from .aws_assumed_role_based_auth import AwsAssumedRoleBasedAuth
 from .aws_bedrock_guardrail_config import AwsBedrockGuardrailConfig
 from .aws_bedrock_guardrail_config_auth_data import AwsBedrockGuardrailConfigAuthData
+from .aws_bedrock_guardrail_config_operation import AwsBedrockGuardrailConfigOperation
 from .aws_bedrock_provider_account import AwsBedrockProviderAccount
 from .aws_bedrock_provider_account_auth_data import AwsBedrockProviderAccountAuthData
 from .aws_ecr import AwsEcr
@@ -166,6 +167,8 @@
 from .custom_blob_storage import CustomBlobStorage
 from .custom_guardrail_config import CustomGuardrailConfig
 from .custom_guardrail_config_auth_data import CustomGuardrailConfigAuthData
+from .custom_guardrail_config_operation import CustomGuardrailConfigOperation
+from .custom_guardrail_config_target import CustomGuardrailConfigTarget
 from .custom_helm_repo import CustomHelmRepo
 from .custom_integrations import CustomIntegrations
 from .custom_jwt_auth_integration import CustomJwtAuthIntegration
@@ -744,6 +747,7 @@
     "AwsAssumedRoleBasedAuth",
     "AwsBedrockGuardrailConfig",
     "AwsBedrockGuardrailConfigAuthData",
+    "AwsBedrockGuardrailConfigOperation",
     "AwsBedrockProviderAccount",
     "AwsBedrockProviderAccountAuthData",
     "AwsEcr",
@@ -843,6 +847,8 @@
     "CustomBlobStorage",
     "CustomGuardrailConfig",
     "CustomGuardrailConfigAuthData",
+    "CustomGuardrailConfigOperation",
+    "CustomGuardrailConfigTarget",
     "CustomHelmRepo",
     "CustomIntegrations",
     "CustomJwtAuthIntegration",

src/truefoundry_sdk/types/aws_bedrock_guardrail_config.py

@@ -5,6 +5,7 @@
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 from .aws_bedrock_guardrail_config_auth_data import AwsBedrockGuardrailConfigAuthData
+from .aws_bedrock_guardrail_config_operation import AwsBedrockGuardrailConfigOperation
 from .aws_region import AwsRegion
 from .base_guardrail_config import BaseGuardrailConfig
 
@@ -20,13 +21,21 @@ class AwsBedrockGuardrailConfig(BaseGuardrailConfig):
     +value=integration/guardrail-config/aws-bedrock
     """
 
+    operation: typing.Optional[AwsBedrockGuardrailConfigOperation] = pydantic.Field(default=None)
+    """
+    +label=Operation
+    +usage=The operation type to use for the Guardrail. Validate guardrails are used to validate requests and mutate can validate as well as mutate requests.
+    Validate guardrails are run in parallel while mutate guardrails are run sequentially.
+    +uiType=Select
+    +sort=50
+    """
+
     auth_data: typing.Optional[AwsBedrockGuardrailConfigAuthData] = pydantic.Field(default=None)
     """
     +label=AWS Account Auth Data
     +usage=Authentication data for the AWS account
     """
 
-    region: typing.Optional[AwsRegion] = None
     guardrail_id: typing.Optional[str] = pydantic.Field(default=None)
     """
     +label=Guardrail ID
@@ -39,11 +48,7 @@ class AwsBedrockGuardrailConfig(BaseGuardrailConfig):
     +usage=The version of the Guardrail to use.
     """
 
-    redact_pii: typing.Optional[bool] = pydantic.Field(default=None)
-    """
-    +label=Redact PII
-    +usage=Whether to redact PII from the response. If this is true, your request will be transformed to redact PII from the response else a validation error will be returned.
-    """
+    region: typing.Optional[AwsRegion] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow")  # type: ignore # Pydantic v2

src/truefoundry_sdk/types/aws_bedrock_guardrail_config_operation.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class AwsBedrockGuardrailConfigOperation(str, enum.Enum):
+    """
+    +label=Operation
+    +usage=The operation type to use for the Guardrail. Validate guardrails are used to validate requests and mutate can validate as well as mutate requests.
+    Validate guardrails are run in parallel while mutate guardrails are run sequentially.
+    +uiType=Select
+    +sort=50
+    """
+
+    VALIDATE = "validate"
+    MUTATE = "mutate"
+
+    def visit(self, validate: typing.Callable[[], T_Result], mutate: typing.Callable[[], T_Result]) -> T_Result:
+        if self is AwsBedrockGuardrailConfigOperation.VALIDATE:
+            return validate()
+        if self is AwsBedrockGuardrailConfigOperation.MUTATE:
+            return mutate()

src/truefoundry_sdk/types/custom_guardrail_config.py

@@ -6,6 +6,8 @@
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 from .base_guardrail_config import BaseGuardrailConfig
 from .custom_guardrail_config_auth_data import CustomGuardrailConfigAuthData
+from .custom_guardrail_config_operation import CustomGuardrailConfigOperation
+from .custom_guardrail_config_target import CustomGuardrailConfigTarget
 
 
 class CustomGuardrailConfig(BaseGuardrailConfig):
@@ -19,33 +21,49 @@ class CustomGuardrailConfig(BaseGuardrailConfig):
     +value=integration/guardrail-config/custom
     """
 
+    operation: typing.Optional[CustomGuardrailConfigOperation] = pydantic.Field(default=None)
+    """
+    +label=Operation
+    +usage=The operation type to use for the Guardrail. Validate guardrails are used to validate requests and mutate can validate as well as mutate requests.
+    Validate guardrails are run in parallel while mutate guardrails are run sequentially.
+    +uiType=Select
+    +sort=100
+    """
+
+    target: typing.Optional[CustomGuardrailConfigTarget] = pydantic.Field(default=None)
+    """
+    +label=Target
+    +usage=Specify whether the guardrail should be applied to the request or response. Guardrails with target "Request" can be only used in input guardrails and guardrails with target "Response" can only be used in output guardrails.
+    +uiType=Select
+    +sort=200
+    """
+
     url: typing.Optional[str] = pydantic.Field(default=None)
     """
     +label=URL
+    +sort=300
     +usage=The URL of the Guardrail to send a post request to.
     """
 
     auth_data: typing.Optional[CustomGuardrailConfigAuthData] = pydantic.Field(default=None)
     """
     +label=Auth Data
+    +sort=450
     +usage=Authentication data for the Guardrail Server.
     """
 
     headers: typing.Optional[typing.Dict[str, str]] = pydantic.Field(default=None)
     """
     +label=Headers
+    +sort=500
     +usage=Headers for the Guardrail Server. Forwarded to the Guardrail Server as is. For example: `{"Authorization": "APIKey <token>"}`
     """
 
-    redact_pii: typing.Optional[bool] = pydantic.Field(default=None)
-    """
-    +label=Redact PII
-    +usage=Whether to redact PII from the response. If this is true, your request will be transformed to redact PII from the response else a validation error will be returned.
-    """
-
     config: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
     """
     +label=Config
+    +sort=600
+    +uiType=JsonInput
     +usage=The config for the Guardrail Server. This is a JSON object that will be sent as a config to Guardrail Server along with the request.
     """
 

src/truefoundry_sdk/types/custom_guardrail_config_operation.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class CustomGuardrailConfigOperation(str, enum.Enum):
+    """
+    +label=Operation
+    +usage=The operation type to use for the Guardrail. Validate guardrails are used to validate requests and mutate can validate as well as mutate requests.
+    Validate guardrails are run in parallel while mutate guardrails are run sequentially.
+    +uiType=Select
+    +sort=100
+    """
+
+    VALIDATE = "validate"
+    MUTATE = "mutate"
+
+    def visit(self, validate: typing.Callable[[], T_Result], mutate: typing.Callable[[], T_Result]) -> T_Result:
+        if self is CustomGuardrailConfigOperation.VALIDATE:
+            return validate()
+        if self is CustomGuardrailConfigOperation.MUTATE:
+            return mutate()

src/truefoundry_sdk/types/custom_guardrail_config_target.py

@@ -0,0 +1,24 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class CustomGuardrailConfigTarget(str, enum.Enum):
+    """
+    +label=Target
+    +usage=Specify whether the guardrail should be applied to the request or response. Guardrails with target "Request" can be only used in input guardrails and guardrails with target "Response" can only be used in output guardrails.
+    +uiType=Select
+    +sort=200
+    """
+
+    REQUEST = "request"
+    RESPONSE = "response"
+
+    def visit(self, request: typing.Callable[[], T_Result], response: typing.Callable[[], T_Result]) -> T_Result:
+        if self is CustomGuardrailConfigTarget.REQUEST:
+            return request()
+        if self is CustomGuardrailConfigTarget.RESPONSE:
+            return response()

src/truefoundry_sdk/types/dynamic_volume_config.py

@@ -8,7 +8,7 @@
 
 class DynamicVolumeConfig(UniversalBaseModel):
     """
-    +label=Dynamic Volume Config
+    +label=Create new volume
     """
 
     type: typing.Literal["dynamic"] = pydantic.Field(default="dynamic")

src/truefoundry_sdk/types/open_ai_moderations_guardrail_config.py

@@ -23,21 +23,21 @@ class OpenAiModerationsGuardrailConfig(BaseGuardrailConfig):
     """
     +value=integration/guardrail-config/openai-moderations
     +sort=50
+    +uiType=Hidden
     """
 
-    auth_data: typing.Optional[OpenaiApiKeyAuth] = None
     base_url: typing.Optional[str] = pydantic.Field(default=None)
     """
     +label=Base URL
-    +sort=300
-    +usage=Optional custom base URL for OpenAI API
-    +message=Base URL must not be empty
+    +sort=100
+    +usage=Optional custom base URL for OpenAI API. If not provided, the default base URL will be used.
     """
 
+    auth_data: typing.Optional[OpenaiApiKeyAuth] = None
     model: typing.Optional[str] = pydantic.Field(default="omni-moderation-latest")
     """
     +label=OpenAI Moderation Model
-    +sort=400
+    +sort=300
     +usage=The model to use for the OpenAI Moderation API.
     """
 
@@ -46,8 +46,9 @@ class OpenAiModerationsGuardrailConfig(BaseGuardrailConfig):
     )
     """
     +label=Category Thresholds
-    +sort=500
+    +sort=400
     +usage=The thresholds for the OpenAI Moderation API.
+    +uiType=Hidden
     """
 
     if IS_PYDANTIC_V2:

src/truefoundry_sdk/types/static_volume_config.py

@@ -8,7 +8,7 @@
 
 class StaticVolumeConfig(UniversalBaseModel):
     """
-    +label=Static Volume Config
+    +label=Use existing volume
     """
 
     type: typing.Literal["static"] = pydantic.Field(default="static")