FastGPT/packages/global/openapi/core/ai/api.ts

import { ObjectIdSchema } from '../../../common/type/mongo';
import { z } from 'zod';

// Query Params
export const GetLLMRequestRecordParamsSchema = z.object({
  requestId: z.string().meta({
    example: 'V1StGXR8_Z5jdHi6B-myT',
    description: 'LLM 请求追踪 ID'
  })
});

export type GetLLMRequestRecordParamsType = z.infer<typeof GetLLMRequestRecordParamsSchema>;

// Response
export const LLMRequestRecordSchema = z.object({
  _id: ObjectIdSchema,
  requestId: z.string().meta({
    example: 'V1StGXR8_Z5jdHi6B-myT',
    description: '请求追踪 ID'
  }),
  body: z.record(z.string(), z.any()).meta({
    description: 'LLM 请求体'
  }),
  response: z.record(z.string(), z.any()).meta({
    description: 'LLM 响应内容'
  }),
  createdAt: z.date().meta({
    example: '2024-01-01T00:00:00.000Z',
    description: '创建时间'
  })
});

export type LLMRequestRecordSchemaType = z.infer<typeof LLMRequestRecordSchema>;

/* ============================================================================
 * 共享 Schema
 * ============================================================================ */

export const ChatMessageSchema = z.object({
  role: z.enum(['user', 'assistant', 'system', 'tool', 'function']).meta({
    example: 'user',
    description: '消息角色'
  }),
  content: z
    .union([z.string(), z.array(z.object())])
    .optional()
    .meta({
      example: '你好',
      description: '消息内容'
    }),
  name: z.string().optional().meta({ description: '发送者名称' }),
  tool_calls: z.array(z.object()).optional().meta({ description: '工具调用' }),
  tool_call_id: z.string().optional().meta({ description: '工具调用 ID' })
});