prompt-optimizer/packages/ui/tests/composables/variable/useSmartVariableValueGeneration.test.ts

import { describe, expect, it } from 'vitest'

import { buildVariableValueGenerationPlan } from '../../../src/composables/variable/useSmartVariableValueGeneration'

describe('buildVariableValueGenerationPlan', () => {
  it('uses filled variables as context while generating only missing variables in batch mode', () => {
    const values: Record<string, string> = {
      topic: 'AI',
      style: '',
      length: '   ',
      audience: 'high school students',
    }

    const plan = buildVariableValueGenerationPlan(
      ['topic', 'style', 'length', 'audience'],
      (name) => values[name] ?? '',
      () => 'test'
    )

    expect(plan.variablesToGenerate).toEqual([
      { name: 'style', source: 'test' },
      { name: 'length', source: 'test' },
    ])
    expect(plan.contextVariables).toEqual([
      { name: 'topic', source: 'test', currentValue: 'AI' },
      { name: 'audience', source: 'test', currentValue: 'high school students' },
    ])
  })

  it('keeps single-variable generation scoped to the target variable', () => {
    const values: Record<string, string> = {
      topic: 'AI',
      style: 'explainer',
    }

    const plan = buildVariableValueGenerationPlan(
      ['topic', 'style'],
      (name) => values[name] ?? '',
      () => 'test',
      'style'
    )

    expect(plan.variablesToGenerate).toEqual([
      { name: 'style', source: 'test', currentValue: 'explainer' },
    ])
    expect(plan.contextVariables).toEqual([])
  })

  it('attaches variable metadata when available', () => {
    const values: Record<string, string> = {
      tone: '',
      audience: 'developers',
    }

    const plan = buildVariableValueGenerationPlan(
      ['tone', 'audience'],
      (name) => values[name] ?? '',
      () => 'test',
      '',
      (name) => name === 'tone'
        ? {
            description: 'Writing tone',
            defaultValue: 'friendly',
          }
        : undefined
    )

    expect(plan.variablesToGenerate).toEqual([
      {
        name: 'tone',
        description: 'Writing tone',
        defaultValue: 'friendly',
        source: 'test',
      },
    ])
  })
})