@output.ai/llm 0.3.0-dev.pr341-1d6ac82 → 0.3.0-dev.pr341-fd8527e

package/package.json

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

package/src/ai_model.js

@@ -2,10 +2,16 @@ import { anthropic } from '@ai-sdk/anthropic';
 import { azure } from '@ai-sdk/azure';
 import { vertex } from '@ai-sdk/google-vertex';
 import { openai } from '@ai-sdk/openai';
+import { ValidationError, z } from '@output.ai/core';
 
 const builtInProviders = { azure, anthropic, openai, vertex };
 const providers = { ...builtInProviders };
 
+const registerProviderSchema = z.object( {
+  name: z.string().min( 1, 'Provider name must be a non-empty string' ),
+  providerFn: z.function()
+} );
+
 /** @internal Resets providers to built-in state. For testing only. */
 export function _resetProviders() {
   for ( const key of Object.keys( providers ) ) {
@@ -15,11 +21,12 @@ export function _resetProviders() {
 }
 
 export function registerProvider( name, providerFn ) {
-  if ( typeof name !== 'string' || name.length === 0 ) {
-    throw new Error( 'Provider name must be a non-empty string' );
-  }
-  if ( typeof providerFn !== 'function' ) {
-    throw new Error( `Provider "${name}" must be a function that creates a model` );
+  const result = registerProviderSchema.safeParse( { name, providerFn } );
+  if ( !result.success ) {
+    throw new ValidationError(
+      `Invalid provider registration: ${z.prettifyError( result.error )}`,
+      { cause: result.error }
+    );
   }
   providers[name] = providerFn;
 }

package/src/ai_model.spec.js

@@ -630,15 +630,15 @@ describe( 'registerProvider', () => {
   } );
 
   it( 'throws when name is not a string', () => {
-    expect( () => registerProvider( 123, vi.fn() ) ).toThrow( 'non-empty string' );
+    expect( () => registerProvider( 123, vi.fn() ) ).toThrow( 'expected string, received number' );
   } );
 
   it( 'throws when providerFn is not a function', () => {
-    expect( () => registerProvider( 'bad', 'not-a-function' ) ).toThrow( 'must be a function' );
+    expect( () => registerProvider( 'bad', 'not-a-function' ) ).toThrow( 'expected function, received string' );
   } );
 
   it( 'throws when providerFn is null', () => {
-    expect( () => registerProvider( 'bad', null ) ).toThrow( 'must be a function' );
+    expect( () => registerProvider( 'bad', null ) ).toThrow( 'expected function, received null' );
   } );
 } );
 

package/src/prompt_validations.js

@@ -34,12 +34,12 @@ const SNAKE_CASE_WARNINGS = {
 
 function warnSnakeCaseFields( config ) {
   for ( const [ snake, camel ] of Object.entries( SNAKE_CASE_WARNINGS ) ) {
-    if ( snake in config ) {
+    if ( Object.hasOwn( config, snake ) ) {
       console.warn( `[output-llm] "${snake}" found in prompt config. Did you mean "${camel}"?` );
     }
   }
   const thinking = config.providerOptions?.thinking;
-  if ( thinking && 'budget_tokens' in thinking ) {
+  if ( thinking && Object.hasOwn( thinking, 'budget_tokens' ) ) {
     console.warn( '[output-llm] "budget_tokens" found in providerOptions.thinking. Did you mean "budgetTokens"?' );
   }
 }