diff options
Diffstat (limited to 'packages/credential-store/src/registry.ts')
| -rw-r--r-- | packages/credential-store/src/registry.ts | 176 |
1 files changed, 88 insertions, 88 deletions
diff --git a/packages/credential-store/src/registry.ts b/packages/credential-store/src/registry.ts index 89d8084..70abf9a 100644 --- a/packages/credential-store/src/registry.ts +++ b/packages/credential-store/src/registry.ts @@ -1,104 +1,104 @@ import type { ModelInfo, ProviderContract } from "@dispatch/kernel"; export interface Credential { - readonly name: string; - readonly providerId: string; + readonly name: string; + readonly providerId: string; } export interface ResolvedModel { - readonly providerId: string; - readonly model: string; + readonly providerId: string; + readonly model: string; } export interface CredentialStore { - /** - * Split a model name on the FIRST "/": name=before, model=after (model may contain "/"). - * Look up the credential by name; return its providerId + the model id, or undefined if - * the name is unknown or there is no model segment. - */ - resolve(modelName: string): ResolvedModel | undefined; - - /** - * The model catalog: for each credential, look up its provider and call listModels(), - * emitting `${credential.name}/${modelInfo.id}`. Skip credentials whose provider is - * missing or has no listModels. - */ - listCatalog(): Promise<readonly string[]>; - - /** - * Returns the full `ModelInfo` for a `<credentialName>/<model>` string, or - * undefined if unknown. Caches the result of `listModels` per credential. - * Used to look up `contextWindow` for auto-compaction. - */ - getModelInfo(modelName: string): Promise<ModelInfo | undefined>; + /** + * Split a model name on the FIRST "/": name=before, model=after (model may contain "/"). + * Look up the credential by name; return its providerId + the model id, or undefined if + * the name is unknown or there is no model segment. + */ + resolve(modelName: string): ResolvedModel | undefined; + + /** + * The model catalog: for each credential, look up its provider and call listModels(), + * emitting `${credential.name}/${modelInfo.id}`. Skip credentials whose provider is + * missing or has no listModels. + */ + listCatalog(): Promise<readonly string[]>; + + /** + * Returns the full `ModelInfo` for a `<credentialName>/<model>` string, or + * undefined if unknown. Caches the result of `listModels` per credential. + * Used to look up `contextWindow` for auto-compaction. + */ + getModelInfo(modelName: string): Promise<ModelInfo | undefined>; } export interface CredentialStoreDeps { - readonly credentials: readonly Credential[]; - readonly getProvider: (id: string) => ProviderContract | undefined; + readonly credentials: readonly Credential[]; + readonly getProvider: (id: string) => ProviderContract | undefined; } export function createCredentialStore(deps: CredentialStoreDeps): CredentialStore { - const credentialMap = new Map<string, string>(); - for (const credential of deps.credentials) { - credentialMap.set(credential.name, credential.providerId); - } - - return { - resolve(modelName: string): ResolvedModel | undefined { - const slashIndex = modelName.indexOf("/"); - if (slashIndex === -1) { - return undefined; - } - - const credentialName = modelName.slice(0, slashIndex); - const model = modelName.slice(slashIndex + 1); - - if (!model) { - return undefined; - } - - const providerId = credentialMap.get(credentialName); - if (!providerId) { - return undefined; - } - - return { providerId, model }; - }, - - async listCatalog(): Promise<readonly string[]> { - const results: string[] = []; - - for (const credential of deps.credentials) { - const provider = deps.getProvider(credential.providerId); - if (!provider?.listModels) { - continue; - } - - const models = await provider.listModels(); - for (const model of models) { - results.push(`${credential.name}/${model.id}`); - } - } - - return results; - }, - - async getModelInfo(modelName: string): Promise<ModelInfo | undefined> { - const slashIndex = modelName.indexOf("/"); - if (slashIndex === -1) return undefined; - const credentialName = modelName.slice(0, slashIndex); - const modelId = modelName.slice(slashIndex + 1); - if (!modelId) return undefined; - - const providerId = credentialMap.get(credentialName); - if (providerId === undefined) return undefined; - - const provider = deps.getProvider(providerId); - if (provider?.listModels === undefined) return undefined; - - const models = await provider.listModels(); - return models.find((m) => m.id === modelId); - }, - }; + const credentialMap = new Map<string, string>(); + for (const credential of deps.credentials) { + credentialMap.set(credential.name, credential.providerId); + } + + return { + resolve(modelName: string): ResolvedModel | undefined { + const slashIndex = modelName.indexOf("/"); + if (slashIndex === -1) { + return undefined; + } + + const credentialName = modelName.slice(0, slashIndex); + const model = modelName.slice(slashIndex + 1); + + if (!model) { + return undefined; + } + + const providerId = credentialMap.get(credentialName); + if (!providerId) { + return undefined; + } + + return { providerId, model }; + }, + + async listCatalog(): Promise<readonly string[]> { + const results: string[] = []; + + for (const credential of deps.credentials) { + const provider = deps.getProvider(credential.providerId); + if (!provider?.listModels) { + continue; + } + + const models = await provider.listModels(); + for (const model of models) { + results.push(`${credential.name}/${model.id}`); + } + } + + return results; + }, + + async getModelInfo(modelName: string): Promise<ModelInfo | undefined> { + const slashIndex = modelName.indexOf("/"); + if (slashIndex === -1) return undefined; + const credentialName = modelName.slice(0, slashIndex); + const modelId = modelName.slice(slashIndex + 1); + if (!modelId) return undefined; + + const providerId = credentialMap.get(credentialName); + if (providerId === undefined) return undefined; + + const provider = deps.getProvider(providerId); + if (provider?.listModels === undefined) return undefined; + + const models = await provider.listModels(); + return models.find((m) => m.id === modelId); + }, + }; } |
