ai-functions 0.2.16 → 0.2.18

package/dist/mjs/functions/ai.d.ts

@@ -14,5 +14,7 @@ type AIFunctions<T = Record<string, string>> = Record<string, (returnSchema: T,
 export declare const AI: (config?: AIConfig) => {
     ai: AIFunctions<Record<string, string>>;
     openai: OpenAI;
+    gpt: (strings: string[], ...values: string[]) => Promise<string | null>;
+    list: (strings: string[], ...values: string[]) => AsyncGenerator<string | undefined, void, unknown>;
 };
 export {};

package/dist/mjs/functions/ai.js

@@ -2,9 +2,13 @@ import { OpenAI, } from 'openai';
 // import { AIDB, AIDBConfig } from '../db/mongo'
 import { dump } from 'js-yaml';
 import { generateSchema } from '../utils/schema';
+import { GPT } from './gpt';
+import { StreamingList } from './list';
 export const AI = (config = {}) => {
-    const { model = 'gpt-4-1106-preview', system, ...rest } = config;
+    const { model = 'gpt-4-turbo-preview', system, ...rest } = config;
     const openai = config.openai ?? new OpenAI(rest);
+    const { gpt } = GPT(config);
+    const { list } = StreamingList(config);
     // const { client, db, cache, events, queue } = config.db ? AIDB(config.db) : {}
     // const prompt = {
     //   model,
@@ -20,7 +24,7 @@ export const AI = (config = {}) => {
         get: (target, functionName, receiver) => {
             target[functionName] = (returnSchema, options) => async (args, callOptions) => {
                 console.log(generateSchema(returnSchema));
-                const { system, description, model = 'gpt-3.5-turbo', meta = false, ...rest } = { ...options, ...callOptions };
+                const { system, description, model = config.model ?? 'gpt-4-turbo-preview', meta = false, ...rest } = { ...options, ...callOptions };
                 const prompt = {
                     model,
                     messages: [
@@ -75,5 +79,5 @@ export const AI = (config = {}) => {
             return target[functionName];
         },
     });
-    return { ai, openai }; //, client, db, cache, events, queue }
+    return { ai, openai, gpt, list }; //, client, db, cache, events, queue }
 };

package/dist/mjs/functions/ai.test.js

@@ -9,21 +9,21 @@ describe('AI Functions', async () => {
         type: 'Noun | Verb | Adjective | Adverb | Pronoun | Preposition | Conjunction | Interjection | Other',
         example: 'use the word in a sentence',
         // partOfSpeech: 'Part of speech'
-    }, { seed: 1, model: 'gpt-3.5-turbo' });
+    }, { seed: 1, model: 'gpt-4o' });
     it('should be a function', () => {
         expect(typeof categorizeWord).toBe('function');
     });
     it('Destroy should be a verb', async () => {
-        expect(await categorizeWord('destroy')).toMatchObject({ type: 'Verb', example: 'I will destroy the old building.' });
+        expect(await categorizeWord('destroy')).toMatchObject({ type: 'Verb', example: 'The earthquake destroyed several buildings in the city.' });
     });
     it('Dog should be a Noun', async () => {
         const dog = await categorizeWord({ word: 'dog' });
-        expect(dog).toMatchObject({ type: 'Noun', example: 'I have a dog.' });
+        expect(dog).toMatchObject({ type: 'Noun', example: 'She loves playing with her dog.' });
     });
     it('Large should be an Adjective', async () => {
         expect(await categorizeWord({ word: 'large' })).toMatchObject({ type: 'Adjective', example: 'She has a large collection of books.' });
     });
     it('To should be an Preposition', async () => {
-        expect(await categorizeWord('to')).toMatchObject({ type: 'Preposition', example: "I'm going to the park." });
+        expect(await categorizeWord('to')).toMatchObject({ type: 'Preposition', example: "I need to go to the store." });
     });
 });

package/functions/ai.test.ts

@@ -15,7 +15,7 @@ describe('AI Functions', async () => {
     type: 'Noun | Verb | Adjective | Adverb | Pronoun | Preposition | Conjunction | Interjection | Other',
     example: 'use the word in a sentence',
     // partOfSpeech: 'Part of speech'
-  }, { seed: 1, model: 'gpt-3.5-turbo' })
+  }, { seed: 1, model: 'gpt-4o' })
 
   it('should be a function', () => {
     
@@ -23,19 +23,19 @@ describe('AI Functions', async () => {
   })
 
   it('Destroy should be a verb', async () => {
-    expect(await categorizeWord('destroy')).toMatchObject({ type: 'Verb', example: 'I will destroy the old building.' })
+    expect(await categorizeWord('destroy')).toMatchObject({ type: 'Verb', example: 'The earthquake destroyed several buildings in the city.' })
   })
 
   it('Dog should be a Noun', async () => {
     const dog = await categorizeWord({ word: 'dog' })
-    expect(dog).toMatchObject({ type: 'Noun', example: 'I have a dog.' })
+    expect(dog).toMatchObject({ type: 'Noun', example: 'She loves playing with her dog.' })
   })
 
   it('Large should be an Adjective', async () => {
     expect(await categorizeWord({ word: 'large' })).toMatchObject({ type: 'Adjective', example: 'She has a large collection of books.' })
   })
   it('To should be an Preposition', async () => {
-    expect(await categorizeWord('to')).toMatchObject({ type: 'Preposition', example: "I'm going to the park." })
+    expect(await categorizeWord('to')).toMatchObject({ type: 'Preposition', example: "I need to go to the store." })
   })
 
 })

package/functions/ai.ts

@@ -4,6 +4,8 @@ import { ChatCompletion, ChatCompletionCreateParamsBase } from 'openai/resources
 import { dump } from 'js-yaml'
 import { generateSchema } from '../utils/schema'
 import { FromSchema } from 'json-schema-to-ts'
+import { GPT } from './gpt'
+import { StreamingList } from './list'
 
 export type AIConfig = ClientOptions & {
   // db?: AIDBConfig
@@ -28,8 +30,10 @@ type AIFunctions<T = Record<string,string>> = Record<string, (
 >
 
 export const AI = (config: AIConfig = {}) => {
-  const { model = 'gpt-4-1106-preview', system, ...rest } = config 
+  const { model = 'gpt-4-turbo-preview', system, ...rest } = config 
   const openai = config.openai ?? new OpenAI(rest)
+  const { gpt } = GPT(config)
+  const { list } = StreamingList(config)
   // const { client, db, cache, events, queue } = config.db ? AIDB(config.db) : {}
   // const prompt = {
   //   model,
@@ -48,7 +52,7 @@ export const AI = (config: AIConfig = {}) => {
       get: (target, functionName: string, receiver) => {
         target[functionName] = (returnSchema: Record<string,any>, options: FunctionCallOptions) => async (args: string | object, callOptions?: FunctionCallOptions) => {
           console.log(generateSchema(returnSchema))
-          const { system, description, model = 'gpt-3.5-turbo', meta = false, ...rest } = { ...options, ...callOptions }
+          const { system, description, model = config.model ?? 'gpt-4-turbo-preview', meta = false, ...rest } = { ...options, ...callOptions }
           const prompt: ChatCompletionCreateParamsBase = {
             model,
             messages: [
@@ -107,5 +111,5 @@ export const AI = (config: AIConfig = {}) => {
       },
     }
   )
-  return { ai, openai } //, client, db, cache, events, queue }
+  return { ai, openai, gpt, list } //, client, db, cache, events, queue }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-functions",
-  "version": "0.2.16",
+  "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",

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

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

package/dist/cjs/db/cache.js

@@ -1,12 +0,0 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.md5 = void 0;
-const node_crypto_1 = __importDefault(require("node:crypto"));
-const md5 = (data) => {
-    const input = typeof data === 'string' ? data : JSON.stringify(data);
-    return node_crypto_1.default.createHash('md5').update(input).digest('hex');
-};
-exports.md5 = md5;

package/dist/cjs/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/cjs/db/mongo.js

@@ -1,47 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.AIDB = exports.actors = exports.queue = exports.events = exports.cacheTTL = exports.cache = exports.db = exports.client = void 0;
-const AIDB = (args) => {
-    var _a, _b, _c, _d;
-    const config = args.client ? args : { client: args };
-    exports.client = config.client;
-    exports.db = typeof config.db === 'string' ? exports.client.db(config.db) : (_a = config.db) !== null && _a !== void 0 ? _a : exports.client.db();
-    exports.cache = typeof config.cache === 'string' ? exports.db.collection(config.cache) : (_b = config.cache) !== null && _b !== void 0 ? _b : exports.db.collection('ai-cache');
-    exports.cacheTTL = (_c = config.cacheTTL) !== null && _c !== void 0 ? _c : 1000 * 60 * 60 * 24 * 30;
-    exports.events =
-        typeof config.events === 'string' ? exports.db.collection(config.events) : (_d = config.events) !== null && _d !== void 0 ? _d : exports.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: exports.client, db: exports.db, cache: exports.cache, actors: exports.actors, events: exports.events, queue: exports.queue, log }; // , send } as AIDB
-};
-exports.AIDB = AIDB;
-// const send = (input: QueueInput | QueueInput[]) =>
-//   Array.isArray(input) ? queue.insertMany(input) : queue.insertOne(input)
-const log = (prompt, completion) => {
-    var _a, _b, _c, _d;
-    const system = (_a = prompt.messages.find((message) => message.role === 'system')) === null || _a === void 0 ? void 0 : _a.content;
-    const userMessages = prompt.messages.filter((message) => message.role === 'user').map((message) => message.content);
-    const user = userMessages.length === 1 ? userMessages[0] : userMessages;
-    const content = (_b = completion.choices) === null || _b === void 0 ? void 0 : _b[0].message.content;
-    const tool = (_d = (_c = completion.choices) === null || _c === void 0 ? void 0 : _c[0].message.tool_calls) === null || _d === void 0 ? void 0 : _d[0].function;
-    let functionData;
-    try {
-        if (tool)
-            functionData = JSON.parse(tool.arguments);
-    }
-    catch (err) {
-        functionData = tool === null || tool === void 0 ? void 0 : tool.arguments;
-    }
-    const timestamp = new Date();
-    const event = {
-        timestamp,
-        user,
-        system,
-        content,
-        functionName: tool === null || tool === void 0 ? void 0 : tool.name,
-        functionData,
-        prompt,
-        completion,
-    };
-    return exports.events.insertOne(event, { ignoreUndefined: true });
-};