@output.ai/llm 0.2.0 → 0.2.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@output.ai/llm",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "Framework abstraction to interact with LLM models",
   "type": "module",
   "main": "src/index.js",

package/src/index.d.ts

@@ -1,5 +1,75 @@
 import type { z } from '@output.ai/core';
 
+/**
+ * Represents a single message in a prompt conversation
+ * @example
+ * const msg: PromptMessage = {
+ *   role: 'user',
+ *   content: 'Hello, Claude!'
+ * };
+ */
+export type PromptMessage = {
+  /** The role of the message. Examples: 'system', 'user', 'assistant' */
+  role: string;
+  /** The content of the message */
+  content: string;
+};
+
+/**
+ * Configuration for LLM prompt generation
+ * @example
+ * const prompt: Prompt = {
+ *   name: 'summarizePrompt',
+ *   config: {
+ *     provider: 'anthropic',
+ *     model: 'claude-opus-4-1',
+ *     temperature: 0.7,
+ *     maxTokens: 2048
+ *   },
+ *   messages: [...]
+ * };
+ */
+export type Prompt = {
+  /** Name of the prompt file */
+  name: string;
+
+  /** General configurations for the LLM */
+  config: {
+    /** LLM Provider */
+    provider: 'anthropic' | 'openai' | 'azure';
+
+    /** Model name/identifier */
+    model: string;
+
+    /** Generation temperature (0-2). Lower = more deterministic */
+    temperature?: number;
+
+    /** Maximum tokens in the response */
+    maxTokens?: number;
+
+    /** Additional provider-specific options */
+    options?: Record<string, Record<string, JSONValue>>;
+
+    /** Provider-specific configurations */
+    providerOptions?: Record<string, unknown>;
+  };
+
+  /** Array of messages in the conversation */
+  messages: PromptMessage[];
+};
+
+/**
+ * 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
+ * @returns {Prompt} Loaded and rendered prompt object
+ */
+export function loadPrompt(
+  name: string,
+  variables?: Record<string, string | number>
+): Prompt;
+
 /**
  * Use an LLM model to generate text.
  *

package/src/index.js

@@ -1,2 +1,3 @@
 export { generateText, generateArray, generateObject, generateEnum } from './ai_sdk.js';
+export { loadPrompt } from './prompt_loader.js';
 export * as ai from 'ai';