@outputai/llm 0.11.1-next.da26845.0 → 0.12.0

package/src/prompt/loader.spec.js

@@ -1,258 +0,0 @@
-import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
-
-const liquidMocks = vi.hoisted( () => ( {
-  createContext: vi.fn(),
-  options: { strictVariables: true },
-  parseAndRenderSync: vi.fn(),
-  registerFilter: vi.fn()
-} ) );
-
-vi.mock( 'liquidjs', () => ( {
-  Context: class Context {
-    constructor( ...args ) {
-      liquidMocks.createContext( this, ...args );
-    }
-  },
-  Liquid: class Liquid {
-    options = liquidMocks.options;
-    parseAndRenderSync = liquidMocks.parseAndRenderSync;
-    registerFilter = liquidMocks.registerFilter;
-  }
-} ) );
-
-vi.mock( 'gray-matter', () => ( {
-  default: vi.fn()
-} ) );
-
-vi.mock( '@outputai/core/sdk/helpers', () => ( {
-  Path: {
-    resolveInvocationDir: vi.fn( () => '/invocation' )
-  }
-} ) );
-
-vi.mock( './interpolations.js', () => ( {
-  pipeInterpolations: vi.fn( value => value ),
-  interpolationFilterToken: '__var_safe',
-  encode: vi.fn(),
-  decode: vi.fn( value => value )
-} ) );
-
-vi.mock( '../utils/file.js', () => ( {
-  searchAndReadFile: vi.fn()
-} ) );
-
-vi.mock( './content.js', () => ( {
-  parseContent: vi.fn()
-} ) );
-
-vi.mock( './validations.js', () => ( {
-  parsePromptSchema: vi.fn( prompt => prompt )
-} ) );
-
-import { loadPrompt } from './loader.js';
-import { Path } from '@outputai/core/sdk/helpers';
-import matter from 'gray-matter';
-import { decode, pipeInterpolations } from './interpolations.js';
-import { searchAndReadFile } from '../utils/file.js';
-import { parseContent } from './content.js';
-import { parsePromptSchema } from './validations.js';
-
-const defaultConfig = {
-  provider: 'anthropic',
-  model: 'claude-3-5-sonnet-20241022'
-};
-
-const defaultParsed = {
-  messages: [ { role: 'user', content: 'Hi' } ],
-  instructions: null
-};
-
-const stubMatter = ( config = defaultConfig ) => {
-  matter.mockImplementation( ( _input, options ) => {
-    if ( options?.engines ) {
-      return { matter: 'RAW_YAML', content: 'RAW_BODY' };
-    }
-    return { data: { ...config } };
-  } );
-};
-
-describe( 'loadPrompt', () => {
-  beforeEach( () => {
-    vi.clearAllMocks();
-    Path.resolveInvocationDir.mockReturnValue( '/invocation' );
-    searchAndReadFile.mockReturnValue( { content: 'FILE', dir: '/mock/dir' } );
-    pipeInterpolations.mockImplementation( value => value );
-    decode.mockImplementation( value => value );
-    liquidMocks.parseAndRenderSync.mockImplementation( template => template );
-    parseContent.mockReturnValue( { ...defaultParsed, messages: [ ...defaultParsed.messages ] } );
-    parsePromptSchema.mockImplementation( prompt => prompt );
-    stubMatter();
-  } );
-
-  afterEach( () => {
-    vi.restoreAllMocks();
-  } );
-
-  it( 'splits, renders, parses, and validates the prompt', () => {
-    const variables = { name: 'World' };
-
-    const result = loadPrompt( 'test', variables );
-
-    expect( searchAndReadFile ).toHaveBeenCalledWith( '/invocation', 'test.prompt' );
-    expect( matter ).toHaveBeenNthCalledWith( 1, 'FILE', expect.objectContaining( {
-      engines: { yaml: expect.any( Function ) }
-    } ) );
-    expect( matter ).toHaveBeenNthCalledWith( 2, '---\nRAW_YAML\n---\n' );
-    expect( decode ).toHaveBeenCalledWith( defaultConfig );
-    expect( parseContent ).toHaveBeenCalledWith( 'RAW_BODY', undefined );
-    expect( parsePromptSchema ).toHaveBeenCalledWith( {
-      name: 'test',
-      config: defaultConfig,
-      messages: defaultParsed.messages,
-      instructions: null,
-      fileDir: '/mock/dir',
-      variables
-    } );
-    expect( result ).toEqual( parsePromptSchema.mock.calls[0][0] );
-  } );
-
-  it( 'renders frontmatter before content using one Liquid context', () => {
-    const variables = { name: 'World' };
-
-    loadPrompt( 'test', variables );
-
-    const context = liquidMocks.createContext.mock.calls[0][0];
-    expect( liquidMocks.createContext ).toHaveBeenCalledWith( context, variables, liquidMocks.options, { sync: true } );
-    expect( pipeInterpolations ).toHaveBeenNthCalledWith( 1, 'RAW_YAML' );
-    expect( pipeInterpolations ).toHaveBeenNthCalledWith( 2, 'RAW_BODY' );
-    expect( liquidMocks.parseAndRenderSync ).toHaveBeenNthCalledWith( 1, 'RAW_YAML', context );
-    expect( liquidMocks.parseAndRenderSync ).toHaveBeenNthCalledWith( 2, 'RAW_BODY', context );
-  } );
-
-  it( 'uses decoded frontmatter for content parsing and validation', () => {
-    const encodedConfig = {
-      ...defaultConfig,
-      messageOptions: { cached: { openai: { label: 'R&D' } } }
-    };
-    const decodedConfig = {
-      ...defaultConfig,
-      messageOptions: { cached: { openai: { label: 'R&D' } } }
-    };
-    stubMatter( encodedConfig );
-    decode.mockReturnValue( decodedConfig );
-
-    loadPrompt( 'test' );
-
-    expect( decode ).toHaveBeenCalledWith( encodedConfig );
-    expect( parseContent ).toHaveBeenCalledWith( 'RAW_BODY', decodedConfig.messageOptions );
-    expect( parsePromptSchema ).toHaveBeenCalledWith( expect.objectContaining( {
-      config: decodedConfig
-    } ) );
-  } );
-
-  it( 'passes the provided prompt directory to searchAndReadFile', () => {
-    loadPrompt( 'test', {}, '/custom/prompts' );
-
-    expect( searchAndReadFile ).toHaveBeenCalledWith( '/custom/prompts', 'test.prompt' );
-    expect( Path.resolveInvocationDir ).not.toHaveBeenCalled();
-  } );
-
-  it( 'renders the piped template, not the raw body', () => {
-    pipeInterpolations.mockReturnValue( 'PIPED' );
-
-    loadPrompt( 'test', { name: 'World' } );
-
-    expect( liquidMocks.parseAndRenderSync ).toHaveBeenNthCalledWith( 2, 'PIPED', expect.any( Object ) );
-    expect( parseContent ).toHaveBeenCalledWith( 'PIPED', undefined );
-  } );
-
-  it( 'passes config.messageOptions to parseContent', () => {
-    const messageOptions = { cached: { anthropic: { cacheControl: { type: 'ephemeral' } } } };
-    stubMatter( { ...defaultConfig, messageOptions } );
-
-    loadPrompt( 'test' );
-
-    expect( parseContent ).toHaveBeenCalledWith( 'RAW_BODY', messageOptions );
-  } );
-
-  it( 'uses parseContent for messages and instructions', () => {
-    parseContent.mockReturnValue( {
-      messages: [],
-      instructions: 'Generate a poster.'
-    } );
-
-    const result = loadPrompt( 'image_prompt' );
-
-    expect( result.messages ).toEqual( [] );
-    expect( result.instructions ).toBe( 'Generate a poster.' );
-  } );
-
-  it( 'throws when the prompt file is not found', () => {
-    searchAndReadFile.mockReturnValue( null );
-
-    expect( () => loadPrompt( 'nonexistent' ) ).toThrow( /Prompt file "nonexistent" not found/ );
-  } );
-
-  it( 'throws when rendered content is empty', () => {
-    liquidMocks.parseAndRenderSync
-      .mockReturnValueOnce( 'RAW_YAML' )
-      .mockReturnValueOnce( '  ' );
-
-    expect( () => loadPrompt( 'empty_prompt' ) ).toThrow( /Prompt "empty_prompt" has no content/ );
-  } );
-
-  it( 'wraps split errors with the prompt name', () => {
-    matter.mockImplementation( () => {
-      throw new Error( 'bad delimiters' );
-    } );
-
-    expect( () => loadPrompt( 'writer@v1' ) ).toThrow(
-      /Error parsing frontmatter on prompt "writer@v1": bad delimiters/
-    );
-  } );
-
-  it( 'wraps content render errors with the prompt name', () => {
-    liquidMocks.parseAndRenderSync
-      .mockReturnValueOnce( 'RAW_YAML' )
-      .mockImplementationOnce( () => {
-        throw new Error( 'unknown variable' );
-      } );
-
-    expect( () => loadPrompt( 'writer@v1' ) ).toThrow(
-      /Error rendering content on prompt "writer@v1": unknown variable/
-    );
-  } );
-
-  it( 'wraps frontmatter render errors with the prompt name', () => {
-    liquidMocks.parseAndRenderSync.mockImplementationOnce( () => {
-      throw new Error( 'unknown filter' );
-    } );
-
-    expect( () => loadPrompt( 'writer@v1' ) ).toThrow(
-      /Error rendering frontmatter on prompt "writer@v1": unknown filter/
-    );
-  } );
-
-  it( 'wraps frontmatter yaml parse errors with the prompt name', () => {
-    matter.mockImplementation( ( _input, options ) => {
-      if ( options?.engines ) {
-        return { matter: 'RAW_YAML', content: 'RAW_BODY' };
-      }
-      throw new Error( 'invalid yaml' );
-    } );
-
-    expect( () => loadPrompt( 'writer@v1' ) ).toThrow(
-      /Error converting frontmatter yaml to js on prompt "writer@v1": invalid yaml/
-    );
-  } );
-
-  it( 'wraps content parse errors with the prompt name', () => {
-    parseContent.mockImplementation( () => {
-      throw new Error( 'unknown option' );
-    } );
-
-    expect( () => loadPrompt( 'writer@v1' ) ).toThrow(
-      /Error parsing content on prompt "writer@v1": unknown option/
-    );
-  } );
-} );

package/src/prompt/markup/attributes.spec.js

@@ -1,132 +0,0 @@
-import { describe, expect, it } from 'vitest';
-import { extractAttributes } from './attributes.js';
-
-describe( 'extractAttributes', () => {
-  it( 'returns no attributes for empty input', () => {
-    expect( extractAttributes( '' ) ).toEqual( {} );
-  } );
-
-  it( 'returns no attributes for whitespace-only input', () => {
-    expect( extractAttributes( ' \n\t ' ) ).toEqual( {} );
-  } );
-
-  it( 'parses bare attributes as true', () => {
-    expect( extractAttributes( 'cached pinned' ) ).toEqual( {
-      cached: true,
-      pinned: true
-    } );
-  } );
-
-  it( 'lowercases attribute names', () => {
-    expect( extractAttributes( 'OPTIONS=cached PINNED' ) ).toEqual( {
-      options: 'cached',
-      pinned: true
-    } );
-  } );
-
-  it( 'preserves unquoted values as strings', () => {
-    expect( extractAttributes( 'enabled=false count=001 ratio=1.25' ) ).toEqual( {
-      enabled: 'false',
-      count: '001',
-      ratio: '1.25'
-    } );
-  } );
-
-  it( 'removes matching double quotes', () => {
-    expect( extractAttributes( 'options="cached shared"' ) ).toEqual( {
-      options: 'cached shared'
-    } );
-  } );
-
-  it( 'removes matching single quotes', () => {
-    expect( extractAttributes( 'options=\'cached shared\'' ) ).toEqual( {
-      options: 'cached shared'
-    } );
-  } );
-
-  it( 'allows the opposite quote type inside quoted values', () => {
-    expect( extractAttributes( 'single="it\'s ready" double=\'say "ready"\'' ) ).toEqual( {
-      single: 'it\'s ready',
-      double: 'say "ready"'
-    } );
-  } );
-
-  it( 'preserves equals and greater-than characters inside quoted values', () => {
-    expect( extractAttributes( 'condition="score >= 10" expression=\'a = b\'' ) ).toEqual( {
-      condition: 'score >= 10',
-      expression: 'a = b'
-    } );
-  } );
-
-  it( 'supports whitespace around equals signs', () => {
-    expect( extractAttributes( 'options = "cached shared" mode = fast pinned' ) ).toEqual( {
-      options: 'cached shared',
-      mode: 'fast',
-      pinned: true
-    } );
-  } );
-
-  it( 'supports tabs and newlines between attributes', () => {
-    expect( extractAttributes( 'options="cached"\nmode=fast\tpinned' ) ).toEqual( {
-      options: 'cached',
-      mode: 'fast',
-      pinned: true
-    } );
-  } );
-
-  it( 'ignores surrounding whitespace', () => {
-    expect( extractAttributes( '  options = "cached" pinned  ' ) ).toEqual( {
-      options: 'cached',
-      pinned: true
-    } );
-  } );
-
-  it( 'parses an explicitly empty value', () => {
-    expect( extractAttributes( 'options=' ) ).toEqual( { options: '' } );
-  } );
-
-  it( 'uses the last value when an attribute is repeated', () => {
-    expect( extractAttributes( 'mode=slow mode=fast' ) ).toEqual( { mode: 'fast' } );
-  } );
-
-  it( 'rejects an unterminated single-quoted value', () => {
-    expect( () => extractAttributes( 'color=\'red' ) )
-      .toThrow( /^Invalid attribute "color": quoted value is missing its closing quote \('\)\.$/ );
-  } );
-
-  it( 'rejects an unterminated double-quoted value', () => {
-    expect( () => extractAttributes( 'color="red' ) )
-      .toThrow( /^Invalid attribute "color": quoted value is missing its closing quote \("\)\.$/ );
-  } );
-
-  it( 'rejects values closed by a different quote type', () => {
-    expect( () => extractAttributes( 'color=\'red"' ) ).toThrow();
-    expect( () => extractAttributes( 'color="red\'' ) ).toThrow();
-  } );
-
-  it( 'rejects stray quotes in unquoted values', () => {
-    expect( () => extractAttributes( 'color=red\'' ) ).toThrow();
-    expect( () => extractAttributes( 'color=red"' ) ).toThrow();
-    expect( () => extractAttributes( 'color=arby\'s' ) ).toThrow();
-    expect( () => extractAttributes( 'color=jack"madman"silva' ) ).toThrow();
-  } );
-
-  it( 'rejects matching quote characters inside quoted values', () => {
-    expect( () => extractAttributes( 'message="say "hello""' ) )
-      .toThrow( /^Invalid attribute "message": quoted value contains an unexpected quote \("\)\.$/ );
-    expect( () => extractAttributes( 'message=\'say \'hello\'\'' ) ).toThrow();
-    expect( () => extractAttributes( 'color="red"dark"' ) ).toThrow();
-    expect( () => extractAttributes( 'color=\'red\'dark\'' ) ).toThrow();
-  } );
-
-  it( 'rejects an empty attribute name', () => {
-    expect( () => extractAttributes( '=value' ) )
-      .toThrow( /^Invalid attribute "=value": name is missing\.$/ );
-  } );
-
-  it( 'rejects invalid valued and bare attribute names', () => {
-    expect( () => extractAttributes( 'bad/name=value' ) )
-      .toThrow( /^Invalid attribute name "bad\/name"\.$/ );
-    expect( () => extractAttributes( 'bad/name' ) ).toThrow();
-  } );
-} );