File size: 827 Bytes
567f740
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
const Joi = require('joi');

const messageSchema = Joi.object({
    role: Joi.string().required(),
    content: Joi.string().required()
});

const chatRequestSchema = Joi.object({
    model: Joi.string().required(),
    messages: Joi.array().items(messageSchema).required(),
    temperature: Joi.number().min(0).max(2).default(0.7),
    max_tokens: Joi.number().integer().min(1).default(4096),
    top_p: Joi.number().min(0).max(1).default(1.0),
    frequency_penalty: Joi.number().min(-2).max(2).default(0.0),
    presence_penalty: Joi.number().min(-2).max(2).default(0.0),
    stop: Joi.array().items(Joi.string()).default([]),
    stream: Joi.boolean().default(false)
});

function validateChatRequest(data) {
    return chatRequestSchema.validate(data);
}

module.exports = { validateChatRequest };