ai-functions 0.1.4 → 0.2.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2023 Nathan Clevenger
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -2,7 +2,8 @@
 
 Library for Developing and Managing AI Functions (including OpenAI GPT4 / GPT3.5)
 
-Key Features: 
+Key Features:
+
 - Enables easy development of AI functions
 
 ```javascript
@@ -12,16 +13,16 @@ const { ai, gpt, list } = AI({ apiKey: OPENAI_API_KEY })
 ```
 
 Then you can use magic `ai` functions:
-```javascript
 
-const categorizeProduct = ai.categorizeProduct({ 
+```javascript
+const categorizeProduct = ai.categorizeProduct({
   productType: 'App | API | Marketplace | Platform | Packaged Service | Professional Service | Website',
   customer: 'ideal customer profile in 3-5 words',
   solution: 'describe the offer in 4-10 words',
   description: 'website meta description',
 })
 
-const product = await (categorizeProduct({ domain: name }))
+const product = await categorizeProduct({ domain: name })
 ```
 
 you can also use `list` tagged template as a convienence function:
@@ -43,7 +44,7 @@ Or in a more complex example:
 
 ```javascript
 const listBlogPosts = (count, topic) => list`${count} blog post titles about ${topic}`
-const writeBlogPost = title => gpt`write a blog post in markdown starting with "# ${title}"`
+const writeBlogPost = (title) => gpt`write a blog post in markdown starting with "# ${title}"`
 
 async function* writeBlog(count, topic) {
   for await (const title of listBlogPosts(count, topic)) {
@@ -55,4 +56,4 @@ async function* writeBlog(count, topic) {
 for await (const post of writeBlog(25, 'future of car sales')) {
   console.log({ post })
 }
-```
+```

package/db/cache.ts

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

@@ -0,0 +1,75 @@
+import type { MongoClient, Db, Collection, InsertOneResult, BSON } from 'mongodb'
+import type { ChatCompletion, ChatCompletionCreateParamsNonStreaming } from 'openai/resources/index.mjs'
+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, 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) => 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: 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 })
+}