@output.ai/llm 0.0.10 → 0.0.12

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@output.ai/llm",
-  "version": "0.0.10",
+  "version": "0.0.12",
   "description": "Framework abstraction to interact with LLM models",
   "type": "module",
   "main": "src/index.js",
@@ -8,14 +8,12 @@
   "files": [
     "./src"
   ],
-  "repository": {
-    "type": "git",
-    "url": "git+https://github.com/growthxai/flow-sdk"
-  },
   "dependencies": {
     "@ai-sdk/anthropic": "2.0.4",
+    "@ai-sdk/azure": "2.0.49",
     "@ai-sdk/openai": "2.0.18",
     "@output.ai/trace": "0.0.1",
     "ai": "5.0.48"
-  }
+  },
+  "license": "UNLICENSED"
 }

package/src/ai_sdk.js

@@ -4,10 +4,12 @@ import {
 } from 'ai';
 import { anthropic } from '@ai-sdk/anthropic';
 import { openai } from '@ai-sdk/openai';
+import { azure } from '@ai-sdk/azure';
 import { trace } from '@output.ai/trace';
 
 const providers = {
   anthropic,
+  azure,
   openai
 };
 
@@ -17,13 +19,18 @@ export async function generateText( options ) {
   const provider = providers[prompt.config.provider];
   const model = provider( prompt.config.model );
 
-  const result = await aiGenerateText( {
+  const aiOptions = {
     model,
     messages: prompt.messages,
     temperature: prompt.config.temperature,
-    maxOutputTokens: prompt.config.max_tokens ?? 64000,
     ...aiSdkOptions // Spread remaining AI SDK options (thinking, etc.)
-  } );
+  };
+
+  if ( prompt.config.max_tokens ) {
+    aiOptions.maxOutputTokens = prompt.config.max_tokens;
+  }
+
+  const result = await aiGenerateText( aiOptions );
 
   trace( { lib: 'llm', event: 'generateText', input: options, output: result } );
 
@@ -43,7 +50,7 @@ export async function generateObject( options ) {
   const provider = providers[prompt.config.provider];
   const model = provider( prompt.config.model );
 
-  const result = await aiGenerateObject( {
+  const aiOptions = {
     model,
     schema,
     schemaName,
@@ -51,9 +58,14 @@ export async function generateObject( options ) {
     output,
     messages: prompt.messages,
     temperature: prompt.config.temperature,
-    maxOutputTokens: prompt.config.max_tokens ?? 64000,
     ...aiSdkOptions // Spread remaining AI SDK options
-  } );
+  };
+
+  if ( prompt.config.max_tokens ) {
+    aiOptions.maxOutputTokens = prompt.config.max_tokens;
+  }
+
+  const result = await aiGenerateObject( aiOptions );
 
   trace( { lib: 'llm', event: 'generateObject', input: options, output: result } );
 

package/src/index.d.ts

@@ -5,7 +5,7 @@ import type {
 
 export interface Prompt {
   config: {
-    provider: 'anthropic' | 'openai';
+    provider: 'anthropic' | 'openai' | 'azure' ;
     model: string;
     temperature?: number;
     max_tokens?: number;

package/src/index.js

@@ -1,2 +1 @@
 export { generateText, generateObject } from './ai_sdk.js';
-export { LLMSchema } from './schema.js';

package/src/schema.js

@@ -1,8 +0,0 @@
-export class LLMSchema {
-  constructor( { name, description, schema, output = 'object' } ) {
-    this.name = name;
-    this.schema = schema;
-    this.description = description;
-    this.output = output;
-  }
-}