ai-functions 0.2.2 → 0.2.3

package/functions/ai.ts

@@ -3,6 +3,7 @@ import { ChatCompletion, ChatCompletionCreateParamsBase } from 'openai/resources
 import { AIDB, AIDBConfig } from '../db/mongo'
 import { dump } from 'js-yaml'
 import { generateSchema } from '../utils/schema'
+import { FromSchema } from 'json-schema-to-ts'
 
 export type AIConfig = ClientOptions & {
   db?: AIDBConfig
@@ -17,14 +18,14 @@ export type FunctionCallOptions = Omit<ChatCompletionCreateParamsBase, 'messages
   description?: string
 }
 
-type AIFunctions<T extends Record<string, any> = Record<string, any>> = {
+type AIFunctions<T extends Record<string, any> = Record<string,any>> = {
   [K in keyof T]: (
-      returnSchema: T[K], 
-      callOptions?: FunctionCallOptions
+    returnSchema: T[K], 
+    callOptions?: FunctionCallOptions
   ) => (
-      args: string | object, 
-      callOptions?: FunctionCallOptions
-  ) => Promise<T[K]>
+    args: string | object, 
+    callOptions?: FunctionCallOptions
+  ) => Promise<{ [P in keyof T[K]]: T[K][P] }>
 }
 
 export const AI = (config: AIConfig = {}) => {
@@ -78,7 +79,9 @@ export const AI = (config: AIConfig = {}) => {
           }
           if (system) prompt.messages.unshift({ role: 'system', content: system })
           const completion = await openai.chat.completions.create(prompt) as ChatCompletion
-          let data, error
+          const schema = generateSchema(returnSchema)
+          let data: FromSchema<typeof schema>
+          let error
           const { message } = completion.choices?.[0]
           console.log({ message })
           prompt.messages.push(message)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-functions",
-  "version": "0.2.2",
+  "version": "0.2.3",
   "description": "Library for Developing and Managing AI Functions (including OpenAI GPT4 / GPT3.5)",
   "main": "index.js",
   "type": "module",
@@ -37,7 +37,7 @@
   },
   "dependencies": {
     "js-yaml": "^4.1.0",
-    "json-schema-to-ts": "^2.9.2",
+    "json-schema-to-ts": "^3.0.0",
     "kafkajs": "^2.2.4",
     "lodash-es": "^4.17.21",
     "mongodb": "^6.2.0",

package/utils/schema.ts

@@ -1,10 +1,10 @@
-import { JSONSchema7 } from 'json-schema-to-ts'
+import { JSONSchema } from 'json-schema-to-ts'
 
 // This is a helper function to generate a JSON schema for string properties.
 // It checks if a string includes '|' which would indicate an enum,
 // or if it starts with 'number: ' or 'boolean: ' which would indicate
 // a number or boolean type respectively, otherwise it defaults to string.
-export const parseStringDescription = (description: string): JSONSchema7 => {
+export const parseStringDescription = (description: string): JSONSchema => {
   // Check if the description indicates an enum for string type
   if (description.includes('|')) {
     return { type: 'string', enum: description.split('|').map((v) => v.trim()) }
@@ -37,13 +37,13 @@ export const parseStringDescription = (description: string): JSONSchema7 => {
  * and values as descriptions or nested property description objects.
  * @returns A JSON schema object based on the provided descriptions.
  */
-export const generateSchema = (propDescriptions: Record<string, string | Record<string, any>>): JSONSchema7 => {
+export const generateSchema = (propDescriptions: Record<string, string | Record<string, any>>): JSONSchema => {
   // If the propDescriptions is for an object structure
   if (typeof propDescriptions !== 'object' || propDescriptions === null || Array.isArray(propDescriptions)) {
     throw new Error('The propDescriptions parameter should be an object.')
   }
 
-  const properties: Record<string, JSONSchema7> = {}
+  const properties: Record<string, JSONSchema> = {}
   const required: string[] = Object.keys(propDescriptions)
 
   for (const [key, description] of Object.entries(propDescriptions)) {