@fgv/ts-extras 5.1.0-20 → 5.1.0-21

package/lib/packlets/ai-assist/apiClient.js

@@ -41,6 +41,7 @@ const ts_json_base_1 = require("@fgv/ts-json-base");
 const ts_utils_1 = require("@fgv/ts-utils");
 const model_1 = require("./model");
 const chatRequestBuilders_1 = require("./chatRequestBuilders");
+const endpoint_1 = require("./endpoint");
 const registry_1 = require("./registry");
 const toolFormats_1 = require("./toolFormats");
 // ============================================================================
@@ -287,9 +288,7 @@ async function callOpenAiCompletion(config, prompt, additionalMessages, temperat
         tail: additionalMessages
     });
     const body = { model: config.model, messages, temperature };
-    const headers = {
-        Authorization: `Bearer ${config.apiKey}`
-    };
+    const headers = (0, endpoint_1.bearerAuthHeader)(config.apiKey);
     /* c8 ignore next 1 - optional logger */
     logger === null || logger === void 0 ? void 0 : logger.info(`OpenAI completion: model=${config.model}`);
     const jsonResult = await fetchJson(url, headers, body, logger, signal);
@@ -342,9 +341,7 @@ async function callOpenAiResponsesCompletion(config, prompt, tools, additionalMe
         tools: (0, toolFormats_1.toResponsesApiTools)(tools),
         temperature
     };
-    const headers = {
-        Authorization: `Bearer ${config.apiKey}`
-    };
+    const headers = (0, endpoint_1.bearerAuthHeader)(config.apiKey);
     /* c8 ignore next 1 - optional logger */
     logger === null || logger === void 0 ? void 0 : logger.info(`OpenAI Responses API: model=${config.model}, tools=${tools.map((t) => t.type).join(',')}`);
     const jsonResult = await fetchJson(url, headers, body, logger, signal);
@@ -509,19 +506,24 @@ async function callGeminiCompletion(config, prompt, additionalMessages, temperat
  * @public
  */
 async function callProviderCompletion(params) {
-    const { descriptor, apiKey, prompt, additionalMessages, temperature = 0.7, modelOverride, logger, tools, signal } = params;
-    if (!descriptor.baseUrl) {
-        return (0, ts_utils_1.fail)(`provider "${descriptor.id}" has no API endpoint configured`);
+    const { descriptor, apiKey, prompt, additionalMessages, temperature = 0.7, modelOverride, logger, tools, signal, endpoint } = params;
+    const baseUrlResult = (0, endpoint_1.resolveEffectiveBaseUrl)(descriptor, endpoint);
+    if (baseUrlResult.isFailure()) {
+        return (0, ts_utils_1.fail)(baseUrlResult.message);
     }
     if (prompt.attachments.length > 0 && !descriptor.acceptsImageInput) {
         return (0, ts_utils_1.fail)(`provider "${descriptor.id}" does not accept image input`);
     }
     const hasTools = tools !== undefined && tools.length > 0;
     const modelContext = hasTools ? 'tools' : undefined;
+    const model = (0, model_1.resolveModel)(modelOverride !== null && modelOverride !== void 0 ? modelOverride : descriptor.defaultModel, modelContext);
+    if (model.length === 0) {
+        return (0, ts_utils_1.fail)(`provider "${descriptor.id}": no model resolved; pass modelOverride or set descriptor.defaultModel`);
+    }
     const config = {
-        baseUrl: descriptor.baseUrl,
+        baseUrl: baseUrlResult.value,
         apiKey,
-        model: (0, model_1.resolveModel)(modelOverride !== null && modelOverride !== void 0 ? modelOverride : descriptor.defaultModel, modelContext)
+        model
     };
     /* c8 ignore next 8 - optional logger diagnostic output */
     if (logger) {
@@ -615,9 +617,7 @@ async function callOpenAiImageGeneration(config, request, defaultMimeType, logge
     var _a, _b, _c;
     const opts = (_a = request.options) !== null && _a !== void 0 ? _a : {};
     const refs = (_b = request.referenceImages) !== null && _b !== void 0 ? _b : [];
-    const headers = {
-        Authorization: `Bearer ${config.apiKey}`
-    };
+    const headers = (0, endpoint_1.bearerAuthHeader)(config.apiKey);
     const n = (_c = opts.count) !== null && _c !== void 0 ? _c : 1;
     const fetched = refs.length > 0
         ? await callOpenAiImagesEdits(config, request, headers, n, refs, logger, signal)
@@ -803,14 +803,20 @@ async function callImagenGeneration(config, request, logger, signal) {
  */
 async function callProviderImageGeneration(params) {
     var _a, _b;
-    const { descriptor, apiKey, params: request, modelOverride, logger, signal } = params;
+    const { descriptor, apiKey, params: request, modelOverride, logger, signal, endpoint } = params;
     if (!(0, registry_1.supportsImageGeneration)(descriptor)) {
         return (0, ts_utils_1.fail)(`provider "${descriptor.id}" does not support image generation`);
     }
-    if (!descriptor.baseUrl) {
-        return (0, ts_utils_1.fail)(`provider "${descriptor.id}" has no API endpoint configured`);
+    const baseUrlResult = (0, endpoint_1.resolveEffectiveBaseUrl)(descriptor, endpoint);
+    if (baseUrlResult.isFailure()) {
+        return (0, ts_utils_1.fail)(baseUrlResult.message);
     }
     const model = (0, model_1.resolveModel)(modelOverride !== null && modelOverride !== void 0 ? modelOverride : descriptor.defaultModel, 'image');
+    if (model.length === 0) {
+        return (0, ts_utils_1.fail)(`provider "${descriptor.id}": no image model resolved; ` +
+            `pass modelOverride or set descriptor.defaultModel ` +
+            `(a plain string, or an object with an "image" entry)`);
+    }
     const capability = (0, registry_1.resolveImageCapability)(descriptor, model);
     if (capability === undefined) {
         return (0, ts_utils_1.fail)(`provider "${descriptor.id}" does not support image generation for model "${model}"`);
@@ -819,7 +825,7 @@ async function callProviderImageGeneration(params) {
         return (0, ts_utils_1.fail)(`model "${model}" does not support reference images`);
     }
     const config = {
-        baseUrl: descriptor.baseUrl,
+        baseUrl: baseUrlResult.value,
         apiKey,
         model
     };
@@ -946,9 +952,7 @@ function buildModelInfo(providerId, id, nativeCapabilities, nativeDisplayName, c
  */
 async function callOpenAiListModels(config, providerId, capabilityConfig, logger, signal) {
     const url = `${config.baseUrl}/models`;
-    const headers = {
-        Authorization: `Bearer ${config.apiKey}`
-    };
+    const headers = (0, endpoint_1.bearerAuthHeader)(config.apiKey);
     /* c8 ignore next 1 - optional logger */
     logger === null || logger === void 0 ? void 0 : logger.info(`List models: provider=${providerId}, format=openai`);
     const jsonResult = await fetchGetJson(url, headers, logger, signal);
@@ -1035,12 +1039,13 @@ async function callGeminiListModels(config, providerId, capabilityConfig, logger
  * @public
  */
 async function callProviderListModels(params) {
-    const { descriptor, apiKey, capability, capabilityConfig, logger, signal } = params;
-    if (!descriptor.baseUrl) {
-        return (0, ts_utils_1.fail)(`provider "${descriptor.id}" has no API endpoint configured`);
+    const { descriptor, apiKey, capability, capabilityConfig, logger, signal, endpoint } = params;
+    const baseUrlResult = (0, endpoint_1.resolveEffectiveBaseUrl)(descriptor, endpoint);
+    if (baseUrlResult.isFailure()) {
+        return (0, ts_utils_1.fail)(baseUrlResult.message);
     }
     const config = {
-        baseUrl: descriptor.baseUrl,
+        baseUrl: baseUrlResult.value,
         apiKey,
         model: '' // unused by listing
     };