ai-functions 0.2.1 → 0.2.2

package/functions/ai.test.ts

@@ -13,6 +13,8 @@ describe('AI Functions', async () => {
   
   const categorizeWord = ai.categorizeWord({
     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' })
 
   it('should be a function', () => {
@@ -20,12 +22,20 @@ describe('AI Functions', async () => {
     expect(typeof categorizeWord).toBe('function')
   })
 
-  it('should return a promise', async () => {
-    expect(await categorizeWord('destroy')).toMatchObject({ type: 'Verb' })
+  it('Destroy should be a verb', async () => {
+    expect(await categorizeWord('destroy')).toMatchObject({ type: 'Verb', example: 'I will destroy the old building.' })
   })
 
-  it('should return a promise', async () => {
-    expect(await categorizeWord('dog')).toMatchObject({ type: 'Noun' })
+  it('Dog should be a Noun', async () => {
+    const dog = await categorizeWord({ word: 'dog' })
+    expect(dog).toMatchObject({ type: 'Noun', example: 'I have a 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." })
+  })
+
 })

package/functions/ai.ts

@@ -17,6 +17,16 @@ export type FunctionCallOptions = Omit<ChatCompletionCreateParamsBase, 'messages
   description?: string
 }
 
+type AIFunctions<T extends Record<string, any> = Record<string, any>> = {
+  [K in keyof T]: (
+      returnSchema: T[K], 
+      callOptions?: FunctionCallOptions
+  ) => (
+      args: string | object, 
+      callOptions?: FunctionCallOptions
+  ) => Promise<T[K]>
+}
+
 export const AI = (config: AIConfig = {}) => {
   const { model = 'gpt-4-1106-preview', system, ...rest } = config 
   const openai = config.openai ?? new OpenAI(rest)
@@ -32,8 +42,7 @@ export const AI = (config: AIConfig = {}) => {
   //   messages: [{ role: 'user', content: 'hello' }],
   // })
 
-  const ai: Record<string, (args: string | object, callOptions?: FunctionCallOptions) => 
-    (args: string | object, callOptions?: FunctionCallOptions) => Promise<any>> = new Proxy(
+  const ai: AIFunctions = new Proxy(
     {},
     {
       get: (target, functionName: string, receiver) => {
@@ -89,7 +98,7 @@ export const AI = (config: AIConfig = {}) => {
                 : completion.usage.prompt_tokens * 0.00015 + completion.usage.completion_tokens * 0.0002) * 100000
             ) / 100000 : undefined
           // completion.usage = camelcaseKeys(completion.usage)
-          console.log({ data, content, error, cost })
+          console.log({ data, content, error, cost, usage: completion.usage })
           return meta ? { prompt, content, data, error, cost, ...completion } : data ?? content
         }
       },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-functions",
-  "version": "0.2.1",
+  "version": "0.2.2",
   "description": "Library for Developing and Managing AI Functions (including OpenAI GPT4 / GPT3.5)",
   "main": "index.js",
   "type": "module",