@sap-ai-sdk/orchestration 1.17.1-20250825032219.0 → 1.17.1-20250920013049.0

package/dist/orchestration-types.d.ts

@@ -1,11 +1,11 @@
 import type { Xor } from '@sap-cloud-sdk/util';
 import type { CustomRequestConfig } from '@sap-cloud-sdk/http-client';
 import type { ChatModel } from './model-types.js';
-import type { ChatMessages, DataRepositoryType, DocumentGroundingFilter, DpiConfig, DpiEntities, FilteringModuleConfig, FilteringStreamOptions, GlobalStreamOptions, GroundingModuleConfig, InputTranslationModuleConfig, LlamaGuard38B, MaskingModuleConfig, LlmModuleConfig as OriginalLlmModuleConfig, Template as OriginalTemplate, TemplateRef, TemplatingChatMessage, OutputTranslationModuleConfig, AzureContentSafetyOutput, AzureContentSafetyOutputFilterConfig, DPIStandardEntity, DPICustomEntity } from './client/api/schema/index.js';
+import type { ChatMessages, DataRepositoryType, DocumentGroundingFilter, DpiConfig, DpiEntities, FilteringStreamOptions, LlamaGuard38B, LLMModelDetails as OriginalLLMModelDetails, Template, TemplateRef, TemplatingChatMessage, DPIStandardEntity, DPICustomEntity, InputFilteringConfig, OutputFilteringConfig, MaskingProviderConfig, SAPDocumentTranslation, GlobalStreamOptions, ErrorResponse, AzureContentSafetyInputFilterConfig, AzureContentSafetyOutputFilterConfig, LlamaGuard38BFilterConfig } from './client/api/schema/index.js';
 /**
- * Prompt configuration.
+ * Chat completion request configuration.
  */
-export interface Prompt {
+export interface ChatCompletionRequest {
     /**
      * Chat History.
      */
@@ -28,20 +28,89 @@ export interface Prompt {
     /**
      * Template parameters.
      * @example
-     * inputParams: {
+     * placeholderValues: {
      *   product: 'SAP Cloud SDK'
      * }
      */
-    inputParams?: Record<string, string>;
+    placeholderValues?: Record<string, string>;
 }
 /**
- * LLM module configuration.
+ * LLM model details.
  */
-export type LlmModuleConfig = OriginalLlmModuleConfig & {
-    /** */
-    model_name: ChatModel;
-    model_params?: LlmModelParams;
+export type LlmModelDetails = Omit<OriginalLLMModelDetails, 'name' | 'params'> & {
+    name: ChatModel;
+    params?: LlmModelParams;
 };
+/**
+ * Prompt templating module configuration.
+ */
+export interface PromptTemplatingModule {
+    /**
+     * The prompt template to be used. Can be either a user defined template or a reference to a template in the prompt registry.
+     */
+    /**
+     * Static prompts. Can be:
+     * - A `template`: an array of templated chat messages (with {{?placeholders}}).
+     * - An `id` or `scenario`, `name` and `version`: reference to a remote prompt template.
+     *
+     * This is meant for static instructions included with every call.
+     * For per-request templating, use `messages` in `.chatCompletion()` instead.
+     * @example
+     * prompt: {
+     *   template: [
+     *     {
+     *       role: 'system',
+     *       content: 'You are an assistant for {{?product}}.'
+     *     }
+     *   ]
+     * }
+     */
+    prompt?: Xor<PromptTemplate, TemplateRef> | string;
+    /**
+     * LLM model details.
+     */
+    model: LlmModelDetails;
+}
+/**
+ * Representation of the `FilteringModuleConfig` schema.
+ */
+export interface FilteringModule {
+    /**
+     * List of provider type and filters.
+     */
+    input?: InputFilteringConfig;
+    /**
+     * List of provider type and filters.
+     */
+    output?: OutputFilteringConfig;
+}
+/**
+ * Representation of the `MaskingModuleConfig` schema.
+ */
+export interface MaskingModule {
+    /**
+     * List of masking service providers
+     * Min Items: 1.
+     */
+    masking_providers: MaskingProviderConfig[];
+}
+/**
+ * Configuration for translation module.
+ */
+export interface TranslationModule {
+    /**
+     * Configuration for input translation.
+     */
+    input?: TranslationInputConfig;
+    /**
+     * Configuration for output translation.
+     */
+    output?: TranslationOutputConfig;
+}
+/**
+ * Orchestration error response.
+ */
+export type OrchestrationErrorResponse = ErrorResponse;
 /**
  * Model Parameters for LLM module configuration.
  */
@@ -56,9 +125,9 @@ export type LlmModelParams = {
 /**
  * Representation of the 'Template' schema.
  */
-export type Template = Omit<OriginalTemplate, 'template'> & {
+export type PromptTemplate = Omit<Template, 'template'> & {
     /**
-     * A chat message array to be formatted with values from `inputParams`.
+     * A chat message array to be formatted with values from `placeholderValues`.
      * Both `role` and `content` can use {{?variable}} placeholders.
      *
      * For dynamic templating (changing per request), pass templated messages directly in `.chatCompletion({ messages })`.
@@ -75,70 +144,39 @@ export type Template = Omit<OriginalTemplate, 'template'> & {
      */
     template?: TemplatingChatMessage;
 };
-/**
- * Representation of the 'TemplatingModuleConfig' schema.
- * The type can be either a `Template` or a `TemplateRef`.
- */
-export type TemplatingModuleConfig = Xor<Template, TemplateRef>;
 /**
  * Orchestration module configuration.
  */
 export interface OrchestrationModuleConfig {
     /**
-     * Templating configuration for static prompts. Can be:
-     * - A `template`: an array of templated chat messages (with {{?placeholders}}).
-     * - An `id` or `scenario`, `name` and `version`: reference to a remote prompt template.
-     *
-     * This is meant for static instructions included with every call.
-     * For per-request templating, use `messages` in `.chatCompletion()` instead.
-     * @example
-     * templating: {
-     *   template: [
-     *     {
-     *       role: 'system',
-     *       content: 'You are an assistant for {{?product}}.'
-     *     }
-     *   ]
-     * }
-     */
-    templating?: TemplatingModuleConfig | string;
-    /**
-     * LLM module configuration.
+     * Prompt templating configuration.
      */
-    llm: LlmModuleConfig;
+    promptTemplating: PromptTemplatingModule;
     /**
      * Filtering module configuration for both input and output filters.
-     * To configure a filter, use convenience functions like `buildAzureContentSafetyFilter`, `buildLlamaGuardFilter`, etc..
+     * To configure a filter, use convenience functions like `buildAzureContentSafetyFilter`, `buildLlamaGuard38BFilter`, etc..
      * @example
      * filtering: {
      *   input: {
      *     filters: [
-     *       buildAzureContentSafetyFilter({ Hate: 'ALLOW_SAFE', Violence: 'ALLOW_SAFE_LOW_MEDIUM' })
+     *       buildAzureContentSafetyFilter('input', { hate: 'ALLOW_SAFE', violence: 'ALLOW_SAFE_LOW_MEDIUM' })
      *     ]
      *   }
      * }
      */
-    filtering?: FilteringModuleConfig;
+    filtering?: FilteringModule;
     /**
      * Masking module configuration.
      */
-    masking?: MaskingModuleConfig;
+    masking?: MaskingModule;
     /**
      * Grounding module configuraton.
      */
-    grounding?: GroundingModuleConfig;
-    /**
-     * Global streaming options.
-     */
-    streaming?: GlobalStreamOptions;
-    /**
-     * Input translation module configuration.
-     */
-    inputTranslation?: InputTranslationModuleConfig;
+    grounding?: GroundingModule;
     /**
-     * Output translation module configuration.
+     * Translation module configuration.
      */
-    outputTranslation?: OutputTranslationModuleConfig;
+    translation?: TranslationModule;
 }
 /**
  * Request options for orchestration.
@@ -147,7 +185,7 @@ export interface RequestOptions {
     /**
      * Prompt configuration.
      */
-    prompt?: Prompt;
+    request?: ChatCompletionRequest;
     /**
      * Custom request configuration.
      */
@@ -168,7 +206,7 @@ export interface StreamOptions {
     /**
      * LLM specific stream options.
      */
-    llm?: {
+    promptTemplating?: {
         include_usage?: boolean;
         [key: string]: any;
     } | null;
@@ -181,6 +219,43 @@ export interface StreamOptions {
      */
     global?: GlobalStreamOptions;
 }
+/**
+ * Representation of the `GroundingModuleConfig` schema.
+ */
+export interface GroundingModule {
+    /**
+     * @example 'document_grounding_service'
+     */
+    type: 'document_grounding_service' | any;
+    /**
+     * Grounding service configuration.
+     */
+    config: {
+        /**
+         * Document grounding service filters to be used.
+         */
+        filters?: DocumentGroundingFilter[];
+        /**
+         * Placeholders to be used for grounding input questions and output.
+         */
+        placeholders: {
+            /**
+             * Contains the input parameters used for grounding input questions
+             * Min Items: 1.
+             */
+            input: string[];
+            /**
+             * Placeholder name for grounding output.
+             * @example 'groundingOutput'
+             */
+            output: string;
+        };
+        /**
+         * Parameter name used for specifying metadata parameters.
+         */
+        metadata_params?: string[];
+    };
+}
 /**
  * Represents a filter configuration for the Document Grounding Service.
  */
@@ -196,18 +271,24 @@ export type DocumentGroundingServiceFilter = Omit<DocumentGroundingFilter, 'data
  */
 export interface DocumentGroundingServiceConfig {
     /**
-     * Defines the filters to apply during the grounding process.
+     * Document grounding service filters to be used.
      */
     filters?: DocumentGroundingServiceFilter[];
     /**
-     * Contains the input parameters used for grounding input questions.
+     * Placeholders to be used for grounding input questions and output.
      */
-    input_params: string[];
-    /**
-     * Parameter name used for grounding output.
-     * @example "groundingOutput"
-     */
-    output_param: string;
+    placeholders: {
+        /**
+         * Contains the input parameters used for grounding input questions
+         * Min Items: 1.
+         */
+        input: string[];
+        /**
+         * Placeholder name for grounding output.
+         * @example "groundingOutput"
+         */
+        output: string;
+    };
     /**
      * Parameter name used for specifying metadata parameters.
      */
@@ -215,9 +296,8 @@ export interface DocumentGroundingServiceConfig {
 }
 /**
  * Defines the type of the DPI masking entity.
- * @internal
  */
-export type DpiEntity = DpiEntities | DPIStandardEntity | (DPICustomEntity & {
+type DpiEntity = DpiEntities | DPIStandardEntity | (DPICustomEntity & {
     type: 'custom';
 });
 /**
@@ -228,26 +308,35 @@ export type DpiMaskingConfig = Omit<DpiConfig, 'type' | 'entities' | 'mask_groun
     mask_grounding_input?: boolean;
 };
 /**
- * Filter configuration for Azure content safety Filter.
+ * Output Parameters for Azure content safety output filter.
  */
-export interface AzureContentFilter {
+export interface AzureContentSafetyFilterOutputParameters {
     /**
      * The filter category for hate content.
      */
-    Hate?: AzureFilterThreshold;
+    hate?: AzureFilterThreshold;
     /**
      * The filter category for self-harm content.
      */
-    SelfHarm?: AzureFilterThreshold;
+    self_harm?: AzureFilterThreshold;
     /**
      * The filter category for sexual content.
      */
-    Sexual?: AzureFilterThreshold;
+    sexual?: AzureFilterThreshold;
     /**
      * The filter category for violence content.
      */
-    Violence?: AzureFilterThreshold;
+    violence?: AzureFilterThreshold;
 }
+/**
+ * Input parameters for Azure content safety input filter.
+ */
+export type AzureContentSafetyFilterInputParameters = AzureContentSafetyFilterOutputParameters & {
+    /**
+     * A flag to use prompt shield.
+     */
+    prompt_shield?: boolean;
+};
 /**
  * A descriptive constant for Azure content safety filter threshold.
  * @internal
@@ -260,35 +349,80 @@ export declare const supportedAzureFilterThresholds: {
 };
 /**
  * The Azure threshold level supported for each azure content filter category.
- *
  */
 export type AzureFilterThreshold = keyof typeof supportedAzureFilterThresholds;
 /**
- * The filter categories supported for Llama guard filter.
+ * The filter categories supported for Llama Guard 3 8B filter.
  */
-export type LlamaGuardCategory = keyof LlamaGuard38B;
+export type LlamaGuard38BCategory = keyof LlamaGuard38B;
 /**
- * Translation configuration for SAP Document Translation.
- * See https://help.sap.com/docs/translation-hub/sap-translation-hub/supported-languages-6854bbb1bd824ffebc3a097a7c0fd45d for list of supported languages.
+ * Type for module configurations (input or output).
+ * @internal
  */
-export interface TranslationConfigParams {
+export type ConfigType = 'input' | 'output';
+/**
+ * Parameters for translation configuration.
+ * @internal
+ */
+interface TranslationConfigParameters {
     /**
      * Language of the text to be translated.
-     * @example sourceLanguage: "de-DE"
+     * @example sourceLanguage: 'de-DE'
      */
     sourceLanguage?: string;
     /**
      * Language to which the text should be translated.
-     * @example targetLanguage: "en-US"
+     * @example targetLanguage: 'en-US'
      */
     targetLanguage: string;
 }
 /**
- * Filter configuration for Azure Content Safety.
+ * Output parameters for translation output configuration.
+ */
+export type TranslationOutputParameters = TranslationConfigParameters;
+/**
+ * Input parameters for translation input configuration.
+ */
+export type TranslationInputParameters = TranslationConfigParameters;
+/**
+ * Parameters for translation configurations.
+ */
+export type TranslationConfigParams<T extends ConfigType> = T extends 'input' ? TranslationInputParameters : TranslationOutputParameters;
+/**
+ * Input configuration for translation module.
+ */
+export type TranslationInputConfig = SAPDocumentTranslation;
+/**
+ * Output configuration for translation module.
+ */
+export type TranslationOutputConfig = SAPDocumentTranslation;
+/**
+ * Return type for translation configurations.
+ */
+export type TranslationReturnType<T extends ConfigType> = T extends 'input' ? TranslationInputConfig : TranslationOutputConfig;
+/**
+ * Parameters for Azure content safety filters.
+ */
+export type AzureContentSafetyFilterParameters<T extends ConfigType> = T extends 'input' ? AzureContentSafetyFilterInputParameters : AzureContentSafetyFilterOutputParameters;
+/**
+ * Filter return type for Azure Content Safety.
+ */
+export type AzureContentSafetyFilterReturnType<T extends ConfigType> = T extends 'input' ? AzureContentSafetyInputFilterConfig : AzureContentSafetyOutputFilterConfig;
+/**
+ * Input filter configuration for Llama Guard 3 8B.
+ */
+export type LlamaGuard38BInputFilterConfig = LlamaGuard38BFilterConfig;
+/**
+ * Output filter configuration for Llama Guard 3 8B.
+ */
+export type LlamaGuard38BOutputFilterConfig = LlamaGuard38BFilterConfig;
+/**
+ * Filter return type for Llama Guard 3 8B.
  */
-export type AzureContentSafety = AzureContentSafetyOutput;
+export type LlamaGuard38BFilterReturnType<T extends ConfigType> = T extends 'input' ? LlamaGuard38BInputFilterConfig : LlamaGuard38BOutputFilterConfig;
 /**
- * Representation of the Azure Content Safety filter config schema.
+ * Representation of the 'DpiConfig' schema.
  */
-export type AzureContentSafetyFilterConfig = AzureContentSafetyOutputFilterConfig;
+export type DpiMaskingProviderConfig = DpiConfig;
+export {};
 //# sourceMappingURL=orchestration-types.d.ts.map

package/dist/orchestration-types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"orchestration-types.d.ts","sourceRoot":"","sources":["../src/orchestration-types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,qBAAqB,CAAC;AAC/C,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AACtE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,KAAK,EACV,YAAY,EACZ,kBAAkB,EAClB,uBAAuB,EACvB,SAAS,EACT,WAAW,EACX,qBAAqB,EACrB,sBAAsB,EACtB,mBAAmB,EACnB,qBAAqB,EACrB,4BAA4B,EAC5B,aAAa,EACb,mBAAmB,EACnB,eAAe,IAAI,uBAAuB,EAC1C,QAAQ,IAAI,gBAAgB,EAC5B,WAAW,EACX,qBAAqB,EACrB,6BAA6B,EAC7B,wBAAwB,EACxB,oCAAoC,EACpC,iBAAiB,EACjB,eAAe,EAChB,MAAM,8BAA8B,CAAC;AAEtC;;GAEG;AACH,MAAM,WAAW,MAAM;IACrB;;OAEG;IACH,eAAe,CAAC,EAAE,YAAY,CAAC;IAE/B;;;;;;;;;;;;;OAaG;IACH,QAAQ,CAAC,EAAE,YAAY,CAAC;IAExB;;;;;;OAMG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACtC;AAED;;GAEG;AACH,MAAM,MAAM,eAAe,GAAG,uBAAuB,GAAG;IACtD,MAAM;IACN,UAAU,EAAE,SAAS,CAAC;IACtB,YAAY,CAAC,EAAE,cAAc,CAAC;CAC/B,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,CAAC,CAAC,EAAE,MAAM,CAAC;CACZ,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;AAExB;;GAEG;AACH,MAAM,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,EAAE,UAAU,CAAC,GAAG;IAC1D;;;;;;;;;;;;;;;OAeG;IACH,QAAQ,CAAC,EAAE,qBAAqB,CAAC;CAClC,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,sBAAsB,GAAG,GAAG,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;AAEhE;;GAEG;AACH,MAAM,WAAW,yBAAyB;IACxC;;;;;;;;;;;;;;;;OAgBG;IACH,UAAU,CAAC,EAAE,sBAAsB,GAAG,MAAM,CAAC;IAC7C;;OAEG;IACH,GAAG,EAAE,eAAe,CAAC;IACrB;;;;;;;;;;;OAWG;IACH,SAAS,CAAC,EAAE,qBAAqB,CAAC;IAClC;;OAEG;IACH,OAAO,CAAC,EAAE,mBAAmB,CAAC;IAC9B;;OAEG;IACH,SAAS,CAAC,EAAE,qBAAqB,CAAC;IAClC;;OAEG;IACH,SAAS,CAAC,EAAE,mBAAmB,CAAC;IAChC;;OAEG;IACH,gBAAgB,CAAC,EAAE,4BAA4B,CAAC;IAChD;;OAEG;IACH,iBAAiB,CAAC,EAAE,6BAA6B,CAAC;CACnD;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B;;OAEG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,aAAa,CAAC,EAAE,mBAAmB,CAAC;IACpC;;OAEG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB;;OAEG;IACH,aAAa,CAAC,EAAE,aAAa,CAAC;CAC/B;AAED;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B;;OAEG;IACH,GAAG,CAAC,EAAE;QAAE,aAAa,CAAC,EAAE,OAAO,CAAC;QAAC,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,GAAG,IAAI,CAAC;IAC7D;;OAEG;IACH,eAAe,CAAC,EAAE,sBAAsB,CAAC;IACzC;;OAEG;IACH,MAAM,CAAC,EAAE,mBAAmB,CAAC;CAC9B;AAED;;GAEG;AACH,MAAM,MAAM,8BAA8B,GAAG,IAAI,CAC/C,uBAAuB,EACvB,sBAAsB,CACvB,GAAG;IACF;;;OAGG;IACH,oBAAoB,CAAC,EAAE,kBAAkB,CAAC;CAC3C,CAAC;AAEF;;GAEG;AACH,MAAM,WAAW,8BAA8B;IAC7C;;OAEG;IACH,OAAO,CAAC,EAAE,8BAA8B,EAAE,CAAC;IAC3C;;OAEG;IACH,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB;;;OAGG;IACH,YAAY,EAAE,MAAM,CAAC;IACrB;;OAEG;IACH,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;CAC5B;AAED;;;GAGG;AACH,MAAM,MAAM,SAAS,GACjB,WAAW,GACX,iBAAiB,GACjB,CAAC,eAAe,GAAG;IACjB,IAAI,EAAE,QAAQ,CAAC;CAChB,CAAC,CAAC;AACP;;GAEG;AACH,MAAM,MAAM,gBAAgB,GAAG,IAAI,CACjC,SAAS,EACT,MAAM,GAAG,UAAU,GAAG,sBAAsB,CAC7C,GAAG;IACF,QAAQ,EAAE,CAAC,SAAS,EAAE,GAAG,SAAS,EAAE,CAAC,CAAC;IACtC,oBAAoB,CAAC,EAAE,OAAO,CAAC;CAChC,CAAC;AAEF;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC;;OAEG;IACH,IAAI,CAAC,EAAE,oBAAoB,CAAC;IAC5B;;OAEG;IACH,QAAQ,CAAC,EAAE,oBAAoB,CAAC;IAChC;;OAEG;IACH,MAAM,CAAC,EAAE,oBAAoB,CAAC;IAC9B;;OAEG;IACH,QAAQ,CAAC,EAAE,oBAAoB,CAAC;CACjC;AAED;;;GAGG;AACH,eAAO,MAAM,8BAA8B;;;;;CAKjC,CAAC;AAEX;;;GAGG;AACH,MAAM,MAAM,oBAAoB,GAAG,MAAM,OAAO,8BAA8B,CAAC;AAE/E;;GAEG;AACH,MAAM,MAAM,kBAAkB,GAAG,MAAM,aAAa,CAAC;AAErD;;;GAGG;AACH,MAAM,WAAW,uBAAuB;IACtC;;;OAGG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;;OAGG;IACH,cAAc,EAAE,MAAM,CAAC;CACxB;AAED;;GAEG;AACH,MAAM,MAAM,kBAAkB,GAAG,wBAAwB,CAAC;AAE1D;;GAEG;AACH,MAAM,MAAM,8BAA8B,GACxC,oCAAoC,CAAC"}
+{"version":3,"file":"orchestration-types.d.ts","sourceRoot":"","sources":["../src/orchestration-types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,qBAAqB,CAAC;AAC/C,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AACtE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,KAAK,EACV,YAAY,EACZ,kBAAkB,EAClB,uBAAuB,EACvB,SAAS,EACT,WAAW,EACX,sBAAsB,EACtB,aAAa,EACb,eAAe,IAAI,uBAAuB,EAC1C,QAAQ,EACR,WAAW,EACX,qBAAqB,EACrB,iBAAiB,EACjB,eAAe,EACf,oBAAoB,EACpB,qBAAqB,EACrB,qBAAqB,EACrB,sBAAsB,EACtB,mBAAmB,EACnB,aAAa,EACb,mCAAmC,EACnC,oCAAoC,EACpC,yBAAyB,EAC1B,MAAM,8BAA8B,CAAC;AAEtC;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC;;OAEG;IACH,eAAe,CAAC,EAAE,YAAY,CAAC;IAE/B;;;;;;;;;;;;;OAaG;IACH,QAAQ,CAAC,EAAE,YAAY,CAAC;IAExB;;;;;;OAMG;IACH,iBAAiB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAC5C;AAED;;GAEG;AACH,MAAM,MAAM,eAAe,GAAG,IAAI,CAChC,uBAAuB,EACvB,MAAM,GAAG,QAAQ,CAClB,GAAG;IACF,IAAI,EAAE,SAAS,CAAC;IAChB,MAAM,CAAC,EAAE,cAAc,CAAC;CACzB,CAAC;AAEF;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACrC;;OAEG;IACH;;;;;;;;;;;;;;;;OAgBG;IACH,MAAM,CAAC,EAAE,GAAG,CAAC,cAAc,EAAE,WAAW,CAAC,GAAG,MAAM,CAAC;IACnD;;OAEG;IACH,KAAK,EAAE,eAAe,CAAC;CACxB;AAED;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B;;OAEG;IACH,KAAK,CAAC,EAAE,oBAAoB,CAAC;IAC7B;;OAEG;IACH,MAAM,CAAC,EAAE,qBAAqB,CAAC;CAChC;AAED;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B;;;OAGG;IACH,iBAAiB,EAAE,qBAAqB,EAAE,CAAC;CAC5C;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC;;OAEG;IACH,KAAK,CAAC,EAAE,sBAAsB,CAAC;IAC/B;;OAEG;IACH,MAAM,CAAC,EAAE,uBAAuB,CAAC;CAClC;AAED;;GAEG;AACH,MAAM,MAAM,0BAA0B,GAAG,aAAa,CAAC;AAEvD;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,CAAC,CAAC,EAAE,MAAM,CAAC;CACZ,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;AAExB;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG,IAAI,CAAC,QAAQ,EAAE,UAAU,CAAC,GAAG;IACxD;;;;;;;;;;;;;;;OAeG;IACH,QAAQ,CAAC,EAAE,qBAAqB,CAAC;CAClC,CAAC;AAEF;;GAEG;AACH,MAAM,WAAW,yBAAyB;IACxC;;OAEG;IACH,gBAAgB,EAAE,sBAAsB,CAAC;IACzC;;;;;;;;;;;OAWG;IACH,SAAS,CAAC,EAAE,eAAe,CAAC;IAC5B;;OAEG;IACH,OAAO,CAAC,EAAE,aAAa,CAAC;IACxB;;OAEG;IACH,SAAS,CAAC,EAAE,eAAe,CAAC;IAC5B;;OAEG;IACH,WAAW,CAAC,EAAE,iBAAiB,CAAC;CACjC;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B;;OAEG;IACH,OAAO,CAAC,EAAE,qBAAqB,CAAC;IAChC;;OAEG;IACH,aAAa,CAAC,EAAE,mBAAmB,CAAC;IACpC;;OAEG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB;;OAEG;IACH,aAAa,CAAC,EAAE,aAAa,CAAC;CAC/B;AAED;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B;;OAEG;IACH,gBAAgB,CAAC,EAAE;QAAE,aAAa,CAAC,EAAE,OAAO,CAAC;QAAC,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,GAAG,IAAI,CAAC;IAC1E;;OAEG;IACH,eAAe,CAAC,EAAE,sBAAsB,CAAC;IACzC;;OAEG;IACH,MAAM,CAAC,EAAE,mBAAmB,CAAC;CAC9B;AACD;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B;;OAEG;IACH,IAAI,EAAE,4BAA4B,GAAG,GAAG,CAAC;IACzC;;OAEG;IACH,MAAM,EAAE;QACN;;WAEG;QACH,OAAO,CAAC,EAAE,uBAAuB,EAAE,CAAC;QACpC;;WAEG;QACH,YAAY,EAAE;YACZ;;;eAGG;YACH,KAAK,EAAE,MAAM,EAAE,CAAC;YAChB;;;eAGG;YACH,MAAM,EAAE,MAAM,CAAC;SAChB,CAAC;QACF;;WAEG;QACH,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;KAC5B,CAAC;CACH;AAED;;GAEG;AACH,MAAM,MAAM,8BAA8B,GAAG,IAAI,CAC/C,uBAAuB,EACvB,sBAAsB,CACvB,GAAG;IACF;;;OAGG;IACH,oBAAoB,CAAC,EAAE,kBAAkB,CAAC;CAC3C,CAAC;AAEF;;GAEG;AACH,MAAM,WAAW,8BAA8B;IAC7C;;OAEG;IACH,OAAO,CAAC,EAAE,8BAA8B,EAAE,CAAC;IAC3C;;OAEG;IACH,YAAY,EAAE;QACZ;;;WAGG;QACH,KAAK,EAAE,MAAM,EAAE,CAAC;QAChB;;;WAGG;QACH,MAAM,EAAE,MAAM,CAAC;KAChB,CAAC;IACF;;OAEG;IACH,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;CAC5B;AAED;;GAEG;AACH,KAAK,SAAS,GACV,WAAW,GACX,iBAAiB,GACjB,CAAC,eAAe,GAAG;IACjB,IAAI,EAAE,QAAQ,CAAC;CAChB,CAAC,CAAC;AAEP;;GAEG;AACH,MAAM,MAAM,gBAAgB,GAAG,IAAI,CACjC,SAAS,EACT,MAAM,GAAG,UAAU,GAAG,sBAAsB,CAC7C,GAAG;IACF,QAAQ,EAAE,CAAC,SAAS,EAAE,GAAG,SAAS,EAAE,CAAC,CAAC;IACtC,oBAAoB,CAAC,EAAE,OAAO,CAAC;CAChC,CAAC;AAEF;;GAEG;AACH,MAAM,WAAW,wCAAwC;IACvD;;OAEG;IACH,IAAI,CAAC,EAAE,oBAAoB,CAAC;IAC5B;;OAEG;IACH,SAAS,CAAC,EAAE,oBAAoB,CAAC;IACjC;;OAEG;IACH,MAAM,CAAC,EAAE,oBAAoB,CAAC;IAC9B;;OAEG;IACH,QAAQ,CAAC,EAAE,oBAAoB,CAAC;CACjC;AACD;;GAEG;AACH,MAAM,MAAM,uCAAuC,GACjD,wCAAwC,GAAG;IACzC;;OAEG;IACH,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB,CAAC;AAEJ;;;GAGG;AACH,eAAO,MAAM,8BAA8B;;;;;CAKjC,CAAC;AAEX;;GAEG;AACH,MAAM,MAAM,oBAAoB,GAAG,MAAM,OAAO,8BAA8B,CAAC;AAE/E;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAAG,MAAM,aAAa,CAAC;AAExD;;;GAGG;AACH,MAAM,MAAM,UAAU,GAAG,OAAO,GAAG,QAAQ,CAAC;AAE5C;;;GAGG;AACH,UAAU,2BAA2B;IACnC;;;OAGG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;;OAGG;IACH,cAAc,EAAE,MAAM,CAAC;CACxB;AAED;;GAEG;AACH,MAAM,MAAM,2BAA2B,GAAG,2BAA2B,CAAC;AAEtE;;GAEG;AACH,MAAM,MAAM,0BAA0B,GAAG,2BAA2B,CAAC;AAErE;;GAEG;AACH,MAAM,MAAM,uBAAuB,CAAC,CAAC,SAAS,UAAU,IAAI,CAAC,SAAS,OAAO,GACzE,0BAA0B,GAC1B,2BAA2B,CAAC;AAEhC;;GAEG;AACH,MAAM,MAAM,sBAAsB,GAAG,sBAAsB,CAAC;AAE5D;;GAEG;AACH,MAAM,MAAM,uBAAuB,GAAG,sBAAsB,CAAC;AAE7D;;GAEG;AACH,MAAM,MAAM,qBAAqB,CAAC,CAAC,SAAS,UAAU,IAAI,CAAC,SAAS,OAAO,GACvE,sBAAsB,GACtB,uBAAuB,CAAC;AAE5B;;GAEG;AACH,MAAM,MAAM,kCAAkC,CAAC,CAAC,SAAS,UAAU,IACjE,CAAC,SAAS,OAAO,GACb,uCAAuC,GACvC,wCAAwC,CAAC;AAE/C;;GAEG;AACH,MAAM,MAAM,kCAAkC,CAAC,CAAC,SAAS,UAAU,IACjE,CAAC,SAAS,OAAO,GACb,mCAAmC,GACnC,oCAAoC,CAAC;AAE3C;;GAEG;AACH,MAAM,MAAM,8BAA8B,GAAG,yBAAyB,CAAC;AAEvE;;GAEG;AACH,MAAM,MAAM,+BAA+B,GAAG,yBAAyB,CAAC;AAExE;;GAEG;AACH,MAAM,MAAM,6BAA6B,CAAC,CAAC,SAAS,UAAU,IAC5D,CAAC,SAAS,OAAO,GACb,8BAA8B,GAC9B,+BAA+B,CAAC;AAEtC;;GAEG;AACH,MAAM,MAAM,wBAAwB,GAAG,SAAS,CAAC"}

package/dist/orchestration-types.js.map

@@ -1 +1 @@
-{"version":3,"file":"orchestration-types.js","sourceRoot":"","sources":["../src/orchestration-types.ts"],"names":[],"mappings":"AAqSA;;;GAGG;AACH,MAAM,CAAC,MAAM,8BAA8B,GAAG;IAC5C,UAAU,EAAE,CAAC;IACb,cAAc,EAAE,CAAC;IACjB,qBAAqB,EAAE,CAAC;IACxB,SAAS,EAAE,CAAC;CACJ,CAAC"}
+{"version":3,"file":"orchestration-types.js","sourceRoot":"","sources":["../src/orchestration-types.ts"],"names":[],"mappings":"AAwYA;;;GAGG;AACH,MAAM,CAAC,MAAM,8BAA8B,GAAG;IAC5C,UAAU,EAAE,CAAC;IACb,cAAc,EAAE,CAAC;IACjB,qBAAqB,EAAE,CAAC;IACxB,SAAS,EAAE,CAAC;CACJ,CAAC"}

package/dist/util/filtering.d.ts

@@ -1,25 +1,19 @@
-import type { InputFilteringConfig, LlamaGuard38BFilterConfig, OutputFilteringConfig } from '../client/api/schema/index.js';
-import type { AzureContentFilter, AzureContentSafety, AzureContentSafetyFilterConfig, LlamaGuardCategory } from '../orchestration-types.js';
-/**
- * Convenience function to build Azure content filter.
- * @param filter - Filtering configuration for Azure filter. If skipped, the default Azure content filter configuration is used.
- * @returns An object with the Azure filtering configuration.
- * @deprecated Since 1.8.0. Use {@link buildAzureContentSafetyFilter()} instead.
- */
-export declare function buildAzureContentFilter(filter?: AzureContentSafety): InputFilteringConfig | OutputFilteringConfig;
+import type { AzureContentSafetyFilterParameters, AzureContentSafetyFilterReturnType, LlamaGuard38BCategory, LlamaGuard38BFilterReturnType, ConfigType } from '../orchestration-types.js';
 /**
  * Convenience function to build Azure content filter.
+ * @param type - Type of the filter, either 'input' or 'output'.
  * @param config - Configuration for Azure content safety filter.
  * If skipped, the default configuration of `ALLOW_SAFE_LOW` is used for all filter categories.
- * @returns Filter config object.
- * @example "buildAzureContentSafetyFilter({ Hate: 'ALLOW_SAFE', Violence: 'ALLOW_SAFE_LOW_MEDIUM' })"
+ * @returns Azure content safety configuration.
+ * @example "buildAzureContentSafetyFilter({ type: 'input', hate: 'ALLOW_SAFE', violence: 'ALLOW_SAFE_LOW_MEDIUM' })"
  */
-export declare function buildAzureContentSafetyFilter(config?: AzureContentFilter): AzureContentSafetyFilterConfig;
+export declare function buildAzureContentSafetyFilter<T extends ConfigType>(type: T, config?: AzureContentSafetyFilterParameters<T>): AzureContentSafetyFilterReturnType<T>;
 /**
- * Convenience function to build Llama guard filter.
+ * Convenience function to build Llama Guard 3 8B filter.
+ * @param type - Type of the filter, either `input` or `output`.
  * @param categories - Categories to be enabled for filtering. Provide at least one category.
- * @returns Filter config object.
- * @example "buildLlamaGuardFilter('self_harm', 'hate')"
+ * @returns Llama Guard 3 8B filter configuration.
+ * @example buildLlamaGuard38BFilter('input', ['elections', 'hate'])
  */
-export declare function buildLlamaGuardFilter(...categories: [LlamaGuardCategory, ...LlamaGuardCategory[]]): LlamaGuard38BFilterConfig;
+export declare function buildLlamaGuard38BFilter<T extends ConfigType>(type: T, categories: [LlamaGuard38BCategory, ...LlamaGuard38BCategory[]]): LlamaGuard38BFilterReturnType<T>;
 //# sourceMappingURL=filtering.d.ts.map

package/dist/util/filtering.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"filtering.d.ts","sourceRoot":"","sources":["../../src/util/filtering.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,oBAAoB,EACpB,yBAAyB,EACzB,qBAAqB,EACtB,MAAM,+BAA+B,CAAC;AACvC,OAAO,KAAK,EACV,kBAAkB,EAClB,kBAAkB,EAClB,8BAA8B,EAE9B,kBAAkB,EACnB,MAAM,2BAA2B,CAAC;AAEnC;;;;;GAKG;AACH,wBAAgB,uBAAuB,CACrC,MAAM,CAAC,EAAE,kBAAkB,GAC1B,oBAAoB,GAAG,qBAAqB,CAY9C;AAED;;;;;;GAMG;AACH,wBAAgB,6BAA6B,CAC3C,MAAM,CAAC,EAAE,kBAAkB,GAC1B,8BAA8B,CAiBhC;AAED;;;;;GAKG;AACH,wBAAgB,qBAAqB,CACnC,GAAG,UAAU,EAAE,CAAC,kBAAkB,EAAE,GAAG,kBAAkB,EAAE,CAAC,GAC3D,yBAAyB,CAO3B"}
+{"version":3,"file":"filtering.d.ts","sourceRoot":"","sources":["../../src/util/filtering.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAEV,kCAAkC,EAClC,kCAAkC,EAElC,qBAAqB,EACrB,6BAA6B,EAC7B,UAAU,EACX,MAAM,2BAA2B,CAAC;AAEnC;;;;;;;GAOG;AACH,wBAAgB,6BAA6B,CAAC,CAAC,SAAS,UAAU,EAChE,IAAI,EAAE,CAAC,EACP,MAAM,CAAC,EAAE,kCAAkC,CAAC,CAAC,CAAC,GAC7C,kCAAkC,CAAC,CAAC,CAAC,CAuCvC;AAED;;;;;;GAMG;AACH,wBAAgB,wBAAwB,CAAC,CAAC,SAAS,UAAU,EAC3D,IAAI,EAAE,CAAC,EACP,UAAU,EAAE,CAAC,qBAAqB,EAAE,GAAG,qBAAqB,EAAE,CAAC,GAC9D,6BAA6B,CAAC,CAAC,CAAC,CAOlC"}

package/dist/util/filtering.js

@@ -1,53 +1,52 @@
 import { supportedAzureFilterThresholds } from '../orchestration-types.js';
 /**
  * Convenience function to build Azure content filter.
- * @param filter - Filtering configuration for Azure filter. If skipped, the default Azure content filter configuration is used.
- * @returns An object with the Azure filtering configuration.
- * @deprecated Since 1.8.0. Use {@link buildAzureContentSafetyFilter()} instead.
- */
-export function buildAzureContentFilter(filter) {
-    if (filter && !Object.keys(filter).length) {
-        throw new Error('Filter property cannot be an empty object');
-    }
-    return {
-        filters: [
-            {
-                type: 'azure_content_safety',
-                ...(filter && { config: filter })
-            }
-        ]
-    };
-}
-/**
- * Convenience function to build Azure content filter.
+ * @param type - Type of the filter, either 'input' or 'output'.
  * @param config - Configuration for Azure content safety filter.
  * If skipped, the default configuration of `ALLOW_SAFE_LOW` is used for all filter categories.
- * @returns Filter config object.
- * @example "buildAzureContentSafetyFilter({ Hate: 'ALLOW_SAFE', Violence: 'ALLOW_SAFE_LOW_MEDIUM' })"
+ * @returns Azure content safety configuration.
+ * @example "buildAzureContentSafetyFilter({ type: 'input', hate: 'ALLOW_SAFE', violence: 'ALLOW_SAFE_LOW_MEDIUM' })"
  */
-export function buildAzureContentSafetyFilter(config) {
-    if (config && !Object.keys(config).length) {
-        throw new Error('Filtering configuration cannot be an empty object');
+export function buildAzureContentSafetyFilter(type, config) {
+    if (!config) {
+        return {
+            type: 'azure_content_safety'
+        };
     }
-    return {
-        type: 'azure_content_safety',
-        ...(config && {
+    if (!Object.keys(config).length) {
+        throw new Error('Filtering parameters cannot be empty');
+    }
+    if (type === 'input') {
+        const { prompt_shield, ...rest } = config;
+        return {
+            type: 'azure_content_safety',
             config: {
-                ...Object.fromEntries(Object.entries(config).map(([key, value]) => [
+                ...Object.fromEntries(Object.entries(rest).map(([key, value]) => [
                     key,
                     supportedAzureFilterThresholds[value]
-                ]))
+                ])),
+                ...(prompt_shield !== undefined && { prompt_shield })
             }
-        })
+        };
+    }
+    return {
+        type: 'azure_content_safety',
+        config: {
+            ...Object.fromEntries(Object.entries(config).map(([key, value]) => [
+                key,
+                supportedAzureFilterThresholds[value]
+            ]))
+        }
     };
 }
 /**
- * Convenience function to build Llama guard filter.
+ * Convenience function to build Llama Guard 3 8B filter.
+ * @param type - Type of the filter, either `input` or `output`.
  * @param categories - Categories to be enabled for filtering. Provide at least one category.
- * @returns Filter config object.
- * @example "buildLlamaGuardFilter('self_harm', 'hate')"
+ * @returns Llama Guard 3 8B filter configuration.
+ * @example buildLlamaGuard38BFilter('input', ['elections', 'hate'])
  */
-export function buildLlamaGuardFilter(...categories) {
+export function buildLlamaGuard38BFilter(type, categories) {
     return {
         type: 'llama_guard_3_8b',
         config: Object.fromEntries([...categories].map(category => [category, true]))

package/dist/util/filtering.js.map

@@ -1 +1 @@
-{"version":3,"file":"filtering.js","sourceRoot":"","sources":["../../src/util/filtering.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,8BAA8B,EAAE,MAAM,2BAA2B,CAAC;AAc3E;;;;;GAKG;AACH,MAAM,UAAU,uBAAuB,CACrC,MAA2B;IAE3B,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,CAAC;QAC1C,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;IAC/D,CAAC;IACD,OAAO;QACL,OAAO,EAAE;YACP;gBACE,IAAI,EAAE,sBAAsB;gBAC5B,GAAG,CAAC,MAAM,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;aAClC;SACF;KACF,CAAC;AACJ,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,6BAA6B,CAC3C,MAA2B;IAE3B,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,CAAC;QAC1C,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;IACvE,CAAC;IACD,OAAO;QACL,IAAI,EAAE,sBAAsB;QAC5B,GAAG,CAAC,MAAM,IAAI;YACZ,MAAM,EAAE;gBACN,GAAG,MAAM,CAAC,WAAW,CACnB,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC;oBAC3C,GAAG;oBACH,8BAA8B,CAAC,KAA6B,CAAC;iBAC9D,CAAC,CACH;aACF;SACF,CAAC;KACH,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,qBAAqB,CACnC,GAAG,UAAyD;IAE5D,OAAO;QACL,IAAI,EAAE,kBAAkB;QACxB,MAAM,EAAE,MAAM,CAAC,WAAW,CACxB,CAAC,GAAG,UAAU,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC,CAClD;KACF,CAAC;AACJ,CAAC"}
+{"version":3,"file":"filtering.js","sourceRoot":"","sources":["../../src/util/filtering.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,8BAA8B,EAAE,MAAM,2BAA2B,CAAC;AAe3E;;;;;;;GAOG;AACH,MAAM,UAAU,6BAA6B,CAC3C,IAAO,EACP,MAA8C;IAE9C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO;YACL,IAAI,EAAE,sBAAsB;SAC7B,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,CAAC;QAChC,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;IAC1D,CAAC;IAED,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;QACrB,MAAM,EAAE,aAAa,EAAE,GAAG,IAAI,EAAE,GAC9B,MAAiD,CAAC;QACpD,OAAO;YACL,IAAI,EAAE,sBAAsB;YAC5B,MAAM,EAAE;gBACN,GAAG,MAAM,CAAC,WAAW,CACnB,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC;oBACzC,GAAG;oBACH,8BAA8B,CAAC,KAA6B,CAAC;iBAC9D,CAAC,CACH;gBACD,GAAG,CAAC,aAAa,KAAK,SAAS,IAAI,EAAE,aAAa,EAAE,CAAC;aACtD;SACqC,CAAC;IAC3C,CAAC;IAED,OAAO;QACL,IAAI,EAAE,sBAAsB;QAC5B,MAAM,EAAE;YACN,GAAG,MAAM,CAAC,WAAW,CACnB,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC;gBAC3C,GAAG;gBACH,8BAA8B,CAAC,KAA6B,CAAC;aAC9D,CAAC,CACH;SACF;KACsC,CAAC;AAC5C,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,wBAAwB,CACtC,IAAO,EACP,UAA+D;IAE/D,OAAO;QACL,IAAI,EAAE,kBAAkB;QACxB,MAAM,EAAE,MAAM,CAAC,WAAW,CACxB,CAAC,GAAG,UAAU,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC,CAClD;KACF,CAAC;AACJ,CAAC"}

package/dist/util/grounding.d.ts

@@ -1,9 +1,8 @@
-import type { GroundingModuleConfig } from '../client/api/schema/index.js';
-import type { DocumentGroundingServiceConfig } from '../orchestration-types.js';
+import type { DocumentGroundingServiceConfig, GroundingModule } from '../orchestration-types.js';
 /**
  * Convenience function to create Document Grounding configuration.
  * @param groundingConfig - Configuration for the document grounding service.
  * @returns An object with the full grounding configuration.
  */
-export declare function buildDocumentGroundingConfig(groundingConfig: DocumentGroundingServiceConfig): GroundingModuleConfig;
+export declare function buildDocumentGroundingConfig(groundingConfig: DocumentGroundingServiceConfig): GroundingModule;
 //# sourceMappingURL=grounding.d.ts.map

package/dist/util/grounding.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"grounding.d.ts","sourceRoot":"","sources":["../../src/util/grounding.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,+BAA+B,CAAC;AAC3E,OAAO,KAAK,EAAE,8BAA8B,EAAE,MAAM,2BAA2B,CAAC;AAEhF;;;;GAIG;AACH,wBAAgB,4BAA4B,CAC1C,eAAe,EAAE,8BAA8B,GAC9C,qBAAqB,CAiBvB"}
+{"version":3,"file":"grounding.d.ts","sourceRoot":"","sources":["../../src/util/grounding.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,8BAA8B,EAC9B,eAAe,EAChB,MAAM,2BAA2B,CAAC;AAEnC;;;;GAIG;AACH,wBAAgB,4BAA4B,CAC1C,eAAe,EAAE,8BAA8B,GAC9C,eAAe,CAmBjB"}

package/dist/util/grounding.js

@@ -7,8 +7,10 @@ export function buildDocumentGroundingConfig(groundingConfig) {
     return {
         type: 'document_grounding_service',
         config: {
-            input_params: groundingConfig.input_params,
-            output_param: groundingConfig.output_param,
+            placeholders: {
+                input: groundingConfig.placeholders.input,
+                output: groundingConfig.placeholders.output
+            },
             ...(groundingConfig.filters && {
                 filters: groundingConfig.filters?.map(filter => ({
                     data_repository_type: 'vector',

package/dist/util/grounding.js.map

@@ -1 +1 @@
-{"version":3,"file":"grounding.js","sourceRoot":"","sources":["../../src/util/grounding.ts"],"names":[],"mappings":"AAGA;;;;GAIG;AACH,MAAM,UAAU,4BAA4B,CAC1C,eAA+C;IAE/C,OAAO;QACL,IAAI,EAAE,4BAA4B;QAClC,MAAM,EAAE;YACN,YAAY,EAAE,eAAe,CAAC,YAAY;YAC1C,YAAY,EAAE,eAAe,CAAC,YAAY;YAC1C,GAAG,CAAC,eAAe,CAAC,OAAO,IAAI;gBAC7B,OAAO,EAAE,eAAe,CAAC,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;oBAC/C,oBAAoB,EAAE,QAAQ;oBAC9B,GAAG,MAAM;iBACV,CAAC,CAAC;aACJ,CAAC;YACF,GAAG,CAAC,eAAe,CAAC,eAAe,IAAI;gBACrC,eAAe,EAAE,eAAe,CAAC,eAAe;aACjD,CAAC;SACH;KACF,CAAC;AACJ,CAAC"}
+{"version":3,"file":"grounding.js","sourceRoot":"","sources":["../../src/util/grounding.ts"],"names":[],"mappings":"AAKA;;;;GAIG;AACH,MAAM,UAAU,4BAA4B,CAC1C,eAA+C;IAE/C,OAAO;QACL,IAAI,EAAE,4BAA4B;QAClC,MAAM,EAAE;YACN,YAAY,EAAE;gBACZ,KAAK,EAAE,eAAe,CAAC,YAAY,CAAC,KAAK;gBACzC,MAAM,EAAE,eAAe,CAAC,YAAY,CAAC,MAAM;aAC5C;YACD,GAAG,CAAC,eAAe,CAAC,OAAO,IAAI;gBAC7B,OAAO,EAAE,eAAe,CAAC,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;oBAC/C,oBAAoB,EAAE,QAAQ;oBAC9B,GAAG,MAAM;iBACV,CAAC,CAAC;aACJ,CAAC;YACF,GAAG,CAAC,eAAe,CAAC,eAAe,IAAI;gBACrC,eAAe,EAAE,eAAe,CAAC,eAAe;aACjD,CAAC;SACH;KACF,CAAC;AACJ,CAAC"}