@output.ai/llm 0.2.4 → 0.2.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@output.ai/llm",
-  "version": "0.2.4",
+  "version": "0.2.5",
   "description": "Framework abstraction to interact with LLM models",
   "type": "module",
   "main": "src/index.js",
@@ -13,7 +13,7 @@
     "@ai-sdk/azure": "2.0.53",
     "@ai-sdk/openai": "2.0.52",
     "@output.ai/core": ">=0.0.1",
-    "ai": "5.0.48",
+    "ai": "5.0.52",
     "gray-matter": "4.0.3",
     "liquidjs": "10.22.0"
   },

package/src/index.d.ts

@@ -59,12 +59,12 @@ export type Prompt = {
  * Load a prompt file and render it with variables.
  *
  * @param {string} name - Name of the prompt file (without .prompt extension)
- * @param {Record<string, string | number>} [variables] - Variables to interpolate
+ * @param {Record<string, string | number | boolean>} [variables] - Variables to interpolate
  * @returns {Prompt} Loaded and rendered prompt object
  */
 export function loadPrompt(
   name: string,
-  variables?: Record<string, string | number>
+  variables?: Record<string, string | number | boolean>
 ): Prompt;
 
 /**
@@ -75,13 +75,13 @@ export function loadPrompt(
  *
  * @param {object} args - Generation arguments
  * @param {string} args.prompt - Prompt file name
- * @param {Record<string, string | number>} args.variables - Variables to interpolate
+ * @param {Record<string, string | number | boolean>} args.variables - Variables to interpolate
  * @returns {Promise<string>} Generated text
  */
 export function generateText(
   args: {
     prompt: string,
-    variables?: Record<string, string | number>
+    variables?: Record<string, string | number | boolean>
   }
 ): Promise<string>;
 
@@ -93,7 +93,7 @@ export function generateText(
  *
  * @param {object} args - Generation arguments
  * @param {string} args.prompt - Prompt file name
- * @param {Record<string, string | number>} args.variables - Variables to interpolate
+ * @param {Record<string, string | number | boolean>} args.variables - Variables to interpolate
  * @param {z.ZodObject} args.schema - Output schema
  * @param {string} [args.schemaName] - Output schema name
  * @param {string} [args.schemaDescription] - Output schema description
@@ -102,7 +102,7 @@ export function generateText(
 export function generateObject<TSchema extends z.ZodObject>(
   args: {
     prompt: string,
-    variables?: Record<string, string | number>,
+    variables?: Record<string, string | number | boolean>,
     schema: TSchema,
     schemaName?: string,
     schemaDescription?: string
@@ -117,7 +117,7 @@ export function generateObject<TSchema extends z.ZodObject>(
  *
  * @param {object} args - Generation arguments
  * @param {string} args.prompt - Prompt file name
- * @param {Record<string, string | number>} args.variables - Variables to interpolate
+ * @param {Record<string, string | number | boolean>} args.variables - Variables to interpolate
  * @param {z.ZodType} args.schema - Output schema (array item)
  * @param {string} [args.schemaName] - Output schema name
  * @param {string} [args.schemaDescription] - Output schema description
@@ -126,7 +126,7 @@ export function generateObject<TSchema extends z.ZodObject>(
 export function generateArray<TSchema extends z.ZodType>(
   args: {
     prompt: string,
-    variables?: Record<string, string | number>,
+    variables?: Record<string, string | number | boolean>,
     schema: TSchema,
     schemaName?: string,
     schemaDescription?: string
@@ -141,14 +141,14 @@ export function generateArray<TSchema extends z.ZodType>(
  *
  * @param {object} args - Generation arguments
  * @param {string} args.prompt - Prompt file name
- * @param {Record<string, string | number>} args.variables - Variables to interpolate
+ * @param {Record<string, string | number | boolean>} args.variables - Variables to interpolate
  * @param {string[]} args.enum - Allowed values for the generation
  * @returns {Promise<string>} One of the provided enum values
  */
 export function generateEnum<const TEnum extends readonly [string, ...string[]]>(
   args: {
     prompt: string,
-    variables?: Record<string, string | number>,
+    variables?: Record<string, string | number | boolean>,
     enum: TEnum
   }
 ): Promise<TEnum[number]>;

package/src/prompt_loader.js

@@ -18,7 +18,7 @@ const renderPrompt = ( name, content, values ) => {
  * Load a prompt file and render it with variables.
  *
  * @param {string} name - Name of the prompt file (without .prompt extension)
- * @param {Record<string, string | number>} [values] - Variables to interpolate
+ * @param {Record<string, string | number | boolean>} [values] - Variables to interpolate
  * @returns {Prompt} Loaded and rendered prompt object
  */
 export const loadPrompt = ( name, values = {} ) => {

package/src/prompt_loader.spec.js

@@ -146,4 +146,34 @@ temperature: 0.7
     expect( result.config.maxTokens ).toBe( 1024 );
   } );
 
+  it( 'should render boolean variables correctly', () => {
+    const promptContent = `---
+provider: anthropic
+model: claude-3-5-sonnet-20241022
+---
+
+<user>{% if debug %}Debug mode enabled{% else %}Debug mode disabled{% endif %}</user>`;
+
+    loadContent.mockReturnValue( promptContent );
+
+    const result = loadPrompt( 'test', { debug: true } );
+
+    expect( result.messages[0].content ).toBe( 'Debug mode enabled' );
+  } );
+
+  it( 'should render false boolean variables', () => {
+    const promptContent = `---
+provider: anthropic
+model: claude-3-5-sonnet-20241022
+---
+
+<user>{% if enabled %}Feature enabled{% else %}Feature disabled{% endif %}</user>`;
+
+    loadContent.mockReturnValue( promptContent );
+
+    const result = loadPrompt( 'test', { enabled: false } );
+
+    expect( result.messages[0].content ).toBe( 'Feature disabled' );
+  } );
+
 } );