@pinecone-database/pinecone
Version:
This is the official Node.js SDK for [Pinecone](https://www.pinecone.io), written in TypeScript.
69 lines • 2.92 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.AssistantHostSingleton = void 0;
const utils_1 = require("../utils");
const asstControlOperationsBuilder_1 = require("./control/asstControlOperationsBuilder");
const control_1 = require("./control");
exports.AssistantHostSingleton = (function () {
const hostUrls = {}; // map of apiKey-assistantName to hostUrl
function ensureAssistantPath(url) {
if (!url.endsWith('/assistant')) {
// Append "/assistant" if it doesn't already end with it
url = url.endsWith('/') ? `${url}assistant` : `${url}/assistant`;
}
return url;
}
const _describeAssistant = async (config, assistantName) => {
const assistantControlApi = (0, asstControlOperationsBuilder_1.asstControlOperationsBuilder)(config);
const describeResponse = await (0, control_1.describeAssistant)(assistantControlApi)(assistantName);
const host = describeResponse?.host;
if (!host) {
// if the host is empty for some reason, default based on region
let defaultHost = 'https://prod-1-data.ke.pinecone.io';
// If 'eu' is specified use that, otherwise default to 'us'
if (config.assistantRegion &&
config.assistantRegion.toLowerCase() === 'eu') {
defaultHost = 'https://prod-eu-data.ke.pinecone.io';
}
return defaultHost;
}
else {
return host;
}
};
const _key = (config, assistantName) => `${config.apiKey}-${assistantName}`;
// singleton object
const singleton = {
getHostUrl: async (config, assistantName) => {
const cacheKey = _key(config, assistantName);
if (cacheKey in hostUrls) {
return hostUrls[cacheKey];
}
else {
const hostUrl = await _describeAssistant(config, assistantName);
hostUrls[cacheKey] = (0, utils_1.normalizeUrl)(ensureAssistantPath(hostUrl));
}
return hostUrls[cacheKey];
},
_reset: () => {
for (const key of Object.keys(hostUrls)) {
delete hostUrls[key];
}
},
_set: (config, assistantName, hostUrl) => {
const normalizedHostUrl = (0, utils_1.normalizeUrl)(ensureAssistantPath(hostUrl));
// prevent adding an empty hostUrl to the cache
if (!hostUrl || !normalizedHostUrl) {
return;
}
const cacheKey = _key(config, assistantName);
hostUrls[cacheKey] = normalizedHostUrl;
},
_delete: (config, assistantName) => {
const cacheKey = _key(config, assistantName);
delete hostUrls[cacheKey];
},
};
return singleton;
})();
//# sourceMappingURL=assistantHostSingleton.js.map