sql-agent.service.ts 341 B

123456789101112
  1. import { Injectable } from '@nestjs/common'
  2. import { chatReason } from '@/module/service/lls.service'
  3. import { buildSqlPrompt } from '@/prompt/sql-prompt/sql'
  4. @Injectable()
  5. export class SqlAgentService {
  6. async chat(chat: string) {
  7. const chatPrompt = buildSqlPrompt(chat)
  8. return await chatReason(JSON.stringify(chatPrompt))
  9. }
  10. }