ai-functions 0.2.18 → 0.3.0

package/package.json

@@ -1,56 +1,72 @@
 {
   "name": "ai-functions",
-  "version": "0.2.18",
-  "description": "Library for Developing and Managing AI Functions (including OpenAI GPT4 / GPT3.5)",
-  "main": "dist/cjs/index.js",
-  "module": "dist/mjs/index.js",
-  "exports": {
-    ".": {
-      "import": "./dist/mjs/index.js",
-      "require": "./dist/cjs/index.js"
-    }
+  "version": "0.3.0",
+  "description": "A powerful TypeScript library for building AI-powered applications with template literals and structured outputs",
+  "type": "module",
+  "main": "dist/index.js",
+  "module": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "bin": {
+    "ai-functions": "bin/cli.js"
   },
+  "files": [
+    "dist",
+    "bin"
+  ],
   "scripts": {
     "build": "tsc",
-    "builds": "rm -fr dist/* && tsc -p tsconfig.json && tsc -p tsconfig-cjs.json && ./fixup",
-    "format": "prettier --write .",
-    "test": "dotenv -- vitest --globals"
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "lint": "eslint src",
+    "format": "prettier --write \"src/**/*.{ts,tsx,js,jsx}\" \"*.{md,mdx}\"",
+    "prepublishOnly": "pnpm run build",
+    "semantic-release": "semantic-release"
   },
+  "keywords": [
+    "ai",
+    "openai",
+    "template-literals",
+    "typescript",
+    "structured-output",
+    "cli"
+  ],
+  "author": "AI Primitives",
+  "license": "MIT",
+  "homepage": "https://github.com/ai-primitives/ai-functions#readme",
   "repository": {
     "type": "git",
-    "url": "git+https://github.com/nathanclevenger/ai-functions.git"
+    "url": "git+https://github.com/ai-primitives/ai-functions.git"
   },
-  "keywords": [],
-  "author": "",
-  "license": "MIT",
   "bugs": {
-    "url": "https://github.com/nathanclevenger/ai-functions/issues"
-  },
-  "homepage": "https://github.com/nathanclevenger/ai-functions#readme",
-  "prettier": {
-    "semi": false,
-    "singleQuote": true,
-    "trailingComma": "es5",
-    "tabWidth": 2,
-    "printWidth": 120
+    "url": "https://github.com/ai-primitives/ai-functions/issues"
   },
   "devDependencies": {
-    "@types/js-yaml": "^4.0.9",
-    "@types/json-schema": "^7.0.14",
-    "dotenv-cli": "^7.3.0",
-    "prettier": "^3.0.3",
-    "typescript": "^5.3.3",
-    "vitest": "^0.33.0"
+    "@ai-sdk/provider": "^1.0.2",
+    "@ai-sdk/provider-utils": "^2.0.4",
+    "@ai-sdk/ui-utils": "^1.0.5",
+    "@eslint/js": "^9.17.0",
+    "@opentelemetry/api": "^1.9.0",
+    "@semantic-release/commit-analyzer": "^13.0.0",
+    "@semantic-release/github": "^11.0.1",
+    "@semantic-release/npm": "^12.0.1",
+    "@semantic-release/release-notes-generator": "^14.0.1",
+    "@types/node": "^22.10.2",
+    "@typescript-eslint/eslint-plugin": "^8.18.0",
+    "@typescript-eslint/parser": "^8.18.0",
+    "dotenv": "^16.4.7",
+    "eslint": "^9.17.0",
+    "prettier": "^3.4.2",
+    "semantic-release": "^24.2.0",
+    "typescript": "^5.7.2",
+    "undici": "^7.2.0",
+    "vitest": "^2.1.8",
+    "web-streams-polyfill": "^4.0.0"
   },
   "dependencies": {
-    "js-yaml": "^4.1.0",
-    "json-schema-to-ts": "^3.0.0",
-    "kafkajs": "^2.2.4",
-    "lodash-es": "^4.17.21",
-    "mongodb": "^6.2.0",
-    "openai": "^4.26.0",
-    "p-queue": "^7.4.1",
-    "partial-json-parser": "^1.2.2",
-    "xstate": "^5.0.0-beta.40"
+    "@ai-sdk/openai": "^1.0.10",
+    "@ai-sdk/openai-compatible": "^0.0.9",
+    "ai": "^4.0.20",
+    "p-queue": "^8.0.1",
+    "zod": "^3.24.1"
   }
 }

package/db/cache.ts

@@ -1,6 +0,0 @@
-import crypto from 'node:crypto'
-
-export const md5 = (data: string | object) => {
-  const input = typeof data === 'string' ? data : JSON.stringify(data)
-  return crypto.createHash('md5').update(input).digest('hex')
-}

package/db/mongo.ts

@@ -1,75 +0,0 @@
-import type { MongoClient, Db, Collection, InsertOneResult, BSON } from 'mongodb'
-import type { ChatCompletion, ChatCompletionCreateParamsNonStreaming } from 'openai/resources'
-import { QueueInput } from '../queue/mongo'
-
-export let client: MongoClient
-export let db: Db
-export let cache: Collection
-export let cacheTTL: number
-export let events: Collection
-export let queue: Collection
-export let actors: Collection
-
-export type AIDBConfig = {
-  client: MongoClient
-  db?: Db | string
-  cache?: Collection | string
-  cacheTTL?: number
-  actors?: Collection | string
-  events?: Collection | string
-  queue?: Collection | string
-}
-
-export type AIDB = {
-  client: MongoClient
-  db: Db
-  cache: Collection
-  cacheTTL: number
-  actors: Collection
-  events: Collection
-  queue: Collection
-  log: (prompt: ChatCompletionCreateParamsNonStreaming, completion: ChatCompletion) => Promise<InsertOneResult<any>>
-  send: (input: QueueInput | QueueInput[]) => Promise<InsertOneResult<any>>
-}
-
-export const AIDB = (args: AIDBConfig | MongoClient) => {
-  const config = (args as AIDBConfig).client ? (args as AIDBConfig) : { client: args as MongoClient }
-  client = config.client
-  db = typeof config.db === 'string' ? client.db(config.db) : config.db ?? client.db()
-  cache = typeof config.cache === 'string' ? db.collection(config.cache) : config.cache ?? db.collection('ai-cache')
-  cacheTTL = config.cacheTTL ?? 1000 * 60 * 60 * 24 * 30
-  events =
-    typeof config.events === 'string' ? db.collection(config.events) : config.events ?? db.collection('ai-events')
-  // queue = typeof config.queue === 'string' ? db.collection(config.queue) : config.queue ?? db.collection('ai-queue')
-  // actors = typeof config.queue === 'string' ? db.collection(config.queue) : config.queue ?? db.collection('ai-actors')
-  return { client, db, cache, actors, events, queue, log }  as AIDB // , send } as AIDB
-}
-
-// const send = (input: QueueInput | QueueInput[]) =>
-//   Array.isArray(input) ? queue.insertMany(input) : queue.insertOne(input)
-
-const log = (prompt: ChatCompletionCreateParamsNonStreaming, completion: ChatCompletion) => {
-  const system = prompt.messages.find((message: any) => message.role === 'system')?.content
-  const userMessages = prompt.messages.filter((message: any) => message.role === 'user').map((message) => message.content)
-  const user = userMessages.length === 1 ? userMessages[0] : userMessages
-  const content = completion.choices?.[0].message.content
-  const tool = completion.choices?.[0].message.tool_calls?.[0].function
-  let functionData: object | string | undefined
-  try {
-    if (tool) functionData = JSON.parse(tool.arguments)
-  } catch (err) {
-    functionData = tool?.arguments
-  }
-  const timestamp = new Date()
-  const event = {
-    timestamp,
-    user,
-    system,
-    content,
-    functionName: tool?.name,
-    functionData,
-    prompt,
-    completion,
-  }
-  return events.insertOne(event, { ignoreUndefined: true })
-}

package/dist/mjs/db/cache.d.ts

@@ -1 +0,0 @@
-export declare const md5: (data: string | object) => string;

package/dist/mjs/db/cache.js

@@ -1,5 +0,0 @@
-import crypto from 'node:crypto';
-export const md5 = (data) => {
-    const input = typeof data === 'string' ? data : JSON.stringify(data);
-    return crypto.createHash('md5').update(input).digest('hex');
-};

package/dist/mjs/db/mongo.d.ts

@@ -1,31 +0,0 @@
-import type { MongoClient, Db, Collection, InsertOneResult } from 'mongodb';
-import type { ChatCompletion, ChatCompletionCreateParamsNonStreaming } from 'openai/resources';
-import { QueueInput } from '../queue/mongo';
-export declare let client: MongoClient;
-export declare let db: Db;
-export declare let cache: Collection;
-export declare let cacheTTL: number;
-export declare let events: Collection;
-export declare let queue: Collection;
-export declare let actors: Collection;
-export type AIDBConfig = {
-    client: MongoClient;
-    db?: Db | string;
-    cache?: Collection | string;
-    cacheTTL?: number;
-    actors?: Collection | string;
-    events?: Collection | string;
-    queue?: Collection | string;
-};
-export type AIDB = {
-    client: MongoClient;
-    db: Db;
-    cache: Collection;
-    cacheTTL: number;
-    actors: Collection;
-    events: Collection;
-    queue: Collection;
-    log: (prompt: ChatCompletionCreateParamsNonStreaming, completion: ChatCompletion) => Promise<InsertOneResult<any>>;
-    send: (input: QueueInput | QueueInput[]) => Promise<InsertOneResult<any>>;
-};
-export declare const AIDB: (args: AIDBConfig | MongoClient) => AIDB;

package/dist/mjs/db/mongo.js

@@ -1,48 +0,0 @@
-export let client;
-export let db;
-export let cache;
-export let cacheTTL;
-export let events;
-export let queue;
-export let actors;
-export const AIDB = (args) => {
-    const config = args.client ? args : { client: args };
-    client = config.client;
-    db = typeof config.db === 'string' ? client.db(config.db) : config.db ?? client.db();
-    cache = typeof config.cache === 'string' ? db.collection(config.cache) : config.cache ?? db.collection('ai-cache');
-    cacheTTL = config.cacheTTL ?? 1000 * 60 * 60 * 24 * 30;
-    events =
-        typeof config.events === 'string' ? db.collection(config.events) : config.events ?? db.collection('ai-events');
-    // queue = typeof config.queue === 'string' ? db.collection(config.queue) : config.queue ?? db.collection('ai-queue')
-    // actors = typeof config.queue === 'string' ? db.collection(config.queue) : config.queue ?? db.collection('ai-actors')
-    return { client, db, cache, actors, events, queue, log }; // , send } as AIDB
-};
-// const send = (input: QueueInput | QueueInput[]) =>
-//   Array.isArray(input) ? queue.insertMany(input) : queue.insertOne(input)
-const log = (prompt, completion) => {
-    const system = prompt.messages.find((message) => message.role === 'system')?.content;
-    const userMessages = prompt.messages.filter((message) => message.role === 'user').map((message) => message.content);
-    const user = userMessages.length === 1 ? userMessages[0] : userMessages;
-    const content = completion.choices?.[0].message.content;
-    const tool = completion.choices?.[0].message.tool_calls?.[0].function;
-    let functionData;
-    try {
-        if (tool)
-            functionData = JSON.parse(tool.arguments);
-    }
-    catch (err) {
-        functionData = tool?.arguments;
-    }
-    const timestamp = new Date();
-    const event = {
-        timestamp,
-        user,
-        system,
-        content,
-        functionName: tool?.name,
-        functionData,
-        prompt,
-        completion,
-    };
-    return events.insertOne(event, { ignoreUndefined: true });
-};