zack3d/big-agi
1
0
Fork 0
mirror of https://github.com/enricoros/big-AGI.git synced 2026-05-10 21:50:14 -07:00
Code Issues Packages Projects Releases Wiki Activity

LLMs: improve DModelVendor typings, and reverse Id->type mappings

Browse Source
This commit is contained in:
Enrico Ros
2026-04-09 14:25:30 -07:00
parent ba06d70c05
commit f4f4ad9373
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/modules/llms/vendors/vendors.registry.ts
Vendored
+9 -3
View File
@@ -45,8 +45,8 @@ export type ModelVendorId =
| 'zai'
;
/** Global: Vendor Instances Registry **/
const MODEL_VENDOR_REGISTRY: Record<ModelVendorId, IModelVendor> = {
/** Global: Vendor Instances Registry (`satisfies` validates keys, TS preserves specific vendor types) **/
const MODEL_VENDOR_REGISTRY = {
alibaba: ModelVendorAlibaba,
anthropic: ModelVendorAnthropic,
azure: ModelVendorAzure,
@@ -66,7 +66,13 @@ const MODEL_VENDOR_REGISTRY: Record<ModelVendorId, IModelVendor> = {
togetherai: ModelVendorTogetherAI,
xai: ModelVendorXAI,
zai: ModelVendorZAI,
} as Record<string, IModelVendor>;
} as const satisfies Record<ModelVendorId, IModelVendor>;
// --- Type Helpers - for Id -> concrete type mappings ---
// NOTE: we haven't ported the full system to type inference, this is just a way forward
export type ModelVendorOf<V extends ModelVendorId> = (typeof MODEL_VENDOR_REGISTRY)[V];
export type ModelVendorAccessOf<V extends ModelVendorId> = ModelVendorOf<V> extends IModelVendor<any, infer TAccess> ? TAccess : never;
export function findAllModelVendors(): IModelVendor[] {