LiuVaayne
67 lines
1.8 KiB
TypeScript
67 lines
1.8 KiB
TypeScript
import express, { Request, Response } from 'express'
|
|
|
|
import { loggerService } from '../../services/LoggerService'
|
|
import { chatCompletionService } from '../services/chat-completion'
|
|
|
|
const logger = loggerService.withContext('ApiServerModelsRoutes')
|
|
|
|
const router = express.Router()
|
|
|
|
/**
|
|
* @swagger
|
|
* /v1/models:
|
|
* get:
|
|
* summary: List available models
|
|
* description: Returns a list of available AI models from all configured providers
|
|
* tags: [Models]
|
|
* responses:
|
|
* 200:
|
|
* description: List of available models
|
|
* content:
|
|
* application/json:
|
|
* schema:
|
|
* type: object
|
|
* properties:
|
|
* object:
|
|
* type: string
|
|
* example: list
|
|
* data:
|
|
* type: array
|
|
* items:
|
|
* $ref: '#/components/schemas/Model'
|
|
* 503:
|
|
* description: Service unavailable
|
|
* content:
|
|
* application/json:
|
|
* schema:
|
|
* $ref: '#/components/schemas/Error'
|
|
*/
|
|
router.get('/', async (_req: Request, res: Response) => {
|
|
try {
|
|
logger.info('Models list request received')
|
|
|
|
const models = await chatCompletionService.getModels()
|
|
|
|
if (models.length === 0) {
|
|
logger.warn('No models available from providers')
|
|
}
|
|
|
|
logger.info(`Returning ${models.length} models`)
|
|
return res.json({
|
|
object: 'list',
|
|
data: models
|
|
})
|
|
} catch (error: any) {
|
|
logger.error('Error fetching models:', error)
|
|
return res.status(503).json({
|
|
error: {
|
|
message: 'Failed to retrieve models',
|
|
type: 'service_unavailable',
|
|
code: 'models_unavailable'
|
|
}
|
|
})
|
|
}
|
|
})
|
|
|
|
export { router as modelsRoutes }
|