pulumi-datarobot 0.8.4__tar.gz → 0.8.5__tar.gz

{pulumi_datarobot-0.8.4 → pulumi_datarobot-0.8.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pulumi_datarobot
-Version: 0.8.4
+Version: 0.8.5
 Summary: A Pulumi package for creating and managing DataRobot resources.
 Home-page: https://www.datarobot.com
 License: Apache-2.0

{pulumi_datarobot-0.8.4 → pulumi_datarobot-0.8.5}/pulumi_datarobot/__init__.py

@@ -18,6 +18,7 @@ from .custom_metric import *
 from .custom_metric_from_job import *
 from .custom_metric_job import *
 from .custom_model import *
+from .custom_model_llm_validation import *
 from .dataset_from_datasource import *
 from .dataset_from_file import *
 from .dataset_from_url import *
@@ -158,6 +159,14 @@ _utilities.register(
    "datarobot:index/customModel:CustomModel": "CustomModel"
   }
  },
+ {
+  "pkg": "datarobot",
+  "mod": "index/customModelLlmValidation",
+  "fqn": "pulumi_datarobot",
+  "classes": {
+   "datarobot:index/customModelLlmValidation:CustomModelLlmValidation": "CustomModelLlmValidation"
+  }
+ },
  {
   "pkg": "datarobot",
   "mod": "index/datasetFromDatasource",

{pulumi_datarobot-0.8.4 → pulumi_datarobot-0.8.5}/pulumi_datarobot/_inputs.py

@@ -137,6 +137,8 @@ __all__ = [
     'DeploymentRuntimeParameterValueArgsDict',
     'DeploymentSegmentAnalysisSettingsArgs',
     'DeploymentSegmentAnalysisSettingsArgsDict',
+    'LlmBlueprintCustomModelLlmSettingsArgs',
+    'LlmBlueprintCustomModelLlmSettingsArgsDict',
     'LlmBlueprintLlmSettingsArgs',
     'LlmBlueprintLlmSettingsArgsDict',
     'LlmBlueprintVectorDatabaseSettingsArgs',
@@ -5225,6 +5227,78 @@ class DeploymentSegmentAnalysisSettingsArgs:
         pulumi.set(self, "attributes", value)
 
 
+if not MYPY:
+    class LlmBlueprintCustomModelLlmSettingsArgsDict(TypedDict):
+        external_llm_context_size: NotRequired[pulumi.Input[int]]
+        """
+        The external LLM's context size, in tokens. This value is only used for pruning documents supplied to the LLM when a vector database is associated with the LLM blueprint. It does not affect the external LLM's actual context size in any way and is not supplied to the LLM.
+        """
+        system_prompt: NotRequired[pulumi.Input[str]]
+        """
+        System prompt guides the style of the LLM response. It is a 'universal' prompt, prepended to all individual prompts.
+        """
+        validation_id: NotRequired[pulumi.Input[str]]
+        """
+        The validation ID of the custom model LLM.
+        """
+elif False:
+    LlmBlueprintCustomModelLlmSettingsArgsDict: TypeAlias = Mapping[str, Any]
+
+@pulumi.input_type
+class LlmBlueprintCustomModelLlmSettingsArgs:
+    def __init__(__self__, *,
+                 external_llm_context_size: Optional[pulumi.Input[int]] = None,
+                 system_prompt: Optional[pulumi.Input[str]] = None,
+                 validation_id: Optional[pulumi.Input[str]] = None):
+        """
+        :param pulumi.Input[int] external_llm_context_size: The external LLM's context size, in tokens. This value is only used for pruning documents supplied to the LLM when a vector database is associated with the LLM blueprint. It does not affect the external LLM's actual context size in any way and is not supplied to the LLM.
+        :param pulumi.Input[str] system_prompt: System prompt guides the style of the LLM response. It is a 'universal' prompt, prepended to all individual prompts.
+        :param pulumi.Input[str] validation_id: The validation ID of the custom model LLM.
+        """
+        if external_llm_context_size is not None:
+            pulumi.set(__self__, "external_llm_context_size", external_llm_context_size)
+        if system_prompt is not None:
+            pulumi.set(__self__, "system_prompt", system_prompt)
+        if validation_id is not None:
+            pulumi.set(__self__, "validation_id", validation_id)
+
+    @property
+    @pulumi.getter(name="externalLlmContextSize")
+    def external_llm_context_size(self) -> Optional[pulumi.Input[int]]:
+        """
+        The external LLM's context size, in tokens. This value is only used for pruning documents supplied to the LLM when a vector database is associated with the LLM blueprint. It does not affect the external LLM's actual context size in any way and is not supplied to the LLM.
+        """
+        return pulumi.get(self, "external_llm_context_size")
+
+    @external_llm_context_size.setter
+    def external_llm_context_size(self, value: Optional[pulumi.Input[int]]):
+        pulumi.set(self, "external_llm_context_size", value)
+
+    @property
+    @pulumi.getter(name="systemPrompt")
+    def system_prompt(self) -> Optional[pulumi.Input[str]]:
+        """
+        System prompt guides the style of the LLM response. It is a 'universal' prompt, prepended to all individual prompts.
+        """
+        return pulumi.get(self, "system_prompt")
+
+    @system_prompt.setter
+    def system_prompt(self, value: Optional[pulumi.Input[str]]):
+        pulumi.set(self, "system_prompt", value)
+
+    @property
+    @pulumi.getter(name="validationId")
+    def validation_id(self) -> Optional[pulumi.Input[str]]:
+        """
+        The validation ID of the custom model LLM.
+        """
+        return pulumi.get(self, "validation_id")
+
+    @validation_id.setter
+    def validation_id(self, value: Optional[pulumi.Input[str]]):
+        pulumi.set(self, "validation_id", value)
+
+
 if not MYPY:
     class LlmBlueprintLlmSettingsArgsDict(TypedDict):
         max_completion_length: NotRequired[pulumi.Input[int]]

pulumi_datarobot-0.8.5/pulumi_datarobot/custom_model_llm_validation.py

@@ -0,0 +1,465 @@
+# coding=utf-8
+# *** WARNING: this file was generated by the Pulumi Terraform Bridge (tfgen) Tool. ***
+# *** Do not edit by hand unless you're certain you know what you are doing! ***
+
+import copy
+import warnings
+import sys
+import pulumi
+import pulumi.runtime
+from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict, TypeAlias
+else:
+    from typing_extensions import NotRequired, TypedDict, TypeAlias
+from . import _utilities
+
+__all__ = ['CustomModelLlmValidationArgs', 'CustomModelLlmValidation']
+
+@pulumi.input_type
+class CustomModelLlmValidationArgs:
+    def __init__(__self__, *,
+                 deployment_id: pulumi.Input[str],
+                 prompt_column_name: pulumi.Input[str],
+                 target_column_name: pulumi.Input[str],
+                 model_id: Optional[pulumi.Input[str]] = None,
+                 name: Optional[pulumi.Input[str]] = None,
+                 prediction_timeout: Optional[pulumi.Input[int]] = None,
+                 use_case_id: Optional[pulumi.Input[str]] = None):
+        """
+        The set of arguments for constructing a CustomModelLlmValidation resource.
+        :param pulumi.Input[str] deployment_id: The ID of the custom model deployment.
+        :param pulumi.Input[str] prompt_column_name: The name of the column the custom model uses for prompt text input.
+        :param pulumi.Input[str] target_column_name: The name of the column the custom model uses for prediction output.
+        :param pulumi.Input[str] model_id: The ID of the model used in the deployment.
+        :param pulumi.Input[str] name: The name to use for the validated custom model.
+        :param pulumi.Input[int] prediction_timeout: The timeout in seconds for the prediction when validating a custom model. Defaults to 300.
+        :param pulumi.Input[str] use_case_id: The ID of the use case to associate with the validated custom model.
+        """
+        pulumi.set(__self__, "deployment_id", deployment_id)
+        pulumi.set(__self__, "prompt_column_name", prompt_column_name)
+        pulumi.set(__self__, "target_column_name", target_column_name)
+        if model_id is not None:
+            pulumi.set(__self__, "model_id", model_id)
+        if name is not None:
+            pulumi.set(__self__, "name", name)
+        if prediction_timeout is not None:
+            pulumi.set(__self__, "prediction_timeout", prediction_timeout)
+        if use_case_id is not None:
+            pulumi.set(__self__, "use_case_id", use_case_id)
+
+    @property
+    @pulumi.getter(name="deploymentId")
+    def deployment_id(self) -> pulumi.Input[str]:
+        """
+        The ID of the custom model deployment.
+        """
+        return pulumi.get(self, "deployment_id")
+
+    @deployment_id.setter
+    def deployment_id(self, value: pulumi.Input[str]):
+        pulumi.set(self, "deployment_id", value)
+
+    @property
+    @pulumi.getter(name="promptColumnName")
+    def prompt_column_name(self) -> pulumi.Input[str]:
+        """
+        The name of the column the custom model uses for prompt text input.
+        """
+        return pulumi.get(self, "prompt_column_name")
+
+    @prompt_column_name.setter
+    def prompt_column_name(self, value: pulumi.Input[str]):
+        pulumi.set(self, "prompt_column_name", value)
+
+    @property
+    @pulumi.getter(name="targetColumnName")
+    def target_column_name(self) -> pulumi.Input[str]:
+        """
+        The name of the column the custom model uses for prediction output.
+        """
+        return pulumi.get(self, "target_column_name")
+
+    @target_column_name.setter
+    def target_column_name(self, value: pulumi.Input[str]):
+        pulumi.set(self, "target_column_name", value)
+
+    @property
+    @pulumi.getter(name="modelId")
+    def model_id(self) -> Optional[pulumi.Input[str]]:
+        """
+        The ID of the model used in the deployment.
+        """
+        return pulumi.get(self, "model_id")
+
+    @model_id.setter
+    def model_id(self, value: Optional[pulumi.Input[str]]):
+        pulumi.set(self, "model_id", value)
+
+    @property
+    @pulumi.getter
+    def name(self) -> Optional[pulumi.Input[str]]:
+        """
+        The name to use for the validated custom model.
+        """
+        return pulumi.get(self, "name")
+
+    @name.setter
+    def name(self, value: Optional[pulumi.Input[str]]):
+        pulumi.set(self, "name", value)
+
+    @property
+    @pulumi.getter(name="predictionTimeout")
+    def prediction_timeout(self) -> Optional[pulumi.Input[int]]:
+        """
+        The timeout in seconds for the prediction when validating a custom model. Defaults to 300.
+        """
+        return pulumi.get(self, "prediction_timeout")
+
+    @prediction_timeout.setter
+    def prediction_timeout(self, value: Optional[pulumi.Input[int]]):
+        pulumi.set(self, "prediction_timeout", value)
+
+    @property
+    @pulumi.getter(name="useCaseId")
+    def use_case_id(self) -> Optional[pulumi.Input[str]]:
+        """
+        The ID of the use case to associate with the validated custom model.
+        """
+        return pulumi.get(self, "use_case_id")
+
+    @use_case_id.setter
+    def use_case_id(self, value: Optional[pulumi.Input[str]]):
+        pulumi.set(self, "use_case_id", value)
+
+
+@pulumi.input_type
+class _CustomModelLlmValidationState:
+    def __init__(__self__, *,
+                 deployment_id: Optional[pulumi.Input[str]] = None,
+                 model_id: Optional[pulumi.Input[str]] = None,
+                 name: Optional[pulumi.Input[str]] = None,
+                 prediction_timeout: Optional[pulumi.Input[int]] = None,
+                 prompt_column_name: Optional[pulumi.Input[str]] = None,
+                 target_column_name: Optional[pulumi.Input[str]] = None,
+                 use_case_id: Optional[pulumi.Input[str]] = None):
+        """
+        Input properties used for looking up and filtering CustomModelLlmValidation resources.
+        :param pulumi.Input[str] deployment_id: The ID of the custom model deployment.
+        :param pulumi.Input[str] model_id: The ID of the model used in the deployment.
+        :param pulumi.Input[str] name: The name to use for the validated custom model.
+        :param pulumi.Input[int] prediction_timeout: The timeout in seconds for the prediction when validating a custom model. Defaults to 300.
+        :param pulumi.Input[str] prompt_column_name: The name of the column the custom model uses for prompt text input.
+        :param pulumi.Input[str] target_column_name: The name of the column the custom model uses for prediction output.
+        :param pulumi.Input[str] use_case_id: The ID of the use case to associate with the validated custom model.
+        """
+        if deployment_id is not None:
+            pulumi.set(__self__, "deployment_id", deployment_id)
+        if model_id is not None:
+            pulumi.set(__self__, "model_id", model_id)
+        if name is not None:
+            pulumi.set(__self__, "name", name)
+        if prediction_timeout is not None:
+            pulumi.set(__self__, "prediction_timeout", prediction_timeout)
+        if prompt_column_name is not None:
+            pulumi.set(__self__, "prompt_column_name", prompt_column_name)
+        if target_column_name is not None:
+            pulumi.set(__self__, "target_column_name", target_column_name)
+        if use_case_id is not None:
+            pulumi.set(__self__, "use_case_id", use_case_id)
+
+    @property
+    @pulumi.getter(name="deploymentId")
+    def deployment_id(self) -> Optional[pulumi.Input[str]]:
+        """
+        The ID of the custom model deployment.
+        """
+        return pulumi.get(self, "deployment_id")
+
+    @deployment_id.setter
+    def deployment_id(self, value: Optional[pulumi.Input[str]]):
+        pulumi.set(self, "deployment_id", value)
+
+    @property
+    @pulumi.getter(name="modelId")
+    def model_id(self) -> Optional[pulumi.Input[str]]:
+        """
+        The ID of the model used in the deployment.
+        """
+        return pulumi.get(self, "model_id")
+
+    @model_id.setter
+    def model_id(self, value: Optional[pulumi.Input[str]]):
+        pulumi.set(self, "model_id", value)
+
+    @property
+    @pulumi.getter
+    def name(self) -> Optional[pulumi.Input[str]]:
+        """
+        The name to use for the validated custom model.
+        """
+        return pulumi.get(self, "name")
+
+    @name.setter
+    def name(self, value: Optional[pulumi.Input[str]]):
+        pulumi.set(self, "name", value)
+
+    @property
+    @pulumi.getter(name="predictionTimeout")
+    def prediction_timeout(self) -> Optional[pulumi.Input[int]]:
+        """
+        The timeout in seconds for the prediction when validating a custom model. Defaults to 300.
+        """
+        return pulumi.get(self, "prediction_timeout")
+
+    @prediction_timeout.setter
+    def prediction_timeout(self, value: Optional[pulumi.Input[int]]):
+        pulumi.set(self, "prediction_timeout", value)
+
+    @property
+    @pulumi.getter(name="promptColumnName")
+    def prompt_column_name(self) -> Optional[pulumi.Input[str]]:
+        """
+        The name of the column the custom model uses for prompt text input.
+        """
+        return pulumi.get(self, "prompt_column_name")
+
+    @prompt_column_name.setter
+    def prompt_column_name(self, value: Optional[pulumi.Input[str]]):
+        pulumi.set(self, "prompt_column_name", value)
+
+    @property
+    @pulumi.getter(name="targetColumnName")
+    def target_column_name(self) -> Optional[pulumi.Input[str]]:
+        """
+        The name of the column the custom model uses for prediction output.
+        """
+        return pulumi.get(self, "target_column_name")
+
+    @target_column_name.setter
+    def target_column_name(self, value: Optional[pulumi.Input[str]]):
+        pulumi.set(self, "target_column_name", value)
+
+    @property
+    @pulumi.getter(name="useCaseId")
+    def use_case_id(self) -> Optional[pulumi.Input[str]]:
+        """
+        The ID of the use case to associate with the validated custom model.
+        """
+        return pulumi.get(self, "use_case_id")
+
+    @use_case_id.setter
+    def use_case_id(self, value: Optional[pulumi.Input[str]]):
+        pulumi.set(self, "use_case_id", value)
+
+
+class CustomModelLlmValidation(pulumi.CustomResource):
+    @overload
+    def __init__(__self__,
+                 resource_name: str,
+                 opts: Optional[pulumi.ResourceOptions] = None,
+                 deployment_id: Optional[pulumi.Input[str]] = None,
+                 model_id: Optional[pulumi.Input[str]] = None,
+                 name: Optional[pulumi.Input[str]] = None,
+                 prediction_timeout: Optional[pulumi.Input[int]] = None,
+                 prompt_column_name: Optional[pulumi.Input[str]] = None,
+                 target_column_name: Optional[pulumi.Input[str]] = None,
+                 use_case_id: Optional[pulumi.Input[str]] = None,
+                 __props__=None):
+        """
+        Custom Model LLM Validation
+
+        ## Example Usage
+
+        ```python
+        import pulumi
+        import pulumi_datarobot as datarobot
+
+        example = datarobot.CustomModelLlmValidation("example",
+            deployment_id=datarobot_deployment["example"]["id"],
+            prompt_column_name="promptText",
+            target_column_name="resultText",
+            prediction_timeout=100,
+            use_case_id=datarobot["use_case"]["example"]["id"])
+        pulumi.export("exampleId", example.id)
+        ```
+
+        :param str resource_name: The name of the resource.
+        :param pulumi.ResourceOptions opts: Options for the resource.
+        :param pulumi.Input[str] deployment_id: The ID of the custom model deployment.
+        :param pulumi.Input[str] model_id: The ID of the model used in the deployment.
+        :param pulumi.Input[str] name: The name to use for the validated custom model.
+        :param pulumi.Input[int] prediction_timeout: The timeout in seconds for the prediction when validating a custom model. Defaults to 300.
+        :param pulumi.Input[str] prompt_column_name: The name of the column the custom model uses for prompt text input.
+        :param pulumi.Input[str] target_column_name: The name of the column the custom model uses for prediction output.
+        :param pulumi.Input[str] use_case_id: The ID of the use case to associate with the validated custom model.
+        """
+        ...
+    @overload
+    def __init__(__self__,
+                 resource_name: str,
+                 args: CustomModelLlmValidationArgs,
+                 opts: Optional[pulumi.ResourceOptions] = None):
+        """
+        Custom Model LLM Validation
+
+        ## Example Usage
+
+        ```python
+        import pulumi
+        import pulumi_datarobot as datarobot
+
+        example = datarobot.CustomModelLlmValidation("example",
+            deployment_id=datarobot_deployment["example"]["id"],
+            prompt_column_name="promptText",
+            target_column_name="resultText",
+            prediction_timeout=100,
+            use_case_id=datarobot["use_case"]["example"]["id"])
+        pulumi.export("exampleId", example.id)
+        ```
+
+        :param str resource_name: The name of the resource.
+        :param CustomModelLlmValidationArgs args: The arguments to use to populate this resource's properties.
+        :param pulumi.ResourceOptions opts: Options for the resource.
+        """
+        ...
+    def __init__(__self__, resource_name: str, *args, **kwargs):
+        resource_args, opts = _utilities.get_resource_args_opts(CustomModelLlmValidationArgs, pulumi.ResourceOptions, *args, **kwargs)
+        if resource_args is not None:
+            __self__._internal_init(resource_name, opts, **resource_args.__dict__)
+        else:
+            __self__._internal_init(resource_name, *args, **kwargs)
+
+    def _internal_init(__self__,
+                 resource_name: str,
+                 opts: Optional[pulumi.ResourceOptions] = None,
+                 deployment_id: Optional[pulumi.Input[str]] = None,
+                 model_id: Optional[pulumi.Input[str]] = None,
+                 name: Optional[pulumi.Input[str]] = None,
+                 prediction_timeout: Optional[pulumi.Input[int]] = None,
+                 prompt_column_name: Optional[pulumi.Input[str]] = None,
+                 target_column_name: Optional[pulumi.Input[str]] = None,
+                 use_case_id: Optional[pulumi.Input[str]] = None,
+                 __props__=None):
+        opts = pulumi.ResourceOptions.merge(_utilities.get_resource_opts_defaults(), opts)
+        if not isinstance(opts, pulumi.ResourceOptions):
+            raise TypeError('Expected resource options to be a ResourceOptions instance')
+        if opts.id is None:
+            if __props__ is not None:
+                raise TypeError('__props__ is only valid when passed in combination with a valid opts.id to get an existing resource')
+            __props__ = CustomModelLlmValidationArgs.__new__(CustomModelLlmValidationArgs)
+
+            if deployment_id is None and not opts.urn:
+                raise TypeError("Missing required property 'deployment_id'")
+            __props__.__dict__["deployment_id"] = deployment_id
+            __props__.__dict__["model_id"] = model_id
+            __props__.__dict__["name"] = name
+            __props__.__dict__["prediction_timeout"] = prediction_timeout
+            if prompt_column_name is None and not opts.urn:
+                raise TypeError("Missing required property 'prompt_column_name'")
+            __props__.__dict__["prompt_column_name"] = prompt_column_name
+            if target_column_name is None and not opts.urn:
+                raise TypeError("Missing required property 'target_column_name'")
+            __props__.__dict__["target_column_name"] = target_column_name
+            __props__.__dict__["use_case_id"] = use_case_id
+        super(CustomModelLlmValidation, __self__).__init__(
+            'datarobot:index/customModelLlmValidation:CustomModelLlmValidation',
+            resource_name,
+            __props__,
+            opts)
+
+    @staticmethod
+    def get(resource_name: str,
+            id: pulumi.Input[str],
+            opts: Optional[pulumi.ResourceOptions] = None,
+            deployment_id: Optional[pulumi.Input[str]] = None,
+            model_id: Optional[pulumi.Input[str]] = None,
+            name: Optional[pulumi.Input[str]] = None,
+            prediction_timeout: Optional[pulumi.Input[int]] = None,
+            prompt_column_name: Optional[pulumi.Input[str]] = None,
+            target_column_name: Optional[pulumi.Input[str]] = None,
+            use_case_id: Optional[pulumi.Input[str]] = None) -> 'CustomModelLlmValidation':
+        """
+        Get an existing CustomModelLlmValidation resource's state with the given name, id, and optional extra
+        properties used to qualify the lookup.
+
+        :param str resource_name: The unique name of the resulting resource.
+        :param pulumi.Input[str] id: The unique provider ID of the resource to lookup.
+        :param pulumi.ResourceOptions opts: Options for the resource.
+        :param pulumi.Input[str] deployment_id: The ID of the custom model deployment.
+        :param pulumi.Input[str] model_id: The ID of the model used in the deployment.
+        :param pulumi.Input[str] name: The name to use for the validated custom model.
+        :param pulumi.Input[int] prediction_timeout: The timeout in seconds for the prediction when validating a custom model. Defaults to 300.
+        :param pulumi.Input[str] prompt_column_name: The name of the column the custom model uses for prompt text input.
+        :param pulumi.Input[str] target_column_name: The name of the column the custom model uses for prediction output.
+        :param pulumi.Input[str] use_case_id: The ID of the use case to associate with the validated custom model.
+        """
+        opts = pulumi.ResourceOptions.merge(opts, pulumi.ResourceOptions(id=id))
+
+        __props__ = _CustomModelLlmValidationState.__new__(_CustomModelLlmValidationState)
+
+        __props__.__dict__["deployment_id"] = deployment_id
+        __props__.__dict__["model_id"] = model_id
+        __props__.__dict__["name"] = name
+        __props__.__dict__["prediction_timeout"] = prediction_timeout
+        __props__.__dict__["prompt_column_name"] = prompt_column_name
+        __props__.__dict__["target_column_name"] = target_column_name
+        __props__.__dict__["use_case_id"] = use_case_id
+        return CustomModelLlmValidation(resource_name, opts=opts, __props__=__props__)
+
+    @property
+    @pulumi.getter(name="deploymentId")
+    def deployment_id(self) -> pulumi.Output[str]:
+        """
+        The ID of the custom model deployment.
+        """
+        return pulumi.get(self, "deployment_id")
+
+    @property
+    @pulumi.getter(name="modelId")
+    def model_id(self) -> pulumi.Output[str]:
+        """
+        The ID of the model used in the deployment.
+        """
+        return pulumi.get(self, "model_id")
+
+    @property
+    @pulumi.getter
+    def name(self) -> pulumi.Output[str]:
+        """
+        The name to use for the validated custom model.
+        """
+        return pulumi.get(self, "name")
+
+    @property
+    @pulumi.getter(name="predictionTimeout")
+    def prediction_timeout(self) -> pulumi.Output[int]:
+        """
+        The timeout in seconds for the prediction when validating a custom model. Defaults to 300.
+        """
+        return pulumi.get(self, "prediction_timeout")
+
+    @property
+    @pulumi.getter(name="promptColumnName")
+    def prompt_column_name(self) -> pulumi.Output[str]:
+        """
+        The name of the column the custom model uses for prompt text input.
+        """
+        return pulumi.get(self, "prompt_column_name")
+
+    @property
+    @pulumi.getter(name="targetColumnName")
+    def target_column_name(self) -> pulumi.Output[str]:
+        """
+        The name of the column the custom model uses for prediction output.
+        """
+        return pulumi.get(self, "target_column_name")
+
+    @property
+    @pulumi.getter(name="useCaseId")
+    def use_case_id(self) -> pulumi.Output[Optional[str]]:
+        """
+        The ID of the use case to associate with the validated custom model.
+        """
+        return pulumi.get(self, "use_case_id")
+

{pulumi_datarobot-0.8.4 → pulumi_datarobot-0.8.5}/pulumi_datarobot/llm_blueprint.py

@@ -21,9 +21,10 @@ __all__ = ['LlmBlueprintArgs', 'LlmBlueprint']
 @pulumi.input_type
 class LlmBlueprintArgs:
     def __init__(__self__, *,
-                 llm_id: pulumi.Input[str],
                  playground_id: pulumi.Input[str],
+                 custom_model_llm_settings: Optional[pulumi.Input['LlmBlueprintCustomModelLlmSettingsArgs']] = None,
                  description: Optional[pulumi.Input[str]] = None,
+                 llm_id: Optional[pulumi.Input[str]] = None,
                  llm_settings: Optional[pulumi.Input['LlmBlueprintLlmSettingsArgs']] = None,
                  name: Optional[pulumi.Input[str]] = None,
                  prompt_type: Optional[pulumi.Input[str]] = None,
@@ -31,19 +32,23 @@ class LlmBlueprintArgs:
                  vector_database_settings: Optional[pulumi.Input['LlmBlueprintVectorDatabaseSettingsArgs']] = None):
         """
         The set of arguments for constructing a LlmBlueprint resource.
-        :param pulumi.Input[str] llm_id: The id of the LLM for the LLM Blueprint.
         :param pulumi.Input[str] playground_id: The id of the Playground for the LLM Blueprint.
+        :param pulumi.Input['LlmBlueprintCustomModelLlmSettingsArgs'] custom_model_llm_settings: The custom model LLM settings for the LLM Blueprint.
         :param pulumi.Input[str] description: The description of the LLM Blueprint.
+        :param pulumi.Input[str] llm_id: The id of the LLM for the LLM Blueprint.
         :param pulumi.Input['LlmBlueprintLlmSettingsArgs'] llm_settings: The LLM settings for the LLM Blueprint.
         :param pulumi.Input[str] name: The name of the LLM Blueprint.
         :param pulumi.Input[str] prompt_type: The prompt type for the LLM Blueprint.
         :param pulumi.Input[str] vector_database_id: The id of the Vector Database for the LLM Blueprint.
         :param pulumi.Input['LlmBlueprintVectorDatabaseSettingsArgs'] vector_database_settings: The Vector Database settings for the LLM Blueprint.
         """
-        pulumi.set(__self__, "llm_id", llm_id)
         pulumi.set(__self__, "playground_id", playground_id)
+        if custom_model_llm_settings is not None:
+            pulumi.set(__self__, "custom_model_llm_settings", custom_model_llm_settings)
         if description is not None:
             pulumi.set(__self__, "description", description)
+        if llm_id is not None:
+            pulumi.set(__self__, "llm_id", llm_id)
         if llm_settings is not None:
             pulumi.set(__self__, "llm_settings", llm_settings)
         if name is not None:
@@ -55,18 +60,6 @@ class LlmBlueprintArgs:
         if vector_database_settings is not None:
             pulumi.set(__self__, "vector_database_settings", vector_database_settings)
 
-    @property
-    @pulumi.getter(name="llmId")
-    def llm_id(self) -> pulumi.Input[str]:
-        """
-        The id of the LLM for the LLM Blueprint.
-        """
-        return pulumi.get(self, "llm_id")
-
-    @llm_id.setter
-    def llm_id(self, value: pulumi.Input[str]):
-        pulumi.set(self, "llm_id", value)
-
     @property
     @pulumi.getter(name="playgroundId")
     def playground_id(self) -> pulumi.Input[str]:
@@ -79,6 +72,18 @@ class LlmBlueprintArgs:
     def playground_id(self, value: pulumi.Input[str]):
         pulumi.set(self, "playground_id", value)
 
+    @property
+    @pulumi.getter(name="customModelLlmSettings")
+    def custom_model_llm_settings(self) -> Optional[pulumi.Input['LlmBlueprintCustomModelLlmSettingsArgs']]:
+        """
+        The custom model LLM settings for the LLM Blueprint.
+        """
+        return pulumi.get(self, "custom_model_llm_settings")
+
+    @custom_model_llm_settings.setter
+    def custom_model_llm_settings(self, value: Optional[pulumi.Input['LlmBlueprintCustomModelLlmSettingsArgs']]):
+        pulumi.set(self, "custom_model_llm_settings", value)
+
     @property
     @pulumi.getter
     def description(self) -> Optional[pulumi.Input[str]]:
@@ -91,6 +96,18 @@ class LlmBlueprintArgs:
     def description(self, value: Optional[pulumi.Input[str]]):
         pulumi.set(self, "description", value)
 
+    @property
+    @pulumi.getter(name="llmId")
+    def llm_id(self) -> Optional[pulumi.Input[str]]:
+        """
+        The id of the LLM for the LLM Blueprint.
+        """
+        return pulumi.get(self, "llm_id")
+
+    @llm_id.setter
+    def llm_id(self, value: Optional[pulumi.Input[str]]):
+        pulumi.set(self, "llm_id", value)
+
     @property
     @pulumi.getter(name="llmSettings")
     def llm_settings(self) -> Optional[pulumi.Input['LlmBlueprintLlmSettingsArgs']]:
@@ -155,6 +172,7 @@ class LlmBlueprintArgs:
 @pulumi.input_type
 class _LlmBlueprintState:
     def __init__(__self__, *,
+                 custom_model_llm_settings: Optional[pulumi.Input['LlmBlueprintCustomModelLlmSettingsArgs']] = None,
                  description: Optional[pulumi.Input[str]] = None,
                  llm_id: Optional[pulumi.Input[str]] = None,
                  llm_settings: Optional[pulumi.Input['LlmBlueprintLlmSettingsArgs']] = None,
@@ -165,6 +183,7 @@ class _LlmBlueprintState:
                  vector_database_settings: Optional[pulumi.Input['LlmBlueprintVectorDatabaseSettingsArgs']] = None):
         """
         Input properties used for looking up and filtering LlmBlueprint resources.
+        :param pulumi.Input['LlmBlueprintCustomModelLlmSettingsArgs'] custom_model_llm_settings: The custom model LLM settings for the LLM Blueprint.
         :param pulumi.Input[str] description: The description of the LLM Blueprint.
         :param pulumi.Input[str] llm_id: The id of the LLM for the LLM Blueprint.
         :param pulumi.Input['LlmBlueprintLlmSettingsArgs'] llm_settings: The LLM settings for the LLM Blueprint.
@@ -174,6 +193,8 @@ class _LlmBlueprintState:
         :param pulumi.Input[str] vector_database_id: The id of the Vector Database for the LLM Blueprint.
         :param pulumi.Input['LlmBlueprintVectorDatabaseSettingsArgs'] vector_database_settings: The Vector Database settings for the LLM Blueprint.
         """
+        if custom_model_llm_settings is not None:
+            pulumi.set(__self__, "custom_model_llm_settings", custom_model_llm_settings)
         if description is not None:
             pulumi.set(__self__, "description", description)
         if llm_id is not None:
@@ -191,6 +212,18 @@ class _LlmBlueprintState:
         if vector_database_settings is not None:
             pulumi.set(__self__, "vector_database_settings", vector_database_settings)
 
+    @property
+    @pulumi.getter(name="customModelLlmSettings")
+    def custom_model_llm_settings(self) -> Optional[pulumi.Input['LlmBlueprintCustomModelLlmSettingsArgs']]:
+        """
+        The custom model LLM settings for the LLM Blueprint.
+        """
+        return pulumi.get(self, "custom_model_llm_settings")
+
+    @custom_model_llm_settings.setter
+    def custom_model_llm_settings(self, value: Optional[pulumi.Input['LlmBlueprintCustomModelLlmSettingsArgs']]):
+        pulumi.set(self, "custom_model_llm_settings", value)
+
     @property
     @pulumi.getter
     def description(self) -> Optional[pulumi.Input[str]]:
@@ -293,6 +326,7 @@ class LlmBlueprint(pulumi.CustomResource):
     def __init__(__self__,
                  resource_name: str,
                  opts: Optional[pulumi.ResourceOptions] = None,
+                 custom_model_llm_settings: Optional[pulumi.Input[Union['LlmBlueprintCustomModelLlmSettingsArgs', 'LlmBlueprintCustomModelLlmSettingsArgsDict']]] = None,
                  description: Optional[pulumi.Input[str]] = None,
                  llm_id: Optional[pulumi.Input[str]] = None,
                  llm_settings: Optional[pulumi.Input[Union['LlmBlueprintLlmSettingsArgs', 'LlmBlueprintLlmSettingsArgsDict']]] = None,
@@ -336,6 +370,7 @@ class LlmBlueprint(pulumi.CustomResource):
 
         :param str resource_name: The name of the resource.
         :param pulumi.ResourceOptions opts: Options for the resource.
+        :param pulumi.Input[Union['LlmBlueprintCustomModelLlmSettingsArgs', 'LlmBlueprintCustomModelLlmSettingsArgsDict']] custom_model_llm_settings: The custom model LLM settings for the LLM Blueprint.
         :param pulumi.Input[str] description: The description of the LLM Blueprint.
         :param pulumi.Input[str] llm_id: The id of the LLM for the LLM Blueprint.
         :param pulumi.Input[Union['LlmBlueprintLlmSettingsArgs', 'LlmBlueprintLlmSettingsArgsDict']] llm_settings: The LLM settings for the LLM Blueprint.
@@ -398,6 +433,7 @@ class LlmBlueprint(pulumi.CustomResource):
     def _internal_init(__self__,
                  resource_name: str,
                  opts: Optional[pulumi.ResourceOptions] = None,
+                 custom_model_llm_settings: Optional[pulumi.Input[Union['LlmBlueprintCustomModelLlmSettingsArgs', 'LlmBlueprintCustomModelLlmSettingsArgsDict']]] = None,
                  description: Optional[pulumi.Input[str]] = None,
                  llm_id: Optional[pulumi.Input[str]] = None,
                  llm_settings: Optional[pulumi.Input[Union['LlmBlueprintLlmSettingsArgs', 'LlmBlueprintLlmSettingsArgsDict']]] = None,
@@ -415,9 +451,8 @@ class LlmBlueprint(pulumi.CustomResource):
                 raise TypeError('__props__ is only valid when passed in combination with a valid opts.id to get an existing resource')
             __props__ = LlmBlueprintArgs.__new__(LlmBlueprintArgs)
 
+            __props__.__dict__["custom_model_llm_settings"] = custom_model_llm_settings
             __props__.__dict__["description"] = description
-            if llm_id is None and not opts.urn:
-                raise TypeError("Missing required property 'llm_id'")
             __props__.__dict__["llm_id"] = llm_id
             __props__.__dict__["llm_settings"] = llm_settings
             __props__.__dict__["name"] = name
@@ -437,6 +472,7 @@ class LlmBlueprint(pulumi.CustomResource):
     def get(resource_name: str,
             id: pulumi.Input[str],
             opts: Optional[pulumi.ResourceOptions] = None,
+            custom_model_llm_settings: Optional[pulumi.Input[Union['LlmBlueprintCustomModelLlmSettingsArgs', 'LlmBlueprintCustomModelLlmSettingsArgsDict']]] = None,
             description: Optional[pulumi.Input[str]] = None,
             llm_id: Optional[pulumi.Input[str]] = None,
             llm_settings: Optional[pulumi.Input[Union['LlmBlueprintLlmSettingsArgs', 'LlmBlueprintLlmSettingsArgsDict']]] = None,
@@ -452,6 +488,7 @@ class LlmBlueprint(pulumi.CustomResource):
         :param str resource_name: The unique name of the resulting resource.
         :param pulumi.Input[str] id: The unique provider ID of the resource to lookup.
         :param pulumi.ResourceOptions opts: Options for the resource.
+        :param pulumi.Input[Union['LlmBlueprintCustomModelLlmSettingsArgs', 'LlmBlueprintCustomModelLlmSettingsArgsDict']] custom_model_llm_settings: The custom model LLM settings for the LLM Blueprint.
         :param pulumi.Input[str] description: The description of the LLM Blueprint.
         :param pulumi.Input[str] llm_id: The id of the LLM for the LLM Blueprint.
         :param pulumi.Input[Union['LlmBlueprintLlmSettingsArgs', 'LlmBlueprintLlmSettingsArgsDict']] llm_settings: The LLM settings for the LLM Blueprint.
@@ -465,6 +502,7 @@ class LlmBlueprint(pulumi.CustomResource):
 
         __props__ = _LlmBlueprintState.__new__(_LlmBlueprintState)
 
+        __props__.__dict__["custom_model_llm_settings"] = custom_model_llm_settings
         __props__.__dict__["description"] = description
         __props__.__dict__["llm_id"] = llm_id
         __props__.__dict__["llm_settings"] = llm_settings
@@ -475,6 +513,14 @@ class LlmBlueprint(pulumi.CustomResource):
         __props__.__dict__["vector_database_settings"] = vector_database_settings
         return LlmBlueprint(resource_name, opts=opts, __props__=__props__)
 
+    @property
+    @pulumi.getter(name="customModelLlmSettings")
+    def custom_model_llm_settings(self) -> pulumi.Output[Optional['outputs.LlmBlueprintCustomModelLlmSettings']]:
+        """
+        The custom model LLM settings for the LLM Blueprint.
+        """
+        return pulumi.get(self, "custom_model_llm_settings")
+
     @property
     @pulumi.getter
     def description(self) -> pulumi.Output[Optional[str]]:
@@ -485,7 +531,7 @@ class LlmBlueprint(pulumi.CustomResource):
 
     @property
     @pulumi.getter(name="llmId")
-    def llm_id(self) -> pulumi.Output[str]:
+    def llm_id(self) -> pulumi.Output[Optional[str]]:
         """
         The id of the LLM for the LLM Blueprint.
         """

{pulumi_datarobot-0.8.4 → pulumi_datarobot-0.8.5}/pulumi_datarobot/outputs.py

@@ -77,6 +77,7 @@ __all__ = [
     'DeploymentRetrainingPolicyTriggerSchedule',
     'DeploymentRuntimeParameterValue',
     'DeploymentSegmentAnalysisSettings',
+    'LlmBlueprintCustomModelLlmSettings',
     'LlmBlueprintLlmSettings',
     'LlmBlueprintVectorDatabaseSettings',
     'NotificationChannelCustomHeader',
@@ -4014,6 +4015,70 @@ class DeploymentSegmentAnalysisSettings(dict):
         return pulumi.get(self, "attributes")
 
 
+@pulumi.output_type
+class LlmBlueprintCustomModelLlmSettings(dict):
+    @staticmethod
+    def __key_warning(key: str):
+        suggest = None
+        if key == "externalLlmContextSize":
+            suggest = "external_llm_context_size"
+        elif key == "systemPrompt":
+            suggest = "system_prompt"
+        elif key == "validationId":
+            suggest = "validation_id"
+
+        if suggest:
+            pulumi.log.warn(f"Key '{key}' not found in LlmBlueprintCustomModelLlmSettings. Access the value via the '{suggest}' property getter instead.")
+
+    def __getitem__(self, key: str) -> Any:
+        LlmBlueprintCustomModelLlmSettings.__key_warning(key)
+        return super().__getitem__(key)
+
+    def get(self, key: str, default = None) -> Any:
+        LlmBlueprintCustomModelLlmSettings.__key_warning(key)
+        return super().get(key, default)
+
+    def __init__(__self__, *,
+                 external_llm_context_size: Optional[int] = None,
+                 system_prompt: Optional[str] = None,
+                 validation_id: Optional[str] = None):
+        """
+        :param int external_llm_context_size: The external LLM's context size, in tokens. This value is only used for pruning documents supplied to the LLM when a vector database is associated with the LLM blueprint. It does not affect the external LLM's actual context size in any way and is not supplied to the LLM.
+        :param str system_prompt: System prompt guides the style of the LLM response. It is a 'universal' prompt, prepended to all individual prompts.
+        :param str validation_id: The validation ID of the custom model LLM.
+        """
+        if external_llm_context_size is not None:
+            pulumi.set(__self__, "external_llm_context_size", external_llm_context_size)
+        if system_prompt is not None:
+            pulumi.set(__self__, "system_prompt", system_prompt)
+        if validation_id is not None:
+            pulumi.set(__self__, "validation_id", validation_id)
+
+    @property
+    @pulumi.getter(name="externalLlmContextSize")
+    def external_llm_context_size(self) -> Optional[int]:
+        """
+        The external LLM's context size, in tokens. This value is only used for pruning documents supplied to the LLM when a vector database is associated with the LLM blueprint. It does not affect the external LLM's actual context size in any way and is not supplied to the LLM.
+        """
+        return pulumi.get(self, "external_llm_context_size")
+
+    @property
+    @pulumi.getter(name="systemPrompt")
+    def system_prompt(self) -> Optional[str]:
+        """
+        System prompt guides the style of the LLM response. It is a 'universal' prompt, prepended to all individual prompts.
+        """
+        return pulumi.get(self, "system_prompt")
+
+    @property
+    @pulumi.getter(name="validationId")
+    def validation_id(self) -> Optional[str]:
+        """
+        The validation ID of the custom model LLM.
+        """
+        return pulumi.get(self, "validation_id")
+
+
 @pulumi.output_type
 class LlmBlueprintLlmSettings(dict):
     @staticmethod

{pulumi_datarobot-0.8.4 → pulumi_datarobot-0.8.5}/pulumi_datarobot.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pulumi-datarobot
-Version: 0.8.4
+Version: 0.8.5
 Summary: A Pulumi package for creating and managing DataRobot resources.
 Home-page: https://www.datarobot.com
 License: Apache-2.0

{pulumi_datarobot-0.8.4 → pulumi_datarobot-0.8.5}/pulumi_datarobot.egg-info/SOURCES.txt

@@ -16,6 +16,7 @@ pulumi_datarobot/custom_metric.py
 pulumi_datarobot/custom_metric_from_job.py
 pulumi_datarobot/custom_metric_job.py
 pulumi_datarobot/custom_model.py
+pulumi_datarobot/custom_model_llm_validation.py
 pulumi_datarobot/dataset_from_datasource.py
 pulumi_datarobot/dataset_from_file.py
 pulumi_datarobot/dataset_from_url.py

{pulumi_datarobot-0.8.4 → pulumi_datarobot-0.8.5}/setup.py

@@ -8,7 +8,7 @@ from setuptools.command.install import install
 from subprocess import check_call
 
 
-VERSION = "0.8.4"
+VERSION = "0.8.5"
 def readme():
     try:
         with open('README.md', encoding='utf-8') as f: