import { MemoryRole } from '../../types'
import type { QuestionClassifierNodeType } from './types'

export const mockData: QuestionClassifierNodeType = {
  title: 'Test',
  desc: 'Test',
  type: 'Test',
  query_variable_selector: ['aaa', 'name'],
  model: {
    provider: 'openai',
    name: 'gpt-4',
    mode: 'chat',
    completion_params: {
      temperature: 0.7,
    },
  },
  topics: [
    {
      id: '1',
      name: 'topic 1',
      topic: 'xxxxx',
    },
    {
      id: '2',
      name: 'topic 2',
      topic: 'xxxxx2',
    },
  ],
  instruction: 'You are an entity extraction model that accepts an input',
  memory: {
    role_prefix: MemoryRole.assistant,
    window: {
      enabled: false,
      size: 0,
    },
  },
}