import { z } from '@botpress/sdk'

export const languageModelId = z
  .enum([
    'accounts/fireworks/models/deepseek-r1-0528',
    'accounts/fireworks/models/deepseek-v3-0324',
    'accounts/fireworks/models/llama4-maverick-instruct-basic',
    'accounts/fireworks/models/llama4-scout-instruct-basic',
    'accounts/fireworks/models/llama-v3p3-70b-instruct',
    'accounts/fireworks/models/llama-v3p1-70b-instruct',
    'accounts/fireworks/models/llama-v3p1-8b-instruct',
    'accounts/fireworks/models/mixtral-8x22b-instruct',
    'accounts/fireworks/models/mythomax-l2-13b',
    'accounts/fireworks/models/gemma2-9b-it',
    'accounts/fireworks/models/gpt-oss-20b',
    'accounts/fireworks/models/gpt-oss-120b',
  ])
  .describe('Model to use for content generation')
  .placeholder('accounts/fireworks/models/llama4-maverick-instruct-basic')
export type LanguageModelId = z.infer<typeof languageModelId>

export const imageModelId = z.enum([
  'accounts/fireworks/models/stable-diffusion-xl-1024-v1-0',
  'accounts/stability/models/sd3',
  'accounts/stability/models/sd3-medium',
  'accounts/fireworks/models/playground-v2-5-1024px-aesthetic',
  'accounts/fireworks/models/playground-v2-5-1024px-aesthetic',
])
export type ImageModelId = z.infer<typeof imageModelId>

export const speechToTextModelId = z.enum(['whisper-v3'])
export type SpeechToTextModelId = z.infer<typeof speechToTextModelId>
