45 lines
1.0 KiB
TypeScript
45 lines
1.0 KiB
TypeScript
import { Body, Controller, Get, Post, Query } from '@nestjs/common';
|
|
import { OllamaService } from './ollama.service';
|
|
import { ChatDto } from './dto/chat.dto';
|
|
|
|
@Controller('ollama')
|
|
export class OllamaController {
|
|
constructor(private readonly ollamaService: OllamaService) {}
|
|
|
|
@Get('models')
|
|
public async getModels(
|
|
@Query('onlyChats') { onlyChats }: { onlyChats: boolean },
|
|
) {
|
|
return await this.ollamaService.getModels({ onlyChats });
|
|
}
|
|
|
|
@Post('chat')
|
|
public async chat(@Body() body: ChatDto) {
|
|
const { message, model } = body;
|
|
|
|
if (!message) {
|
|
throw new Error('O atributo message é obrigatório');
|
|
}
|
|
|
|
if (!model) {
|
|
const [defaultModel] = await this.ollamaService.getModels({
|
|
onlyChats: true,
|
|
});
|
|
|
|
if (!defaultModel) {
|
|
throw new Error('Nenhum modelo encontrado');
|
|
}
|
|
|
|
return await this.ollamaService.generateChat({
|
|
model: defaultModel.name,
|
|
message,
|
|
});
|
|
}
|
|
|
|
return await this.ollamaService.generateChat({
|
|
model,
|
|
message,
|
|
});
|
|
}
|
|
}
|