salesforce-data-customcode 6.0.1__py3-none-any.whl → 6.0.2.dev1__py3-none-any.whl

datacustomcode/__init__.py

@@ -17,8 +17,11 @@ __all__ = [
     "AuthType",
     "Client",
     "Credentials",
+    "DefaultSparkLLMGateway",
     "PrintDataCloudWriter",
     "QueryAPIDataCloudReader",
+    "SparkLLMGateway",
+    "llm_gateway_generate_text_col",
 ]
 
 
@@ -44,4 +47,16 @@ def __getattr__(name: str):
         from datacustomcode.io.reader.query_api import QueryAPIDataCloudReader
 
         return QueryAPIDataCloudReader
+    elif name == "SparkLLMGateway":
+        from datacustomcode.llm_gateway import SparkLLMGateway
+
+        return SparkLLMGateway
+    elif name == "DefaultSparkLLMGateway":
+        from datacustomcode.llm_gateway import DefaultSparkLLMGateway
+
+        return DefaultSparkLLMGateway
+    elif name == "llm_gateway_generate_text_col":
+        from datacustomcode.client import llm_gateway_generate_text_col
+
+        return llm_gateway_generate_text_col
     raise AttributeError(f"module {__name__!r} has no attribute {name!r}")

datacustomcode/cli.py

@@ -198,11 +198,6 @@ def zip(path: str, network: str):
     default=None,
     help="SF CLI org alias or username. Fetches credentials via `sf org display`.",
 )
-@click.option(
-    "--use-in-feature",
-    default="SearchIndexChunking",
-    help="Feature where this function will be used.",
-)
 def deploy(
     path: str,
     name: str,
@@ -212,7 +207,6 @@ def deploy(
     profile: str,
     network: str,
     sf_cli_org: Optional[str],
-    use_in_feature: Optional[str],
 ):
     from datacustomcode.constants import USE_IN_FEATURE_MAPPING_FOR_CONNECT_API
     from datacustomcode.deploy import (
@@ -248,18 +242,20 @@ def deploy(
     )
 
     if package_type == "function":
-        # Try to infer use_in_feature from function signature; fall back to
-        # the explicit flag value (defaults to SearchIndexChunking)
+        # Infer use_in_feature from function signature
         entrypoint_path = os.path.join(path, ENTRYPOINT_FILE)
-        inferred = infer_use_in_feature(entrypoint_path)
-        if inferred:
-            use_in_feature = inferred
+        use_in_feature = infer_use_in_feature(entrypoint_path)
+        if use_in_feature:
             logger.info(f"Inferred use_in_feature: {use_in_feature}")
         else:
-            logger.info(f"Using use_in_feature: {use_in_feature}")
+            click.secho(
+                "Error: Could not infer function invoke options. "
+                "Please provide --use-in-feature",
+                fg="red",
+            )
+            raise click.Abort()
 
-        assert use_in_feature is not None
-        # Map feature names to Connect API names
+        # Map user-provided feature names to API names
         mapped_feature = USE_IN_FEATURE_MAPPING_FOR_CONNECT_API.get(
             use_in_feature, use_in_feature
         )

datacustomcode/client.py

@@ -18,24 +18,80 @@ from enum import Enum
 from typing import (
     TYPE_CHECKING,
     ClassVar,
+    Dict,
     Optional,
+    Union,
 )
 
 from datacustomcode.config import config
 from datacustomcode.file.path.default import DefaultFindFilePath
 from datacustomcode.io.reader.base import BaseDataCloudReader
+from datacustomcode.llm_gateway_config import spark_llm_gateway_config
 from datacustomcode.spark.default import DefaultSparkSessionProvider
 
 if TYPE_CHECKING:
     from pathlib import Path
 
-    from pyspark.sql import DataFrame as PySparkDataFrame
+    from pyspark.sql import Column, DataFrame as PySparkDataFrame
 
     from datacustomcode.io.reader.base import BaseDataCloudReader
     from datacustomcode.io.writer.base import BaseDataCloudWriter, WriteMode
+    from datacustomcode.llm_gateway.spark_base import SparkLLMGateway
     from datacustomcode.spark.base import BaseSparkSessionProvider
 
 
+def _build_spark_llm_gateway() -> "SparkLLMGateway":
+    """Instantiate the SDK-configured :class:`SparkLLMGateway`.
+
+    Raises:
+        RuntimeError: If no ``spark_llm_gateway_config`` has been loaded.
+    """
+    cfg = spark_llm_gateway_config.spark_llm_gateway_config
+    if cfg is None:
+        raise RuntimeError(
+            "spark_llm_gateway_config is not configured. Add a "
+            "'spark_llm_gateway_config' section to config.yaml."
+        )
+    return cfg.to_object()
+
+
+def llm_gateway_generate_text_col(
+    template: str,
+    values: Union[Dict[str, "Column"], "Column"],
+    model_id: Optional[str] = None,
+    max_tokens: Optional[int] = None,
+) -> "Column":
+    """Build a Spark Column that runs the LLM Gateway per row.
+
+    Example:
+
+        >>> df.withColumn(
+        ...     "greeting__c",
+        ...     llm_gateway_generate_text_col(
+        ...         "In one sentence, greet {name} from {city}.",
+        ...         {"name": col("name__c"), "city": col("homecity__c")},
+        ...         model_id="sfdc_ai__DefaultGPT4Omni",
+        ...         max_tokens=100,
+        ...     ),
+        ... )
+
+    Args:
+        template: The prompt template, with ``{field}`` placeholders matching
+            keys in ``values``. Substitution uses ``str.format``.
+        values: Either a mapping from placeholder name to Spark ``Column``, or
+            a single ``Column`` whose value is already a struct.
+        model_id: LLM model id. Defaults to ``sfdc_ai__DefaultGPT4Omni``.
+        max_tokens: Maximum tokens to generate. Defaults to 200.
+
+    Returns:
+        A Spark ``Column`` that, when evaluated, produces the generated text.
+    """
+    gateway = Client()._get_spark_llm_gateway()
+    return gateway.llm_gateway_generate_text_col(
+        template, values, model_id=model_id, max_tokens=max_tokens
+    )
+
+
 class DataCloudObjectType(Enum):
     DLO = "dlo"
     DMO = "dmo"
@@ -94,18 +150,23 @@ class Client:
         finder: Find a file path
         reader: A custom reader to use for reading Data Cloud objects.
         writer: A custom writer to use for writing Data Cloud objects.
+        spark_llm_gateway: Optional custom :class:`SparkLLMGateway`. When
+            omitted, the gateway is lazily resolved from
+            ``spark_llm_gateway_config``.
 
     Example:
     >>> client = Client()
     >>> file_path = client.find_file_path("data.csv")
     >>> dlo = client.read_dlo("my_dlo")
     >>> client.write_to_dmo("my_dmo", dlo)
+    >>> answer = client.llm_gateway_generate_text("Generate a greeting message")
     """
 
     _instance: ClassVar[Optional[Client]] = None
     _reader: BaseDataCloudReader
     _writer: BaseDataCloudWriter
     _file: DefaultFindFilePath
+    _spark_llm_gateway: Optional[SparkLLMGateway]
     _data_layer_history: dict[DataCloudObjectType, set[str]]
     _code_type: str
 
@@ -114,11 +175,13 @@ class Client:
         reader: Optional[BaseDataCloudReader] = None,
         writer: Optional[BaseDataCloudWriter] = None,
         spark_provider: Optional[BaseSparkSessionProvider] = None,
+        spark_llm_gateway: Optional[SparkLLMGateway] = None,
         code_type: str = "script",
     ) -> Client:
 
         if cls._instance is None:
             cls._instance = super().__new__(cls)
+            cls._instance._spark_llm_gateway = spark_llm_gateway
             # Initialize Readers and Writers from config
             # and/or provided reader and writer
             if reader is None or writer is None:
@@ -225,6 +288,44 @@ class Client:
 
         return self._file.find_file_path(file_name)  # type: ignore[no-any-return]
 
+    def llm_gateway_generate_text(
+        self,
+        prompt: str,
+        model_id: Optional[str] = None,
+        max_tokens: Optional[int] = None,
+    ) -> str:
+        """Issue a one-shot LLM Gateway call. This is the scalar counterpart to
+        :func:`llm_gateway_generate_text_col`: it runs **once**  — not per row.
+        Use the column helper method instead when you want to fan a prompt out across
+        every row of a DataFrame.
+
+        Example:
+
+            >>> response = Client().llm_gateway_generate_text(
+            ...     "Generate a greeting message"
+            ... )
+
+        Args:
+            prompt: The literal prompt to send. Plain text — no
+                ``{field}`` substitution is performed on this string.
+            model_id: LLM model id to target. Defaults to
+                ``sfdc_ai__DefaultGPT4Omni`` when ``None``.
+            max_tokens: Hard upper bound on the number of tokens the model
+                may generate. Defaults to 200 when ``None``.
+
+        Returns:
+            The generated text as a plain Python ``str``; empty when the
+            gateway response carries no generated text.
+        """
+        return self._get_spark_llm_gateway().llm_gateway_generate_text(
+            prompt, model_id=model_id, max_tokens=max_tokens
+        )
+
+    def _get_spark_llm_gateway(self) -> SparkLLMGateway:
+        if self._spark_llm_gateway is None:
+            self._spark_llm_gateway = _build_spark_llm_gateway()
+        return self._spark_llm_gateway
+
     def _validate_data_layer_history_does_not_contain(
         self, data_cloud_object_type: DataCloudObjectType
     ) -> None:

datacustomcode/config.yaml

@@ -28,3 +28,6 @@ llm_gateway_config:
   type_config_name: DefaultLLMGateway
   options:
     credentials_profile: default
+
+spark_llm_gateway_config:
+  type_config_name: DefaultSparkLLMGateway

datacustomcode/einstein_platform_config.py

@@ -15,20 +15,23 @@
 
 from typing import (
     ClassVar,
+    Generic,
     Optional,
     Type,
-    cast,
+    TypeVar,
 )
 
 from datacustomcode.common_config import BaseObjectConfig
 
+_T = TypeVar("_T")
 
-class CredentialsObjectConfig(BaseObjectConfig):
+
+class CredentialsObjectConfig(BaseObjectConfig, Generic[_T]):
     type_to_create: ClassVar[Type]
     credentials_profile: Optional[str] = None
     sf_cli_org: Optional[str] = None
 
-    def to_object(self):
+    def to_object(self) -> _T:
         """Create an object instance, automatically including credentials in options"""
 
         options = self.options.copy()
@@ -38,4 +41,5 @@ class CredentialsObjectConfig(BaseObjectConfig):
             options["sf_cli_org"] = self.sf_cli_org
 
         type_ = self.type_to_create.subclass_from_config_name(self.type_config_name)
-        return cast(type_, type_(**options))
+        instance: _T = type_(**options)
+        return instance

datacustomcode/einstein_predictions_config.py

@@ -28,7 +28,7 @@ from datacustomcode.einstein_predictions.base import EinsteinPredictions
 _E = TypeVar("_E", bound=EinsteinPredictions)
 
 
-class EinsteinPredictionsObjectConfig(CredentialsObjectConfig, Generic[_E]):
+class EinsteinPredictionsObjectConfig(CredentialsObjectConfig[_E], Generic[_E]):
     type_to_create: ClassVar[Type[EinsteinPredictions]] = EinsteinPredictions  # type: ignore[type-abstract]
 
 

datacustomcode/function/feature_types/chunking.py

@@ -16,6 +16,7 @@
 """
 Pydantic models for Search Index Chunking V1
 """
+
 from enum import Enum
 from typing import (
     Dict,

datacustomcode/io/writer/base.py

@@ -27,21 +27,11 @@ if TYPE_CHECKING:
 class WriteMode(str, Enum):
     APPEND = "append"
     OVERWRITE = "overwrite"
-    OVERWRITE_PARTITIONS = "overwrite_partitions"  # Deprecated: raises error if used
+    OVERWRITE_PARTITIONS = "overwrite_partitions"
     MERGE = "merge"
     MERGE_UPSERT_DELETE = "merge_upsert_delete"
 
 
-class MergeRecordType(str, Enum):
-    """Values for the _merge_record_type column used by MERGE_UPSERT_DELETE."""
-
-    UPSERT = "UPSERT"
-    DELETE = "DELETE"
-
-
-MERGE_RECORD_TYPE_COLUMN = "_merge_record_type"
-
-
 class BaseDataCloudWriter(BaseDataAccessLayer):
     """Base class for Data Cloud writers."""
 

datacustomcode/llm_gateway/__init__.py

@@ -15,8 +15,12 @@
 
 from datacustomcode.llm_gateway.base import LLMGateway
 from datacustomcode.llm_gateway.default import DefaultLLMGateway
+from datacustomcode.llm_gateway.spark_base import SparkLLMGateway
+from datacustomcode.llm_gateway.spark_default import DefaultSparkLLMGateway
 
 __all__ = [
     "DefaultLLMGateway",
+    "DefaultSparkLLMGateway",
     "LLMGateway",
+    "SparkLLMGateway",
 ]

datacustomcode/llm_gateway/default.py

@@ -34,6 +34,8 @@ class DefaultLLMGateway(EinsteinPlatformClient, LLMGateway):
 
         payload: Dict[str, Any] = {"prompt": request.prompt}
 
+        if request.max_tokens is not None:
+            payload["max_tokens"] = request.max_tokens
         if request.localization:
             payload["localization"] = request.localization
         if request.tags:

datacustomcode/llm_gateway/spark_base.py

@@ -0,0 +1,55 @@
+# Copyright (c) 2025, Salesforce, Inc.
+# SPDX-License-Identifier: Apache-2
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+from __future__ import annotations
+
+from abc import ABC, abstractmethod
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    Dict,
+    Optional,
+    Union,
+)
+
+from datacustomcode.mixin import UserExtendableNamedConfigMixin
+
+if TYPE_CHECKING:
+    from pyspark.sql import Column
+
+
+class SparkLLMGateway(ABC, UserExtendableNamedConfigMixin):
+    CONFIG_NAME: str
+
+    def __init__(self, **kwargs: Any) -> None:
+        pass
+
+    @abstractmethod
+    def llm_gateway_generate_text(
+        self,
+        prompt: str,
+        model_id: Optional[str] = None,
+        max_tokens: Optional[int] = None,
+    ) -> str:
+        """Issue a one-shot LLM Gateway call and return the generated text."""
+
+    @abstractmethod
+    def llm_gateway_generate_text_col(
+        self,
+        template: str,
+        values: Union[Dict[str, "Column"], "Column"],
+        model_id: Optional[str] = None,
+        max_tokens: Optional[int] = None,
+    ) -> "Column":
+        """Build a Spark ``Column`` that invokes the LLM Gateway per row."""

datacustomcode/llm_gateway/spark_default.py

@@ -0,0 +1,119 @@
+# Copyright (c) 2025, Salesforce, Inc.
+# SPDX-License-Identifier: Apache-2
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+from __future__ import annotations
+
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    Dict,
+    Optional,
+    Union,
+)
+
+from datacustomcode.llm_gateway.spark_base import SparkLLMGateway
+
+if TYPE_CHECKING:
+    from pyspark.sql import Column
+
+    from datacustomcode.llm_gateway.base import LLMGateway
+
+
+_DEFAULT_LLM_MODEL_ID = "sfdc_ai__DefaultGPT4Omni"
+_DEFAULT_LLM_MAX_TOKENS = 200
+
+
+class DefaultSparkLLMGateway(SparkLLMGateway):
+
+    CONFIG_NAME = "DefaultSparkLLMGateway"
+
+    def __init__(
+        self,
+        llm_gateway: Optional["LLMGateway"] = None,
+        **kwargs: Any,
+    ) -> None:
+        super().__init__(**kwargs)
+        if llm_gateway is None:
+            llm_gateway = _build_underlying_gateway()
+        self._llm_gateway: "LLMGateway" = llm_gateway
+
+    def llm_gateway_generate_text(
+        self,
+        prompt: str,
+        model_id: Optional[str] = None,
+        max_tokens: Optional[int] = None,
+    ) -> str:
+        return _invoke_llm_gateway(self._llm_gateway, prompt, model_id, max_tokens)
+
+    def llm_gateway_generate_text_col(
+        self,
+        template: str,
+        values: Union[Dict[str, "Column"], "Column"],
+        model_id: Optional[str] = None,
+        max_tokens: Optional[int] = None,
+    ) -> "Column":
+
+        from pyspark.sql.functions import struct, udf
+        from pyspark.sql.types import StringType
+
+        if isinstance(values, dict):
+            values_col = struct(*[v.alias(k) for k, v in values.items()])
+        else:
+            values_col = values
+
+        gateway = self._llm_gateway
+
+        def _generate(values_row: Any) -> str:
+            if values_row is None:
+                return ""
+            subs = (
+                values_row.asDict()
+                if hasattr(values_row, "asDict")
+                else dict(values_row)
+            )
+            prompt = template.format(**subs)
+            return _invoke_llm_gateway(gateway, prompt, model_id, max_tokens)
+
+        return udf(_generate, StringType())(values_col)
+
+
+def _build_underlying_gateway() -> "LLMGateway":
+    from datacustomcode.llm_gateway_config import llm_gateway_config
+
+    cfg = llm_gateway_config.llm_gateway_config
+    if cfg is None:
+        raise RuntimeError(
+            "llm_gateway_config is not configured. Add an 'llm_gateway_config' "
+            "section to config.yaml."
+        )
+    return cfg.to_object()
+
+
+def _invoke_llm_gateway(
+    gateway: "LLMGateway",
+    prompt: str,
+    model_id: Optional[str],
+    max_tokens: Optional[int],
+) -> str:
+    from datacustomcode.llm_gateway.types.generate_text_request_builder import (
+        GenerateTextRequestBuilder,
+    )
+
+    builder = (
+        GenerateTextRequestBuilder()
+        .set_prompt(prompt)
+        .set_model(model_id or _DEFAULT_LLM_MODEL_ID)
+        .set_max_tokens(max_tokens or _DEFAULT_LLM_MAX_TOKENS)
+    )
+    return gateway.generate_text(builder.build()).text

datacustomcode/llm_gateway/types/generate_text_request.py

@@ -40,6 +40,13 @@ class GenerateTextRequest(BaseModel):
     )
     model_name: str = Field(..., min_length=1, description="Name of the model to use")
     prompt: str = Field(..., description="Input prompt")
+    max_tokens: Optional[int] = Field(
+        default=None,
+        ge=1,
+        description=(
+            "Maximum number of tokens to generate. If None, server default applies."
+        ),
+    )
     localization: Optional[Dict[str, Any]] = Field(
         default=None, description="Localization settings"
     )

datacustomcode/llm_gateway/types/generate_text_request_builder.py

@@ -26,6 +26,7 @@ class GenerateTextRequestBuilder:
     def __init__(self) -> None:
         self._prompt = ""
         self._model_name = ""
+        self._max_tokens: Optional[int] = None
         self._localization: Optional[Dict[str, Any]] = None
         self._tags: Optional[Dict[str, Any]] = None
 
@@ -37,6 +38,10 @@ class GenerateTextRequestBuilder:
         self._model_name = model_name
         return self
 
+    def set_max_tokens(self, max_tokens: int) -> "GenerateTextRequestBuilder":
+        self._max_tokens = max_tokens
+        return self
+
     def set_localization(
         self,
         localization: Optional[Dict[str, Any]] = None,
@@ -75,6 +80,7 @@ class GenerateTextRequestBuilder:
         request = GenerateTextRequest(
             prompt=self._prompt,
             model_name=self._model_name,
+            max_tokens=self._max_tokens,
             localization=self._localization,
             tags=self._tags,
         )

datacustomcode/llm_gateway_config.py

@@ -21,14 +21,20 @@ from typing import (
     Union,
 )
 
-from datacustomcode.common_config import BaseConfig, default_config_file
+from datacustomcode.common_config import (
+    BaseConfig,
+    BaseObjectConfig,
+    default_config_file,
+)
 from datacustomcode.einstein_platform_config import CredentialsObjectConfig
 from datacustomcode.llm_gateway.base import LLMGateway
+from datacustomcode.llm_gateway.spark_base import SparkLLMGateway
 
 _E = TypeVar("_E", bound=LLMGateway)
+_S = TypeVar("_S", bound=SparkLLMGateway)
 
 
-class LLMGatewayObjectConfig(CredentialsObjectConfig, Generic[_E]):
+class LLMGatewayObjectConfig(CredentialsObjectConfig[_E], Generic[_E]):
     type_to_create: ClassVar[Type[LLMGateway]] = LLMGateway  # type: ignore[type-abstract]
 
 
@@ -52,6 +58,41 @@ class LLMGatewayConfig(BaseConfig):
         return self
 
 
+class SparkLLMGatewayObjectConfig(BaseObjectConfig, Generic[_S]):
+    type_to_create: ClassVar[Type[SparkLLMGateway]] = SparkLLMGateway  # type: ignore[type-abstract]
+
+    def to_object(self) -> SparkLLMGateway:
+        type_ = self.type_to_create.subclass_from_config_name(self.type_config_name)
+        return type_(**self.options)
+
+
+class SparkLLMGatewayConfig(BaseConfig):
+    spark_llm_gateway_config: Union[
+        SparkLLMGatewayObjectConfig[SparkLLMGateway], None
+    ] = None
+
+    def update(self, other: "SparkLLMGatewayConfig") -> "SparkLLMGatewayConfig":
+        def merge(
+            config_a: Union[SparkLLMGatewayObjectConfig, None],
+            config_b: Union[SparkLLMGatewayObjectConfig, None],
+        ) -> Union[SparkLLMGatewayObjectConfig, None]:
+            if config_a is not None and config_a.force:
+                return config_a
+            if config_b:
+                return config_b
+            return config_a
+
+        self.spark_llm_gateway_config = merge(
+            self.spark_llm_gateway_config, other.spark_llm_gateway_config
+        )
+        return self
+
+
 # Global LLM Gateway config instance
 llm_gateway_config = LLMGatewayConfig()
 llm_gateway_config.load(default_config_file())
+
+
+# Global Spark LLM Gateway config instance
+spark_llm_gateway_config = SparkLLMGatewayConfig()
+spark_llm_gateway_config.load(default_config_file())

datacustomcode/templates/function/.devcontainer/devcontainer.json

@@ -5,6 +5,6 @@
 		"dockerfile": "../Dockerfile"
 	},
 	"features": {
-		"ghcr.io/devcontainers/features/git:1": {}
+		"ghcr.io/devcontainers/features/git:1": {},
 	}
 }

datacustomcode/templates/function/example/chunking_with_llm/entrypoint.py

@@ -7,6 +7,9 @@ This function demonstrates the new signature-based invocation with Pydantic mode
 - Requires Runtime parameter (for agentic capabilities)
 - Type-safe with direct field access (no wrappers)
 - Automatic validation and conversion
+
+You can use your AI models configured in Salesforce to generate texts.
+See README.md for how to test locally before deploying to Data Cloud.
 """
 
 import logging

datacustomcode/templates/function/example/chunking_with_prediction/entrypoint.py

@@ -12,6 +12,9 @@ Model: YH_Regression_Python_Predicted_SalePrice_CM_12l_ATC937af934
 Type: Regression
 Input: Year_Built__c (numeric)
 Output: Predicted_SalePrice
+
+You can use your AI models configured in Salesforce to make predictions.
+See README.md for how to test locally before deploying to Data Cloud.
 """
 
 import logging

datacustomcode/templates/function/payload/entrypoint.py

@@ -65,13 +65,9 @@ def make_einstein_prediction(runtime: Runtime) -> None:
     )
 
 
-def generate_text(runtime: Runtime):
+def generate_text(runtime: Runtime, prompt: str, model: str = "sfdc_ai__DefaultGPT52"):
     builder = GenerateTextRequestBuilder()
-    llm_request = (
-        builder.set_prompt("Generate 2 dog names")
-        .set_model("sfdc_ai__DefaultGPT52")
-        .build()
-    )
+    llm_request = builder.set_prompt(prompt).set_model(model).build()
     llm_response = runtime.llm_gateway.generate_text(llm_request)
     logger.info(
         f"LLM Gateway generate text results - success: [{llm_response.is_success}] "
@@ -88,13 +84,16 @@ def function(request: dict, runtime: Runtime) -> dict:
     current_seq_no = 1  # Start sequence number from 1
 
     """
-    You can use your AI models configured in Salesforce
-    to generate texts or predict an outcome.
-    First configure an external client app before using these AI APIs
-    https://developer.salesforce.com/docs/ai/agentforce/guide/agent-api-get-started.html#create-a-salesforce-app"
+    You can use your AI models configured in Salesforce to generate texts
+    or predict an outcome. See README.md for how to test locally before
+    deploying to Data Cloud.
+
+    Example:
+
+        >>> generated_text = generate_text(runtime, "Generate a greeting message")
+        ... prediction = make_einstein_prediction(runtime)
+
     """
-    # generate_text(runtime)
-    # make_einstein_prediction(runtime)
 
     for item in items:
         # Item is DocElement as dict

datacustomcode/templates/script/.devcontainer/devcontainer.json

@@ -5,6 +5,6 @@
 		"dockerfile": "../Dockerfile"
 	},
 	"features": {
-		"ghcr.io/devcontainers/features/git:1": {}
+		"ghcr.io/devcontainers/features/git:1": {},
 	}
 }

datacustomcode/templates/script/payload/entrypoint.py

@@ -12,6 +12,33 @@ def main():
     # Perform transformations on the DataFrame
     df_upper1 = df.withColumn("description__c", upper(col("description__c")))
 
+    """
+    You can use your AI models configured in Salesforce to generate column
+    values. See README.md for how to test locally before deploying to Data Cloud.
+
+    Example:
+
+        >>> from datacustomcode.client import llm_gateway_generate_text_col
+            df_generated = df.withColumn(
+            ...     "greeting__c",
+            ...     llm_gateway_generate_text_col(
+            ...         "In one sentence, greet {name} from {city}.",
+            ...         {"name": col("name__c"), "city": col("homecity__c")},
+            ...         model_id="sfdc_ai__DefaultGPT4Omni",
+            ...         max_tokens=100,
+            ...     ),
+            ... )
+
+    You can also invoke the LLM with a literal plain text prompt — no
+    ``{field}`` substitution is performed on this string.
+
+    Example:
+
+        >>> generated_text = client.llm_gateway_generate_text(
+        ...     prompt, model_id, max_tokens
+        ... )
+    """
+
     # Drop specific columns related to relationships
     df_upper1 = df_upper1.drop("sfdcorganizationid__c")
     df_upper1 = df_upper1.drop("kq_id__c")

datacustomcode/token_provider.py

@@ -101,75 +101,48 @@ class SFCLITokenProvider(TokenProvider):
 
         from datacustomcode.deploy import AccessTokenResponse
 
-        def _run_sf_command(args: list[str], description: str) -> dict:
-            try:
-                result = subprocess.run(
-                    args,
-                    capture_output=True,
-                    text=True,
-                    check=True,
-                    timeout=30,
-                )
-            except FileNotFoundError as exc:
-                raise RuntimeError(
-                    "The 'sf' command was not found. "
-                    "Install Salesforce CLI: "
-                    "https://developer.salesforce.com/tools/salesforcecli"
-                ) from exc
-            except subprocess.TimeoutExpired as exc:
-                raise RuntimeError(
-                    f"'{description}' timed out for org '{self.sf_cli_org}'"
-                ) from exc
-            except subprocess.CalledProcessError as exc:
-                raise RuntimeError(
-                    f"'{description}' failed for org '{self.sf_cli_org}': "
-                    f"{exc.stderr}"
-                ) from exc
+        try:
+            result = subprocess.run(
+                ["sf", "org", "display", "--target-org", self.sf_cli_org, "--json"],
+                capture_output=True,
+                text=True,
+                check=True,
+                timeout=30,
+            )
+        except FileNotFoundError as exc:
+            raise RuntimeError(
+                "The 'sf' command was not found. "
+                "Install Salesforce CLI: https://developer.salesforce.com/tools/salesforcecli"
+            ) from exc
+        except subprocess.TimeoutExpired as exc:
+            raise RuntimeError(
+                f"'sf org display' timed out for org '{self.sf_cli_org}'"
+            ) from exc
+        except subprocess.CalledProcessError as exc:
+            raise RuntimeError(
+                f"'sf org display' failed for org '{self.sf_cli_org}': {exc.stderr}"
+            ) from exc
 
-            try:
-                data = json.loads(result.stdout)
-            except json.JSONDecodeError as exc:
-                raise RuntimeError(
-                    f"Failed to parse JSON from '{description}': {result.stdout}"
-                ) from exc
+        try:
+            data = json.loads(result.stdout)
+        except json.JSONDecodeError as exc:
+            raise RuntimeError(
+                f"Failed to parse JSON from 'sf org display': {result.stdout}"
+            ) from exc
 
-            if data.get("status") != 0:
-                raise RuntimeError(
-                    f"SF CLI error for org '{self.sf_cli_org}': "
-                    f"{data.get('message', 'unknown error')}"
-                )
-            return dict(data)
-
-        # Get instanceUrl from sf org display
-        display_data = _run_sf_command(
-            ["sf", "org", "display", "--target-org", self.sf_cli_org, "--json"],
-            "sf org display",
-        )
-        instance_url = display_data.get("result", {}).get("instanceUrl")
-        if not instance_url:
+        if data.get("status") != 0:
             raise RuntimeError(
-                f"'sf org display' did not return an instance URL "
-                f"for org '{self.sf_cli_org}'"
+                f"SF CLI error for org '{self.sf_cli_org}': "
+                f"{data.get('message', 'unknown error')}"
             )
 
-        # Get access token via show-access-token (newer SF CLI versions
-        # redact the token in sf org display)
-        token_data = _run_sf_command(
-            [
-                "sf",
-                "org",
-                "auth",
-                "show-access-token",
-                "--target-org",
-                self.sf_cli_org,
-                "--json",
-            ],
-            "sf org auth show-access-token",
-        )
-        access_token = token_data.get("result", {}).get("accessToken")
-        if not access_token:
+        result_data = data.get("result", {})
+        access_token = result_data.get("accessToken")
+        instance_url = result_data.get("instanceUrl")
+
+        if not access_token or not instance_url:
             raise RuntimeError(
-                f"'sf org auth show-access-token' did not return an access token "
+                f"'sf org display' did not return an access token or instance URL "
                 f"for org '{self.sf_cli_org}'"
             )
 

{salesforce_data_customcode-6.0.1.dist-info → salesforce_data_customcode-6.0.2.dev1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: salesforce-data-customcode
-Version: 6.0.1
+Version: 6.0.2.dev1
 Summary: Data Cloud Custom Code SDK
 License-Expression: Apache-2.0
 License-File: LICENSE.txt
@@ -330,6 +330,53 @@ Options:
 - `--function-invoke-opt TEXT`: Currently we support only `UnstructuredChunking` for functions.
 
 
+## Testing LLM Gateway
+
+You can use AI models configured in Salesforce to generate responses while transforming your data. Below is a sample code example:
+
+```
+from datacustomcode.client import Client, llm_gateway_generate_text_col
+
+
+def main():
+  client = Client()
+  df = client.read_dlo("Input__dll")
+  df_generated = df.withColumn(
+    "greeting__c",
+    llm_gateway_generate_text_col(
+        "In one sentence, greet {name} from {city}.",
+        {"name": col("name__c"), "city": col("homecity__c")},
+        model_id="sfdc_ai__DefaultGPT4Omni", # An AI model in your org
+        max_tokens=100,
+    ),
+  )
+
+  dlo_name = "Output_dll"
+  client.write_to_dlo(dlo_name, df_upper1, write_mode=WriteMode.APPEND)
+
+  greeting = client.llm_gateway_generate_text("In one sentence, generate a greeting message", "sfdc_ai__DefaultGPT52")
+
+if __name__ == "__main__":
+  main()
+```
+
+In order to test this code on your local machine before deploying it to Data Cloud, you must first set up an External Client App that allows access to the Agent API. Follow this guide to create the ECA https://developer.salesforce.com/docs/ai/agentforce/guide/agent-api-get-started.html#create-a-salesforce-app. You must use `http://localhost:1717/OauthRedirect` as the callback URL.
+
+Once the ECA is set up, log in to your org using this ECA
+```
+sf org login web \
+  --alias myorg \
+  --instance-url https://{MY_DOMAIN_URL} \
+  --client-id {CONSUMER_KEY} \
+  --scopes "sfap_api api"
+```
+
+then you can test your code using `myorg` alias
+```
+datacustomcode run ./payload/entrypoint.py --sf-cli-org myorg
+```
+
+
 ## Docker usage
 
 The SDK provides Docker-based development options that allow you to test your code in an environment that closely resembles Data Cloud's execution environment.

{salesforce_data_customcode-6.0.1.dist-info → salesforce_data_customcode-6.0.2.dev1.dist-info}/RECORD

@@ -1,21 +1,21 @@
-datacustomcode/__init__.py,sha256=Fj2pIYQqLV_kQrMywrYPj0NsMtHjx4tkvwyqSt-m0E0,1530
+datacustomcode/__init__.py,sha256=qx-X3U9lU_791ZbAxhGqQeyDw6eqTi9-E9AbYnwJfxU,2071
 datacustomcode/auth.py,sha256=fpSjhIBdv9trC8yq2vuljAix_Euu-4Ah7HDCGhYjOxI,8309
-datacustomcode/cli.py,sha256=tAtSSUMoVQtME_y1fxpNq4Lw1vfQsP1a6UxlEPiV9Sw,13237
-datacustomcode/client.py,sha256=TO9TkIsJjCEMlQMnD-NBGF26fk2K9jpOZjOT_XWKJ0g,9340
+datacustomcode/cli.py,sha256=ObqhOasZSZ8ZNFP6pywV3FpsUOz45_KaZdMNT4BHOKs,13069
+datacustomcode/client.py,sha256=prAufc2_KAKdy0XX-nYhuD22a93GqQCsQuql9ru-8dY,13294
 datacustomcode/cmd.py,sha256=ZMs46aydJw2EaU26JgCtZmnqESQFHvvaJz10hnjZTBk,3537
 datacustomcode/common_config.py,sha256=SAUnxj3kqmOeWwPmFoYq4tuxokMgURVm4QwcGi-avL4,1928
 datacustomcode/config.py,sha256=2Pk61ieQsEWSxKDxlS66_rliKE8iX0j-9LweEm0aaGo,4072
-datacustomcode/config.yaml,sha256=iMs3HPMLHwnesJ9KRD0-CpJqtHRehQvq2w2od9uE0aU,775
+datacustomcode/config.yaml,sha256=Wd8Zd2f1__hRv7166H5IBbU5YoOEOdQNHgAb4ONbFnA,845
 datacustomcode/constants.py,sha256=SskQpUPnQE7LjKh4Z3AdZ2p191_0Kf73jV8gdBXDPLU,1436
 datacustomcode/credentials.py,sha256=D-7Zd3Nh_wStdj8_wUy5cnC8M_mdbfBijhIAi-2EbXs,9467
 datacustomcode/deploy.py,sha256=9yaekNew5ha6bd6rIFERknFROomOeIfikrKXN6jCoq4,18811
 datacustomcode/einstein_platform_client.py,sha256=ON2B_m--vdbCVVMVcLc81T-BRZ5-UuxVCer8E6OEQPA,4083
-datacustomcode/einstein_platform_config.py,sha256=EYnAKz0z1hNN9A8GHtaYm-Ek4GqUi68iYrJ1TSR-J9w,1416
+datacustomcode/einstein_platform_config.py,sha256=6lb_FRbEzdt5a8-2cR15f13ZKw674uvRJ4Xq0o_pQXE,1490
 datacustomcode/einstein_predictions/__init__.py,sha256=o-iask36phwGA7hGi_sxJpwbITO6h_STDe0J92qD2_M,859
 datacustomcode/einstein_predictions/base.py,sha256=OBV445dpZRtCF5cpdHH49w_6RU_jZY1A-e7VtHTQCFU,1061
 datacustomcode/einstein_predictions/impl/default.py,sha256=fEFhrosozhG0yT87w4Gu8wbe1NgHxBqdAduwWHdWGN4,3011
 datacustomcode/einstein_predictions/types.py,sha256=W7H3sFzm_NdBtnia7O576w0OvM7jwzfGVQmFiZTO8Og,6098
-datacustomcode/einstein_predictions_config.py,sha256=dwY3Q5MFBHm60lL27m-Srt5BpcURc_PuKpaW-ALOqw4,2131
+datacustomcode/einstein_predictions_config.py,sha256=oxKgp2u7t5SdHVsasBBxsvd3KAfXuZ9uQXTrtpcCUlU,2135
 datacustomcode/file/__init__.py,sha256=gamfOD1VnAtEslRBpqh-yKiVjkG_wYWYdSatGIfsN-w,621
 datacustomcode/file/base.py,sha256=2IUlQufPvpHccgyoeZlKhcktO4pS4NKdha03Whq33Zk,745
 datacustomcode/file/path/__init__.py,sha256=gamfOD1VnAtEslRBpqh-yKiVjkG_wYWYdSatGIfsN-w,621
@@ -23,7 +23,7 @@ datacustomcode/file/path/default.py,sha256=jyDv61JUlD5H6lUPAp55_B2toMryKItA8Hu2k
 datacustomcode/function/__init__.py,sha256=oag8FevDxbx1vR0tFhtxPd5SvmUEYDj8esRo4v2qu3Q,749
 datacustomcode/function/base.py,sha256=H_i80GcsFScvHJJ6bYLry5phTpsEU1KubpR8D3qnsjo,691
 datacustomcode/function/feature_types/__init__.py,sha256=gamfOD1VnAtEslRBpqh-yKiVjkG_wYWYdSatGIfsN-w,621
-datacustomcode/function/feature_types/chunking.py,sha256=ftP90Wp5w7V5WYDIpgD-eIyCRQr7VJUlfnrlvwrZZ8w,7179
+datacustomcode/function/feature_types/chunking.py,sha256=_tCCHj7slD59-aLLEZUFRPOX37q_MEKkL4thi9CNwMw,7180
 datacustomcode/function/runtime.py,sha256=jUduf-pve-0JAv0Lg0HPyLCZ9_Hgcxe25dI7t-DgX9o,3696
 datacustomcode/function_utils.py,sha256=y6vaoSSaJ9CeHhjLCNyqzJT4vx6yCePWRxNLz38ddjQ,12920
 datacustomcode/io/__init__.py,sha256=gamfOD1VnAtEslRBpqh-yKiVjkG_wYWYdSatGIfsN-w,621
@@ -34,23 +34,21 @@ datacustomcode/io/reader/query_api.py,sha256=eVrohrcnTnhSMsGfPRHu5XltjFtGG0hyHt1
 datacustomcode/io/reader/sf_cli.py,sha256=x5QacVqRZaZSyph1_wwxo67s8r29wsOcUsOaiF9cDWE,6324
 datacustomcode/io/reader/utils.py,sha256=HlHhPZoHfmWA3mF8kTnd3m4Nd2exaz4NsOJJfQY7Pew,1656
 datacustomcode/io/writer/__init__.py,sha256=gamfOD1VnAtEslRBpqh-yKiVjkG_wYWYdSatGIfsN-w,621
-datacustomcode/io/writer/base.py,sha256=6e2Yszb6BiuN1zCV2rjvZ5CQ0YikQdUgdYAkutoy9pE,1787
+datacustomcode/io/writer/base.py,sha256=RHafJ-Xmh_TnYcN4QfSh0VdIZeIevKmz9sf05adMgxA,1540
 datacustomcode/io/writer/csv.py,sha256=asty5teBpNQ1fMGHZ7wA3suLhq0sk0lVQPN5x7TOSRk,1555
 datacustomcode/io/writer/print.py,sha256=0g2sP_1Wb95UIyEELWJt3dqM18Iv_CWhDW3mDxcIhns,5105
-datacustomcode/llm_gateway/__init__.py,sha256=xT7J9qZByD80WWk-835JcJSvQx8_NiX6qsBPhq7mU6k,800
+datacustomcode/llm_gateway/__init__.py,sha256=ZYEMvQlnWzdSdBwYrhBQGmlfBBqsfmhAy4KViyCBFVA,995
 datacustomcode/llm_gateway/base.py,sha256=CTUhZiZ0JFlmWj6kQpTR4_-mUvqkBKl_Fkdvv9VwxY8,1262
-datacustomcode/llm_gateway/default.py,sha256=QiwrDaUvrh00I-fYYhmogTtqC5r0fmm-JrXxNQLvTc0,1854
+datacustomcode/llm_gateway/default.py,sha256=HVUc5NihNAQMVgRAJ8w2CBDX4mRLl8FRkpCJWv9FAOU,1952
+datacustomcode/llm_gateway/spark_base.py,sha256=h9fBrhBSSthCMBi6DjryxMKp_WSLdoMDg6a3FgO4oKI,1637
+datacustomcode/llm_gateway/spark_default.py,sha256=2-u9Oy936kj35wv8_qBSc-xtqnC7WNkgprhcuroP7U8,3524
 datacustomcode/llm_gateway/types/__init__.py,sha256=gamfOD1VnAtEslRBpqh-yKiVjkG_wYWYdSatGIfsN-w,621
-datacustomcode/llm_gateway/types/generate_text_request.py,sha256=ChxOCzp7St1EYgwl4tWiwfrPyVci4zPyBhnEjWc1Vfs,1448
-datacustomcode/llm_gateway/types/generate_text_request_builder.py,sha256=R4hCgatMPSfWYqaOmeC1IMQj753iDgu7wU992c4tkmk,2565
+datacustomcode/llm_gateway/types/generate_text_request.py,sha256=pHFp9r12xT_GlFm873kQEfrHCdoJI5sJGrCRVqqN3vg,1647
+datacustomcode/llm_gateway/types/generate_text_request_builder.py,sha256=P2lBwyjGQvqf8G9PR4pNeN02AeYri1XCOvlBeRqAJNw,2791
 datacustomcode/llm_gateway/types/generate_text_response.py,sha256=NSvj6nvzR4cZ0cVs2-PE9Ct3ZY-xL4MQYknJMd_xdDI,1912
 datacustomcode/llm_gateway/types/generate_text_response_builder.py,sha256=6MV6JZuTyHUId3Slw9k7m0KBgjj-5S9HI4qtd94Jl2A,1272
-datacustomcode/llm_gateway_config.py,sha256=oBfBlLipM3y8uBtFrynzy7HYXt_R9zMdb8gECXNpR9E,1919
+datacustomcode/llm_gateway_config.py,sha256=ECm9CFqOFOaLHwenvVFAjiRSzG9Eg9fY4pb8nHbQ34c,3312
 datacustomcode/mixin.py,sha256=ZtROqO1W1_D7MV8lw4cw6zzcchJLoZS0aAiXrolYR4k,4914
-datacustomcode/proxy/__init__.py,sha256=gamfOD1VnAtEslRBpqh-yKiVjkG_wYWYdSatGIfsN-w,621
-datacustomcode/proxy/base.py,sha256=Gw8of0cBgcOSpjnR7PX8pqWnsjHpt1AgKIK7u6k6DBU,846
-datacustomcode/proxy/client/__init__.py,sha256=gamfOD1VnAtEslRBpqh-yKiVjkG_wYWYdSatGIfsN-w,621
-datacustomcode/proxy/client/base.py,sha256=dWFcc5_5vSBxZmo7k5cwKnJl1_gcjvOZYI2EjBUXuaA,1073
 datacustomcode/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datacustomcode/run.py,sha256=78IpkuEih3GCO6sQPZLFfiKnrCI-eMtEMLsj-1vUvSE,7362
 datacustomcode/scan.py,sha256=Zb9mE733H_xaqyDI-LZjjnIcctMC9j1AaD_kcr9qTh4,14325
@@ -59,7 +57,7 @@ datacustomcode/spark/base.py,sha256=tlGqM4LxuLoDa7OxJF5nVeb7phO5uvD1DHBQeH7NMMs,
 datacustomcode/spark/default.py,sha256=aMB8CaTPYwHbQE_7XqBLQUZPGRdDJcRL72qTVh0aG7M,1433
 datacustomcode/template.py,sha256=FNNi8YPfd-JB-hUt1DDWSFK5AqwKnTnRSy_KpqYluaU,3282
 datacustomcode/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-datacustomcode/templates/function/.devcontainer/devcontainer.json,sha256=21RNTadhC-rynON2-VOtr5U174Hxnr_MA4RQ4592rsg,166
+datacustomcode/templates/function/.devcontainer/devcontainer.json,sha256=lWMFk-SdTKBlQt_VJrMn8gZPmvT4tfcjRpCNhLdElZc,167
 datacustomcode/templates/function/Dockerfile.dependencies,sha256=AfHRddm5l3ujv4vdrf0d-SMB-qxPCOa0XQm6ptP2Euw,174
 datacustomcode/templates/function/README.md,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datacustomcode/templates/function/build_native_dependencies.sh,sha256=dk9vhzlObdsFAO5BtxP7ioOGDBM8JVwJHp_W0712Nlk,222
@@ -67,17 +65,17 @@ datacustomcode/templates/function/chunking/payload/config.json,sha256=RBNvo1WzZ4
 datacustomcode/templates/function/chunking/payload/entrypoint.py,sha256=tzpQJjODulwzLilOoKdOo8GjpdIlCDByjnawpHIQYQA,4517
 datacustomcode/templates/function/chunking/requirements.txt,sha256=Ih0KsVdiTdo7KuIsqFB_AJMRT2r0ao_xDEb3hiDmf48,46
 datacustomcode/templates/function/example/chunking_with_llm/config.json,sha256=EvjuOVssRg2SjcUb5F50RRMGDqag_f69bv8AOflaK1g,38
-datacustomcode/templates/function/example/chunking_with_llm/entrypoint.py,sha256=BgxWWQeWUVtLDDH21HXHX9HZ5atO6A2_fokw8LsSeyI,3295
+datacustomcode/templates/function/example/chunking_with_llm/entrypoint.py,sha256=6AMVf_busCCDALktG8vkCtjnzJheMgM-OOLML5iLKOg,3437
 datacustomcode/templates/function/example/chunking_with_llm/files/chunking_prompt.txt,sha256=w_gSYNXwZmbMTMjZ_YYwm1IpMsVpI3hyC3ew60zqs_0,446
 datacustomcode/templates/function/example/chunking_with_llm/tests/test.json,sha256=ZcH6QGEWu03dc922Za3vdx6hJY08nfGPie9kkeup-sc,9457
 datacustomcode/templates/function/example/chunking_with_prediction/config.json,sha256=EvjuOVssRg2SjcUb5F50RRMGDqag_f69bv8AOflaK1g,38
-datacustomcode/templates/function/example/chunking_with_prediction/entrypoint.py,sha256=4Rtv9HAPzDNmxbDt0OBoS4Y5SmCDGnQDpy1i800fXPY,7578
+datacustomcode/templates/function/example/chunking_with_prediction/entrypoint.py,sha256=w25_WucA-tUN3O6j-Od_WsW4HDH1Dfp2uJYNfVs_0M4,7722
 datacustomcode/templates/function/example/chunking_with_prediction/tests/test.json,sha256=2EE0c0n-s6eOayb_JrBqeKkXzyxieOyHK3_Mbm--Nmo,1073
 datacustomcode/templates/function/payload/config.json,sha256=RBNvo1WzZ4oRRq0W9-hknpT7T8If536DEMBg9hyq_4o,2
-datacustomcode/templates/function/payload/entrypoint.py,sha256=DrH73aAkU_ua6IHBRYr9SkMtX5pEFvNJYew8zXQL5TQ,5705
+datacustomcode/templates/function/payload/entrypoint.py,sha256=uWb9mT7GaOvVWBK8P7vHS1GNbNaH--rSLV8J1v0eTHY,5667
 datacustomcode/templates/function/requirements-dev.txt,sha256=qT2-m2FOgiYPTvqaNy0yRzIAXqHgR5b8stccYXsXVi8,88
 datacustomcode/templates/function/requirements.txt,sha256=qJbqzs5z0Hrx590U3T7dHq_m8UQEx2TdhgrJSz-QHIQ,40
-datacustomcode/templates/script/.devcontainer/devcontainer.json,sha256=21RNTadhC-rynON2-VOtr5U174Hxnr_MA4RQ4592rsg,166
+datacustomcode/templates/script/.devcontainer/devcontainer.json,sha256=lWMFk-SdTKBlQt_VJrMn8gZPmvT4tfcjRpCNhLdElZc,167
 datacustomcode/templates/script/Dockerfile,sha256=l8HEy5X7BsW9Gi7rBYM6G2czwk-1832RRIos7xyFzpI,463
 datacustomcode/templates/script/Dockerfile.dependencies,sha256=vBbTfgcBkVQVCnrJZE_QEx89ddTQrLZGeyVTZXU8TI8,206
 datacustomcode/templates/script/README.md,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -87,13 +85,13 @@ datacustomcode/templates/script/examples/employee_hierarchy/employee_data.csv,sh
 datacustomcode/templates/script/examples/employee_hierarchy/entrypoint.py,sha256=Mfm3iQtEHTQRW6cmZTwRKdTh3IeGWR-teXrd6x-YLSY,2223
 datacustomcode/templates/script/jupyterlab.sh,sha256=IHR3YQ8d_busuyvesByhJgzCKULIFPI-Hqogs0NPhCs,2432
 datacustomcode/templates/script/payload/config.json,sha256=0d2mEMt4NHIeOUTsgiPMuRKdOLIQxmh-Wq-IfEqU-gc,28
-datacustomcode/templates/script/payload/entrypoint.py,sha256=qpFA-jkhaYYF2CFNQ7_7MZVefU7SxnoMQIQYi7ox2dY,691
+datacustomcode/templates/script/payload/entrypoint.py,sha256=HFrTe7--w5X78Sh4DVkoGGR6-zeqxemqp80RM20Rr4k,1642
 datacustomcode/templates/script/requirements-dev.txt,sha256=OWwuy1awesqOZOS3Zizmdd6BDnSePpGFN5bq5IrALvc,176
 datacustomcode/templates/script/requirements.txt,sha256=qJbqzs5z0Hrx590U3T7dHq_m8UQEx2TdhgrJSz-QHIQ,40
-datacustomcode/token_provider.py,sha256=a16r5xYeCOPR3M3JKvEV7JVIG7DOzDswObRBI33MQOk,6461
+datacustomcode/token_provider.py,sha256=KPj_ca6mcBmhXHaPmZjAI4it8QvOWdwKJL8yfqxfYwU,5446
 datacustomcode/version.py,sha256=9LlbVrzwBvut1L308QbwNBgxdQk5CrghH39JARVSxms,989
-salesforce_data_customcode-6.0.1.dist-info/METADATA,sha256=PVoEUlTf9pjQlJq8Yi3Mt7migPd0K16ORl80T2gsMbc,20162
-salesforce_data_customcode-6.0.1.dist-info/WHEEL,sha256=eY7nduwzv-ldUxpzbRlxwvC693Hg6PX8bWDjEHjZ_dk,88
-salesforce_data_customcode-6.0.1.dist-info/entry_points.txt,sha256=WpQ94UB7UuRCYGOLtJV3vAgm1tl7iVf43VYRWIuNu5Y,74
-salesforce_data_customcode-6.0.1.dist-info/licenses/LICENSE.txt,sha256=iOi8EmQpfkFhMENi7VYtkl2EqS14pEOccoXHiW2dyPU,11443
-salesforce_data_customcode-6.0.1.dist-info/RECORD,,
+salesforce_data_customcode-6.0.2.dev1.dist-info/METADATA,sha256=8lK5D1Zf3TelM0vlib8QBQLtdmixh79g4ce2Owuh8QQ,21748
+salesforce_data_customcode-6.0.2.dev1.dist-info/WHEEL,sha256=eY7nduwzv-ldUxpzbRlxwvC693Hg6PX8bWDjEHjZ_dk,88
+salesforce_data_customcode-6.0.2.dev1.dist-info/entry_points.txt,sha256=WpQ94UB7UuRCYGOLtJV3vAgm1tl7iVf43VYRWIuNu5Y,74
+salesforce_data_customcode-6.0.2.dev1.dist-info/licenses/LICENSE.txt,sha256=iOi8EmQpfkFhMENi7VYtkl2EqS14pEOccoXHiW2dyPU,11443
+salesforce_data_customcode-6.0.2.dev1.dist-info/RECORD,,

datacustomcode/proxy/__init__.py

@@ -1,14 +0,0 @@
-# Copyright (c) 2025, Salesforce, Inc.
-# SPDX-License-Identifier: Apache-2
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.

datacustomcode/proxy/base.py

@@ -1,24 +0,0 @@
-# Copyright (c) 2025, Salesforce, Inc.
-# SPDX-License-Identifier: Apache-2
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-from __future__ import annotations
-
-from abc import ABC
-
-from datacustomcode.mixin import UserExtendableNamedConfigMixin
-
-
-class BaseProxyAccessLayer(ABC, UserExtendableNamedConfigMixin):
-    def __init__(self):
-        pass

datacustomcode/proxy/client/__init__.py

@@ -1,14 +0,0 @@
-# Copyright (c) 2025, Salesforce, Inc.
-# SPDX-License-Identifier: Apache-2
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.

datacustomcode/proxy/client/base.py

@@ -1,32 +0,0 @@
-# Copyright (c) 2025, Salesforce, Inc.
-# SPDX-License-Identifier: Apache-2
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-from __future__ import annotations
-
-from abc import abstractmethod
-
-from datacustomcode.proxy.base import BaseProxyAccessLayer
-
-
-class BaseProxyClient(BaseProxyAccessLayer):
-    def __init__(self):
-        pass
-
-    @abstractmethod
-    def call_llm_gateway(self, llmModelId: str, prompt: str, maxTokens: int) -> str: ...
-
-    @abstractmethod
-    def llm_gateway_generate_text(
-        self, template, values, llmModelId: str, maxTokens: int
-    ): ...