ai-functions 0.2.10 → 0.2.13

package/db/mongo.ts

@@ -1,5 +1,5 @@
 import type { MongoClient, Db, Collection, InsertOneResult, BSON } from 'mongodb'
-import type { ChatCompletion, ChatCompletionCreateParamsNonStreaming } from 'openai/resources/index.mjs'
+import type { ChatCompletion, ChatCompletionCreateParamsNonStreaming } from 'openai/resources'
 import { QueueInput } from '../queue/mongo'
 
 export let client: MongoClient
@@ -49,8 +49,8 @@ export const AIDB = (args: AIDBConfig | MongoClient) => {
 //   Array.isArray(input) ? queue.insertMany(input) : queue.insertOne(input)
 
 const log = (prompt: ChatCompletionCreateParamsNonStreaming, completion: ChatCompletion) => {
-  const system = prompt.messages.find((message) => message.role === 'system')?.content
-  const userMessages = prompt.messages.filter((message) => message.role === 'user').map((message) => message.content)
+  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

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

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

package/dist/cjs/db/cache.js

@@ -0,0 +1,12 @@
+"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

@@ -0,0 +1,31 @@
+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

@@ -0,0 +1,47 @@
+"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 });
+};