diff --git a/src/collections.ts b/src/collections.ts index b59b094..e68ff65 100644 --- a/src/collections.ts +++ b/src/collections.ts @@ -33,6 +33,15 @@ export interface Collection { includeByDefault?: boolean; // Include in queries by default (default: true) } +/** + * Model configuration for embedding, reranking, and generation + */ +export interface ModelsConfig { + embed?: string; + rerank?: string; + generate?: string; +} + /** * The complete configuration file structure */ @@ -41,6 +50,7 @@ export interface CollectionConfig { editor_uri?: string; // Editor URI template for terminal hyperlinks editor_uri_template?: string; // Alias for editor_uri collections: Record; // Collection name -> config + models?: ModelsConfig; } /**