@modelence/ai 0.1.0 → 0.1.1-dev.1

package/README.md

@@ -1,6 +1,6 @@
 # @modelence/ai
 
-AI engine for Modelence applications.
+AI engine for Modelence applications with built-in telemetry, based on https://ai-sdk.dev
 
 ## Installation
 
@@ -10,19 +10,35 @@ npm install @modelence/ai
 
 ## Usage
 
+### generateText
+
+A wrapper around [AI SDK](https://ai-sdk.dev)'s `generateText` with built-in Modelence configuration and telemetry.
+
 ```typescript
-import { getOpenAIConfig } from '@modelence/ai';
+import { generateText } from '@modelence/ai';
+
+const response = await generateText({
+  provider: 'openai',
+  model: 'gpt-4o',
+  messages: [
+    { role: 'user', content: 'Hello, world!' }
+  ],
+});
 
-const config = getOpenAIConfig();
-// Returns: { apiKey: string }
+console.log(response.text);
 ```
 
-## Functions
+#### Supported Providers
 
-### `getOpenAIConfig()`
+- **OpenAI**: `provider: 'openai'`, models: `gpt-4o`, `gpt-4o-mini`, `gpt-3.5-turbo`
+- **Anthropic**: `provider: 'anthropic'`, models: `claude-3-5-sonnet-20241022`, `claude-3-haiku-20240307`
+- **Google**: `provider: 'google'`, models: `gemini-1.5-pro`, `gemini-1.5-flash`
 
-Returns the OpenAI configuration from Modelence server config.
+#### Configuration
 
-**Returns:** `{ apiKey: string }`
+The function automatically uses API keys from Modelence configuration:
+- OpenAI: `_system.openai.apiKey`
+- Anthropic: `_system.anthropic.apiKey`
+- Google: `_system.google.apiKey`
 
-The API key is retrieved from the `_system.openai.apiKey` configuration key using Modelence's server-side config system.
+You don't need to manually set any of these configs as long as your application is using a [Modelence Cloud](https://modelence.com/cloud) backend - simply use the AI > Integrations tab in your Modelence Cloud dashboard to configure keys, and it will be automatically used and recognized by this package.

package/dist/index.d.ts

@@ -1,5 +1,48 @@
-declare function getOpenAIConfig(): {
-    apiKey: string;
-};
+import * as ai from 'ai';
+import { generateText as generateText$1 } from 'ai';
 
-export { getOpenAIConfig };
+/**
+ * Supported AI providers for text generation.
+ */
+type Provider = 'openai' | 'anthropic' | 'google';
+type OriginalGenerateTextParams = Parameters<typeof generateText$1>[0];
+/**
+ * Options for the Modelence generateText function.
+ *
+ * This interface extends all the standard AI SDK generateText options,
+ * but replaces the model parameter with separate provider and model parameters.
+ */
+interface GenerateTextOptions extends Omit<OriginalGenerateTextParams, 'model'> {
+    /** The AI provider name */
+    provider: Provider;
+    /** The specific model name */
+    model: string;
+}
+/**
+ * Generates text using AI models with built-in Modelence configuration and telemetry.
+ *
+ * This is a wrapper around the AI SDK's generateText function that automatically
+ * configures providers using Modelence's server-side configuration system.
+ *
+ * @param options - Configuration options for text generation
+ * @returns A promise that resolves to the generated text result
+ *
+ * @example
+ * ```typescript
+ * import { generateText } from '@modelence/ai';
+ *
+ * const response = await generateText({
+ *   provider: 'openai',
+ *   model: 'gpt-4o',
+ *   messages: [
+ *     { role: 'user', content: 'Write a haiku about programming' }
+ *   ],
+ *   temperature: 0.7
+ * });
+ *
+ * console.log(response.text);
+ * ```
+ */
+declare function generateText(options: GenerateTextOptions): Promise<ai.GenerateTextResult<ai.ToolSet, unknown>>;
+
+export { type GenerateTextOptions, generateText };

package/dist/index.js

@@ -1,2 +1,2 @@
-import {getConfig}from'modelence/server';function n(){return {apiKey:String(getConfig("_system.openai.apiKey"))}}export{n as getOpenAIConfig};//# sourceMappingURL=index.js.map
+import {getConfig}from'modelence/server';import {startTransaction,captureError}from'modelence/telemetry';import {generateText}from'ai';import {createOpenAI}from'@ai-sdk/openai';import {createAnthropic}from'@ai-sdk/anthropic';import {createGoogleGenerativeAI}from'@ai-sdk/google';function d(e,r){switch(e){case "openai":return createOpenAI({apiKey:String(getConfig("_system.openai.apiKey"))})(r);case "anthropic":return createAnthropic({apiKey:String(getConfig("_system.anthropic.apiKey"))})(r);case "google":return createGoogleGenerativeAI({apiKey:String(getConfig("_system.google.apiKey"))})(r);default:throw new Error(`Unsupported provider: ${e}`)}}async function G(e){let{provider:r,model:o,...i}=e,n=startTransaction("ai","ai:generateText",{provider:r,model:o,messageCount:Array.isArray(e.messages)?e.messages.length:0,temperature:e.temperature});try{let t=await generateText({model:d(r,o),...i});return n.end(),t}catch(t){throw captureError(t),n.end("error"),t}}export{G as generateText};//# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts"],"names":["getOpenAIConfig","getConfig"],"mappings":"yCAEO,SAASA,GAAkB,CAChC,OAAO,CACL,MAAA,CAAQ,MAAOC,CAAAA,SAAAA,CAAU,uBAAuB,CAAC,CACnD,CACF","file":"index.js","sourcesContent":["import { getConfig } from 'modelence/server';\n\nexport function getOpenAIConfig() {\n  return {\n    apiKey: String(getConfig('_system.openai.apiKey'))\n  };\n}\n"]}
+{"version":3,"sources":["../src/index.ts"],"names":["getProviderModel","provider","model","createOpenAI","getConfig","createAnthropic","createGoogleGenerativeAI","generateText","options","restOptions","transaction","startTransaction","result","originalGenerateText","error","captureError"],"mappings":"uRA4BA,SAASA,CAAiBC,CAAAA,CAAAA,CAAoBC,CAAe,CAAA,CAC3D,OAAQD,CAAU,EAChB,KAAK,QAAA,CACH,OAAOE,YAAa,CAAA,CAClB,MAAQ,CAAA,MAAA,CAAOC,UAAU,uBAAuB,CAAC,CACnD,CAAC,EAAEF,CAAK,CAAA,CAEV,KAAK,WAAA,CACH,OAAOG,eAAgB,CAAA,CACrB,MAAQ,CAAA,MAAA,CAAOD,UAAU,0BAA0B,CAAC,CACtD,CAAC,EAAEF,CAAK,CAAA,CAEV,KAAK,QAAA,CACH,OAAOI,wBAAyB,CAAA,CAC9B,MAAQ,CAAA,MAAA,CAAOF,UAAU,uBAAuB,CAAC,CACnD,CAAC,CAAA,CAAEF,CAAK,CAEV,CAAA,QACE,MAAM,IAAI,MAAM,CAAyBD,sBAAAA,EAAAA,CAAQ,CAAE,CAAA,CACvD,CACF,CA2BA,eAAsBM,CAAaC,CAAAA,CAAAA,CAA8B,CAC/D,GAAM,CAAE,QAAAP,CAAAA,CAAAA,CAAU,MAAAC,CAAO,CAAA,GAAGO,CAAY,CAAA,CAAID,EAEtCE,CAAcC,CAAAA,gBAAAA,CAAiB,IAAM,CAAA,iBAAA,CAAmB,CAC5D,QAAAV,CAAAA,CAAAA,CACA,KAAAC,CAAAA,CAAAA,CACA,aAAc,KAAM,CAAA,OAAA,CAAQM,EAAQ,QAAQ,CAAA,CAAIA,EAAQ,QAAS,CAAA,MAAA,CAAS,CAC1E,CAAA,WAAA,CAAaA,EAAQ,WACvB,CAAC,CAED,CAAA,GAAI,CACF,IAAMI,CAAAA,CAAS,MAAMC,YAAAA,CAAqB,CACxC,KAAOb,CAAAA,CAAAA,CAAiBC,CAAUC,CAAAA,CAAK,EACvC,GAAGO,CACL,CAAC,CAAA,CAED,OAAAC,CAAY,CAAA,GAAA,EACLE,CAAAA,CACT,OAASE,CAAO,CAAA,CACd,MAAAC,YAAAA,CAAaD,CAAc,CAC3BJ,CAAAA,CAAAA,CAAY,IAAI,OAAO,CAAA,CACjBI,CACR,CACF","file":"index.js","sourcesContent":["import { getConfig } from 'modelence/server';\nimport { startTransaction, captureError } from 'modelence/telemetry';\nimport { generateText as originalGenerateText } from 'ai';\nimport { createOpenAI } from '@ai-sdk/openai';\nimport { createAnthropic } from '@ai-sdk/anthropic';\nimport { createGoogleGenerativeAI } from '@ai-sdk/google';\n\n/**\n * Supported AI providers for text generation.\n */\ntype Provider = 'openai' | 'anthropic' | 'google';\n\n// Extract the original generateText parameters and override the model property\ntype OriginalGenerateTextParams = Parameters<typeof originalGenerateText>[0];\n\n/**\n * Options for the Modelence generateText function.\n * \n * This interface extends all the standard AI SDK generateText options,\n * but replaces the model parameter with separate provider and model parameters.\n */\nexport interface GenerateTextOptions extends Omit<OriginalGenerateTextParams, 'model'> {\n  /** The AI provider name */\n  provider: Provider;\n  /** The specific model name */\n  model: string;\n}\n\nfunction getProviderModel(provider: Provider, model: string) {\n  switch (provider) {\n    case 'openai':\n      return createOpenAI({\n        apiKey: String(getConfig('_system.openai.apiKey')),\n      })(model);\n    \n    case 'anthropic':\n      return createAnthropic({\n        apiKey: String(getConfig('_system.anthropic.apiKey')),\n      })(model);\n    \n    case 'google':\n      return createGoogleGenerativeAI({\n        apiKey: String(getConfig('_system.google.apiKey')),\n      })(model);\n    \n    default:\n      throw new Error(`Unsupported provider: ${provider}`);\n  }\n}\n\n/**\n * Generates text using AI models with built-in Modelence configuration and telemetry.\n * \n * This is a wrapper around the AI SDK's generateText function that automatically\n * configures providers using Modelence's server-side configuration system.\n * \n * @param options - Configuration options for text generation\n * @returns A promise that resolves to the generated text result\n * \n * @example\n * ```typescript\n * import { generateText } from '@modelence/ai';\n * \n * const response = await generateText({\n *   provider: 'openai',\n *   model: 'gpt-4o',\n *   messages: [\n *     { role: 'user', content: 'Write a haiku about programming' }\n *   ],\n *   temperature: 0.7\n * });\n * \n * console.log(response.text);\n * ```\n */\nexport async function generateText(options: GenerateTextOptions) {\n  const { provider, model, ...restOptions } = options;\n  \n  const transaction = startTransaction('ai', 'ai:generateText', {\n    provider, \n    model,\n    messageCount: Array.isArray(options.messages) ? options.messages.length : 0,\n    temperature: options.temperature\n  });\n\n  try {\n    const result = await originalGenerateText({\n      model: getProviderModel(provider, model),\n      ...restOptions,\n    });\n    \n    transaction.end();\n    return result;\n  } catch (error) {\n    captureError(error as Error);\n    transaction.end('error');\n    throw error;\n  }\n}\n"]}

package/package.json

@@ -1,7 +1,7 @@
 {
   "type": "module",
   "name": "@modelence/ai",
-  "version": "0.1.0",
+  "version": "0.1.1-dev.1",
   "description": "Modelence AI engine",
   "exports": {
     ".": {
@@ -19,7 +19,11 @@
   "author": "Modelence",
   "license": "SEE LICENSE IN LICENSE",
   "dependencies": {
-    "modelence": "^0.5.0"
+    "@ai-sdk/anthropic": "^1.2.12",
+    "@ai-sdk/google": "^1.2.19",
+    "@ai-sdk/openai": "^1.3.22",
+    "ai": "^4.3.16",
+    "modelence": "^0.5.1-dev.2"
   },
   "devDependencies": {
     "tsup": "^8.3.6",

package/src/index.ts

@@ -1,7 +1,99 @@
 import { getConfig } from 'modelence/server';
+import { startTransaction, captureError } from 'modelence/telemetry';
+import { generateText as originalGenerateText } from 'ai';
+import { createOpenAI } from '@ai-sdk/openai';
+import { createAnthropic } from '@ai-sdk/anthropic';
+import { createGoogleGenerativeAI } from '@ai-sdk/google';
 
-export function getOpenAIConfig() {
-  return {
-    apiKey: String(getConfig('_system.openai.apiKey'))
-  };
+/**
+ * Supported AI providers for text generation.
+ */
+type Provider = 'openai' | 'anthropic' | 'google';
+
+// Extract the original generateText parameters and override the model property
+type OriginalGenerateTextParams = Parameters<typeof originalGenerateText>[0];
+
+/**
+ * Options for the Modelence generateText function.
+ * 
+ * This interface extends all the standard AI SDK generateText options,
+ * but replaces the model parameter with separate provider and model parameters.
+ */
+export interface GenerateTextOptions extends Omit<OriginalGenerateTextParams, 'model'> {
+  /** The AI provider name */
+  provider: Provider;
+  /** The specific model name */
+  model: string;
+}
+
+function getProviderModel(provider: Provider, model: string) {
+  switch (provider) {
+    case 'openai':
+      return createOpenAI({
+        apiKey: String(getConfig('_system.openai.apiKey')),
+      })(model);
+    
+    case 'anthropic':
+      return createAnthropic({
+        apiKey: String(getConfig('_system.anthropic.apiKey')),
+      })(model);
+    
+    case 'google':
+      return createGoogleGenerativeAI({
+        apiKey: String(getConfig('_system.google.apiKey')),
+      })(model);
+    
+    default:
+      throw new Error(`Unsupported provider: ${provider}`);
+  }
+}
+
+/**
+ * Generates text using AI models with built-in Modelence configuration and telemetry.
+ * 
+ * This is a wrapper around the AI SDK's generateText function that automatically
+ * configures providers using Modelence's server-side configuration system.
+ * 
+ * @param options - Configuration options for text generation
+ * @returns A promise that resolves to the generated text result
+ * 
+ * @example
+ * ```typescript
+ * import { generateText } from '@modelence/ai';
+ * 
+ * const response = await generateText({
+ *   provider: 'openai',
+ *   model: 'gpt-4o',
+ *   messages: [
+ *     { role: 'user', content: 'Write a haiku about programming' }
+ *   ],
+ *   temperature: 0.7
+ * });
+ * 
+ * console.log(response.text);
+ * ```
+ */
+export async function generateText(options: GenerateTextOptions) {
+  const { provider, model, ...restOptions } = options;
+  
+  const transaction = startTransaction('ai', 'ai:generateText', {
+    provider, 
+    model,
+    messageCount: Array.isArray(options.messages) ? options.messages.length : 0,
+    temperature: options.temperature
+  });
+
+  try {
+    const result = await originalGenerateText({
+      model: getProviderModel(provider, model),
+      ...restOptions,
+    });
+    
+    transaction.end();
+    return result;
+  } catch (error) {
+    captureError(error as Error);
+    transaction.end('error');
+    throw error;
+  }
 }