@wix/auto_sdk_ai-gateway_prompts 1.0.0 → 1.0.2

package/build/cjs/index.js

@@ -29,6 +29,7 @@ __export(index_exports, {
   ClipGuidancePreset: () => ClipGuidancePreset,
   ContentRole: () => ContentRole,
   CreatePredictionModel: () => CreatePredictionModel,
+  DynamicRetrievalConfigMode: () => DynamicRetrievalConfigMode,
   EditImageWithPromptRequestModel: () => EditImageWithPromptRequestModel,
   FinishReason: () => FinishReason,
   GatewayMessageDefinitionRole: () => GatewayMessageDefinitionRole,
@@ -92,7 +93,8 @@ __export(index_exports, {
   V1ToolChoiceType: () => V1ToolChoiceType,
   VideoGenModel: () => VideoGenModel,
   VideoModel: () => VideoModel,
-  generateContentByPromptObject: () => generateContentByPromptObject4
+  generateContentByPromptObject: () => generateContentByPromptObject4,
+  generateTextByPromptObjectStreamed: () => generateTextByPromptObjectStreamed4
 });
 module.exports = __toCommonJS(index_exports);
 
@@ -107,7 +109,14 @@ var import_bytes = require("@wix/sdk-runtime/transformations/bytes");
 var import_transform_paths = require("@wix/sdk-runtime/transformations/transform-paths");
 var import_rest_modules = require("@wix/sdk-runtime/rest-modules");
 function resolveWixApiInfraV1WixAiExternalGatewayUrl(opts) {
-  const domainToMappings = {};
+  const domainToMappings = {
+    "www.wixapis.com": [
+      {
+        srcPath: "/ai-external-gateway-poc",
+        destPath: ""
+      }
+    ]
+  };
   return (0, import_rest_modules.resolveUrl)(Object.assign(opts, { domainToMappings }));
 }
 var PACKAGE_NAME = "@wix/auto_sdk_ai-gateway_prompts";
@@ -407,6 +416,141 @@ function generateContentByPromptObject(payload) {
   }
   return __generateContentByPromptObject;
 }
+function generateTextByPromptObjectStreamed(payload) {
+  function __generateTextByPromptObjectStreamed({ host }) {
+    const serializedData = (0, import_transform_paths.transformPaths)(payload, [
+      {
+        transformFn: import_float.transformSDKFloatToRESTFloat,
+        paths: [
+          { path: "prompt.openAiChatCompletionRequest.temperature" },
+          { path: "prompt.openAiChatCompletionRequest.topP" },
+          { path: "prompt.openAiChatCompletionRequest.presencePenalty" },
+          { path: "prompt.openAiChatCompletionRequest.frequencyPenalty" },
+          { path: "prompt.googleTextBisonRequest.parameters.temperature" },
+          { path: "prompt.googleTextBisonRequest.parameters.topP" },
+          { path: "prompt.googleChatBisonRequest.parameters.temperature" },
+          { path: "prompt.googleChatBisonRequest.parameters.topP" },
+          { path: "prompt.azureChatCompletionRequest.temperature" },
+          { path: "prompt.azureChatCompletionRequest.topP" },
+          { path: "prompt.azureChatCompletionRequest.presencePenalty" },
+          { path: "prompt.azureChatCompletionRequest.frequencyPenalty" },
+          {
+            path: "prompt.googleGeminiGenerateContentRequest.generationConfig.temperature"
+          },
+          {
+            path: "prompt.googleGeminiGenerateContentRequest.generationConfig.topP"
+          },
+          { path: "prompt.anthropicClaudeRequest.temperature" },
+          { path: "prompt.anthropicClaudeRequest.topP" },
+          { path: "prompt.googleAnthropicClaudeRequest.temperature" },
+          { path: "prompt.googleAnthropicClaudeRequest.topP" },
+          { path: "prompt.invokeAnthropicModelRequest.temperature" },
+          { path: "prompt.invokeAnthropicModelRequest.topP" },
+          { path: "prompt.stabilityAiTextToImageRequest.textPrompts.weight" },
+          { path: "prompt.stabilityAiStableDiffusionRequest.strength" },
+          {
+            path: "prompt.blackForestLabsGenerateImageRequest.imagePromptStrength"
+          },
+          { path: "prompt.replicateCreatePredictionRequest.fluxPulid.trueCfg" },
+          {
+            path: "prompt.replicateCreatePredictionRequest.fluxPulid.idWeight"
+          },
+          {
+            path: "prompt.replicateCreatePredictionRequest.fluxPulid.guidanceScale"
+          },
+          {
+            path: "prompt.replicateCreatePredictionRequest.fluxDevControlnet.loraStrength"
+          },
+          {
+            path: "prompt.replicateCreatePredictionRequest.fluxDevControlnet.guidanceScale"
+          },
+          {
+            path: "prompt.replicateCreatePredictionRequest.fluxDevControlnet.controlStrength"
+          },
+          {
+            path: "prompt.replicateCreatePredictionRequest.fluxDevControlnet.imageToImageStrength"
+          },
+          { path: "prompt.stabilityAiEditWithPromptRequest.creativity" },
+          { path: "prompt.runwareTextToImageRequest.strength" },
+          { path: "prompt.mlPlatformGenerateImageRequest.fluxPulid.trueCfg" },
+          { path: "prompt.mlPlatformGenerateImageRequest.fluxPulid.idWeight" },
+          {
+            path: "prompt.mlPlatformGenerateImageRequest.fluxPulid.guidanceScale"
+          },
+          { path: "prompt.googleCreateChatCompletionRequest.temperature" },
+          { path: "prompt.googleCreateChatCompletionRequest.topP" },
+          { path: "prompt.googleCreateChatCompletionRequest.presencePenalty" },
+          { path: "prompt.googleCreateChatCompletionRequest.frequencyPenalty" },
+          { path: "prompt.mlPlatformOpenAiRawRequest.temperature" },
+          { path: "prompt.mlPlatformOpenAiRawRequest.topP" },
+          { path: "prompt.mlPlatformOpenAiRawRequest.presencePenalty" },
+          { path: "prompt.mlPlatformOpenAiRawRequest.frequencyPenalty" },
+          { path: "prompt.openAiResponsesRequest.temperature" },
+          { path: "prompt.openAiResponsesRequest.topP" },
+          { path: "prompt.azureOpenAiResponsesRequest.temperature" },
+          { path: "prompt.azureOpenAiResponsesRequest.topP" },
+          { path: "prompt.llamaModelRequest.temperature" },
+          { path: "prompt.llamaModelRequest.topP" },
+          { path: "prompt.runwareTextToImageRequest.cfgScale" },
+          { path: "prompt.runwareTextToImageRequest.loraModels.weight" },
+          { path: "prompt.mlPlatformLlamaModelRequest.temperature" },
+          { path: "prompt.mlPlatformLlamaModelRequest.topP" },
+          { path: "prompt.perplexityChatCompletionRequest.temperature" },
+          { path: "prompt.perplexityChatCompletionRequest.topP" },
+          { path: "prompt.perplexityChatCompletionRequest.topK" },
+          { path: "prompt.perplexityChatCompletionRequest.presencePenalty" },
+          { path: "prompt.perplexityChatCompletionRequest.frequencyPenalty" },
+          { path: "prompt.runwareVideoInferenceRequest.cfgScale" }
+        ]
+      }
+    ]);
+    const metadata = {
+      entityFqdn: "wix.api_infra.v1.prompt_proxy",
+      method: "POST",
+      methodFqn: "wix.api_infra.v1.WixAiExternalGateway.GenerateTextByPromptObjectStreamed",
+      packageName: PACKAGE_NAME,
+      migrationOptions: {
+        optInTransformResponse: true
+      },
+      url: resolveWixApiInfraV1WixAiExternalGatewayUrl({
+        protoPath: "/v1/generate-by-prompt-object-streamed",
+        data: serializedData,
+        host
+      }),
+      data: serializedData,
+      transformResponse: (payload2) => (0, import_transform_paths.transformPaths)(payload2, [
+        {
+          transformFn: import_float2.transformRESTFloatToSDKFloat,
+          paths: [
+            {
+              path: "googleGeminiStreamChunk.candidates.safetyRatings.probabilityScore"
+            },
+            {
+              path: "googleGeminiStreamChunk.candidates.safetyRatings.severityScore"
+            },
+            {
+              path: "googleGeminiStreamChunk.candidates.groundingMetadata.groundingSupports.confidenceScores",
+              isRepeated: true
+            },
+            {
+              path: "googleGeminiStreamChunk.candidates.groundingMetadata.retrievalMetadata.googleSearchDynamicRetrievalScore"
+            }
+          ]
+        },
+        {
+          transformFn: import_bytes.transformRESTBytesToSDKBytes,
+          paths: [
+            {
+              path: "googleGeminiStreamChunk.candidates.groundingMetadata.searchEntryPoint.sdkBlob"
+            }
+          ]
+        }
+      ])
+    };
+    return metadata;
+  }
+  return __generateTextByPromptObjectStreamed;
+}
 
 // src/api-infra-v1-prompt-proxy-prompts.universal.ts
 var OpenaiproxyV1Model = /* @__PURE__ */ ((OpenaiproxyV1Model2) => {
@@ -537,11 +681,11 @@ var Outcome = /* @__PURE__ */ ((Outcome2) => {
   Outcome2["OUTCOME_DEADLINE_EXCEEDED"] = "OUTCOME_DEADLINE_EXCEEDED";
   return Outcome2;
 })(Outcome || {});
-var Mode = /* @__PURE__ */ ((Mode2) => {
-  Mode2["MODE_UNSPECIFIED"] = "MODE_UNSPECIFIED";
-  Mode2["MODE_DYNAMIC"] = "MODE_DYNAMIC";
-  return Mode2;
-})(Mode || {});
+var DynamicRetrievalConfigMode = /* @__PURE__ */ ((DynamicRetrievalConfigMode2) => {
+  DynamicRetrievalConfigMode2["MODE_UNSPECIFIED"] = "MODE_UNSPECIFIED";
+  DynamicRetrievalConfigMode2["MODE_DYNAMIC"] = "MODE_DYNAMIC";
+  return DynamicRetrievalConfigMode2;
+})(DynamicRetrievalConfigMode || {});
 var HarmCategory = /* @__PURE__ */ ((HarmCategory2) => {
   HarmCategory2["UNKNOWN_CATEGORY"] = "UNKNOWN_CATEGORY";
   HarmCategory2["HARM_CATEGORY_SEXUALLY_EXPLICIT"] = "HARM_CATEGORY_SEXUALLY_EXPLICIT";
@@ -565,6 +709,14 @@ var Modality = /* @__PURE__ */ ((Modality2) => {
   Modality2["AUDIO"] = "AUDIO";
   return Modality2;
 })(Modality || {});
+var Mode = /* @__PURE__ */ ((Mode2) => {
+  Mode2["UNKNOWN"] = "UNKNOWN";
+  Mode2["AUTO"] = "AUTO";
+  Mode2["ANY"] = "ANY";
+  Mode2["NONE"] = "NONE";
+  Mode2["VALIDATED"] = "VALIDATED";
+  return Mode2;
+})(Mode || {});
 var Type = /* @__PURE__ */ ((Type2) => {
   Type2["UNKNOWN"] = "UNKNOWN";
   Type2["EPHEMERAL"] = "EPHEMERAL";
@@ -1078,6 +1230,47 @@ async function generateContentByPromptObject2(options) {
     throw transformedError;
   }
 }
+async function generateTextByPromptObjectStreamed2(options) {
+  const { httpClient, sideEffects } = arguments[1];
+  const payload = (0, import_rename_all_nested_keys.renameKeysFromSDKRequestToRESTRequest)({
+    prompt: options?.prompt,
+    params: options?.params,
+    userRequestInfo: options?.userRequestInfo,
+    fallbackProperties: options?.fallbackProperties,
+    dynamicProperties: options?.dynamicProperties,
+    dynamicRequestConfig: options?.dynamicRequestConfig,
+    asyncResultTopic: options?.asyncResultTopic
+  });
+  const reqOpts = generateTextByPromptObjectStreamed(
+    payload
+  );
+  sideEffects?.onSiteCall?.();
+  try {
+    const result = await httpClient.request(reqOpts);
+    sideEffects?.onSuccess?.(result);
+    return (0, import_rename_all_nested_keys.renameKeysFromRESTResponseToSDKResponse)(result.data);
+  } catch (err) {
+    const transformedError = (0, import_transform_error.transformError)(
+      err,
+      {
+        spreadPathsToArguments: {},
+        explicitPathsToArguments: {
+          prompt: "$[0].prompt",
+          params: "$[0].params",
+          userRequestInfo: "$[0].userRequestInfo",
+          fallbackProperties: "$[0].fallbackProperties",
+          dynamicProperties: "$[0].dynamicProperties",
+          dynamicRequestConfig: "$[0].dynamicRequestConfig",
+          asyncResultTopic: "$[0].asyncResultTopic"
+        },
+        singleArgumentUnchanged: false
+      },
+      ["options"]
+    );
+    sideEffects?.onError?.(err);
+    throw transformedError;
+  }
+}
 
 // src/api-infra-v1-prompt-proxy-prompts.public.ts
 function generateContentByPromptObject3(httpClient) {
@@ -1087,10 +1280,18 @@ function generateContentByPromptObject3(httpClient) {
     { httpClient }
   );
 }
+function generateTextByPromptObjectStreamed3(httpClient) {
+  return (options) => generateTextByPromptObjectStreamed2(
+    options,
+    // @ts-ignore
+    { httpClient }
+  );
+}
 
 // src/api-infra-v1-prompt-proxy-prompts.context.ts
 var import_rest_modules2 = require("@wix/sdk-runtime/rest-modules");
 var generateContentByPromptObject4 = /* @__PURE__ */ (0, import_rest_modules2.createRESTModule)(generateContentByPromptObject3);
+var generateTextByPromptObjectStreamed4 = /* @__PURE__ */ (0, import_rest_modules2.createRESTModule)(generateTextByPromptObjectStreamed3);
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   AnthropicModel,
@@ -1102,6 +1303,7 @@ var generateContentByPromptObject4 = /* @__PURE__ */ (0, import_rest_modules2.cr
   ClipGuidancePreset,
   ContentRole,
   CreatePredictionModel,
+  DynamicRetrievalConfigMode,
   EditImageWithPromptRequestModel,
   FinishReason,
   GatewayMessageDefinitionRole,
@@ -1165,6 +1367,7 @@ var generateContentByPromptObject4 = /* @__PURE__ */ (0, import_rest_modules2.cr
   V1ToolChoiceType,
   VideoGenModel,
   VideoModel,
-  generateContentByPromptObject
+  generateContentByPromptObject,
+  generateTextByPromptObjectStreamed
 });
 //# sourceMappingURL=index.js.map