Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 13 additions & 0 deletions packages/qvac-lib-registry-server/client/index.d.ts
Original file line number Diff line number Diff line change
Expand Up @@ -56,6 +56,17 @@ export interface QVACModelQuery {
lt?: Record<string, any>
}

export interface FindByParams {
/** Filter by name (partial match, case-insensitive) */
name?: string
/** Filter by engine (exact match) */
engine?: string
/** Filter by quantization (partial match, case-insensitive) */
quantization?: string
/** Include deprecated models (default: false) */
includeDeprecated?: boolean
}

export class QVACRegistryClient {
constructor (opts?: QVACRegistryClientOptions)

Expand All @@ -64,6 +75,8 @@ export class QVACRegistryClient {

getModel (path: string, source: string): Promise<QVACModelEntry | null>
downloadModel (path: string, source: string, options?: QVACDownloadOptions): Promise<QVACDownloadResult>

findBy (params?: FindByParams): Promise<QVACModelEntry[]>
findModels (query?: QVACModelQuery): Promise<QVACModelEntry[]>
findModelsByEngine (query?: QVACModelQuery): Promise<QVACModelEntry[]>
findModelsByName (query?: QVACModelQuery): Promise<QVACModelEntry[]>
Expand Down
16 changes: 16 additions & 0 deletions packages/qvac-lib-registry-server/client/lib/client.js
Original file line number Diff line number Diff line change
Expand Up @@ -129,6 +129,22 @@ class QVACRegistryClient extends ReadyResource {
return this.db.findModelsByQuantization(query).toArray()
}

/**
* Find models with optional filters.
* Uses the database's findBy method for efficient indexed queries.
* @param {Object} params - Filter parameters
* @param {string} [params.name] - Filter by name (partial match)
* @param {string} [params.engine] - Filter by engine (exact match)
* @param {string} [params.quantization] - Filter by quantization (partial match)
* @param {boolean} [params.includeDeprecated=false] - Include deprecated models
* @returns {Promise<Array>} Array of matching models
*/
async findBy (params = {}) {
await this.ready()
this.logger.debug('findBy called', { params })
return this.db.findBy(params)
}

_validateString (value, name) {
if (typeof value !== 'string' || value.length === 0) {
throw new Error(`Invalid ${name}: ${value}`)
Expand Down
2 changes: 1 addition & 1 deletion packages/qvac-lib-registry-server/client/package.json
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@
},
"dependencies": {
"@qvac/error": "^0.1.0",
"@tetherto/qvac-registry-schema-mono": "^0.1.0",
"@tetherto/qvac-registry-schema-mono": "^0.2.1",
"b4a": "^1.6.7",
"bare-fs": "^2.1.5",
"bare-os": "^2.2.0",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@ test('client api surface', async t => {
t.ok(QVACRegistryClient.prototype.findModelsByEngine, 'has findModelsByEngine method')
t.ok(QVACRegistryClient.prototype.findModelsByName, 'has findModelsByName method')
t.ok(QVACRegistryClient.prototype.findModelsByQuantization, 'has findModelsByQuantization method')
t.ok(QVACRegistryClient.prototype.findBy, 'has findBy method')
t.ok(QVACRegistryClient.prototype.ready, 'has ready method')
t.ok(QVACRegistryClient.prototype.close, 'has close method')
})
Expand All @@ -24,4 +25,5 @@ test('client find methods are async functions', async t => {
t.ok(QVACRegistryClient.prototype.findModelsByEngine.constructor.name === 'AsyncFunction', 'findModelsByEngine is async')
t.ok(QVACRegistryClient.prototype.findModelsByName.constructor.name === 'AsyncFunction', 'findModelsByName is async')
t.ok(QVACRegistryClient.prototype.findModelsByQuantization.constructor.name === 'AsyncFunction', 'findModelsByQuantization is async')
t.ok(QVACRegistryClient.prototype.findBy.constructor.name === 'AsyncFunction', 'findBy is async')
})
Loading