import { describe, expect, it } from 'vitest'
import { MistralTextAdapter } from './adapters/text'

// Constructing the SDK client makes no network call.
const config = { apiKey: 'test-key' }

describe('Mistral text adapter inputModalities', () => {
  it('gives the input list of each known model', () => {
    expect(
      new MistralTextAdapter(config, 'mistral-large-latest').inputModalities,
    ).toEqual(['text'])
    expect(
      new MistralTextAdapter(config, 'mistral-medium-latest').inputModalities,
    ).toEqual(['text', 'image', 'document'])
  })

  it('is undefined for a model the metadata does not list', () => {
    // A JS caller, or a model id newer than this package, reaches the adapter.
    // @ts-expect-error - 'mistral-unknown-9000' is not a declared model
    const adapter = new MistralTextAdapter(config, 'mistral-unknown-9000')

    expect(adapter.inputModalities).toBeUndefined()
  })
})