@drax/ai-back 0.13.0

package/test/KnowledgeService.test.ts

@@ -0,0 +1,36 @@
+import {describe, test, expect} from 'vitest'
+import {KnowledgeService, OpenAiProvider, OpenAiProviderFactory} from "../src";
+
+
+describe('KnowledgeService Test', () => {
+
+    test('KnowledgeService URL', async () => {
+        const ks = new KnowledgeService()
+        const content = await ks.getFromUrl('https://exigroup.com.ar/preguntas-frecuentes/')
+        console.log(content)
+
+        const openAi = OpenAiProviderFactory.instance()
+        expect(openAi).toBeInstanceOf(OpenAiProvider)
+
+        const systemPrompt = `
+        Por favor, optimiza el siguiente contenido extraído de una página web o documento PDF:
+        1. Elimina cualquier código JavaScript, CSS o HTML residual.
+        2. Organiza el contenido de manera lógica y coherente.
+        3. Elimina información redundante o irrelevante.
+        4. Asegúrate de que el contenido sea claro y fácil de entender.
+        5. No uses marcadores de posición, formato markdown o etiquetas HTML, solo texto simple.
+        `
+        const userInput = content
+        const r = await openAi.prompt({systemPrompt, userInput})
+
+        const contentOptimal = r.output
+
+        console.log(contentOptimal)
+        const chunks = ks.chunkTextBySentence(contentOptimal)
+        console.log(chunks)
+
+        expect(chunks).toBeInstanceOf(Array)
+
+    }, 30000) // 10 seconds timeout
+
+})

package/test/OpenAiProvider.test.ts

@@ -0,0 +1,184 @@
+import {describe, test, expect} from 'vitest'
+import {OpenAiProvider, OpenAiProviderFactory} from "../src";
+import z from "zod";
+import {IPromptMemory, IPromptMessage} from "../src/interfaces/IAIProvider";
+
+
+describe('OpenAi Test', () => {
+
+    test('OpenAi Factory', () => {
+        const openAi = OpenAiProviderFactory.instance()
+        expect(openAi).toBeInstanceOf(OpenAiProvider)
+    })
+
+    test('OpenAi prompt', async () => {
+        const openAi = OpenAiProviderFactory.instance()
+        expect(openAi).toBeInstanceOf(OpenAiProvider)
+
+        const systemPrompt = "You are an AI assistant."
+        const userInput = "What is the capital of Argentina?"
+        const r = await openAi.prompt({systemPrompt, userInput})
+        console.log(r)
+        expect(r.output).toBeDefined()
+        expect(/Buenos Aires/i.test(r.output)).toBeTruthy()
+    })
+
+    test('OpenAi prompt jsonSchema', async () => {
+        const openAi = OpenAiProviderFactory.instance()
+        expect(openAi).toBeInstanceOf(OpenAiProvider)
+
+        const systemPrompt = "You are an AI assistant."
+        const userInput = "What is the most famous pokemon"
+        const jsonSchema = {
+            type: "json_schema",
+            json_schema: {
+                name: "element_description",
+                schema: {
+                    type: "object",
+                    properties: {
+                        name: {
+                            type: "string",
+                            description: "El nombre del elemento"
+                        },
+                        type: {
+                            type: "string",
+                            description: "El tipo de elemento, como archivo, carpeta, enlace, etc."
+                        },
+                        description: {
+                            type: "string",
+                            description: "Una breve descripción del elemento"
+                        }
+                    },
+                    required: ["name", "type", "description"]
+                }
+            }
+        }
+
+
+        const r = await openAi.prompt({systemPrompt, userInput, jsonSchema})
+        console.log(r)
+        expect(r.output).toBeDefined
+        const pokemon = JSON.parse(r.output)
+        expect(pokemon.name).toBe("Pikachu")
+    })
+
+    test('OpenAi prompt zodSchema', async () => {
+        const openAi = OpenAiProviderFactory.instance()
+        expect(openAi).toBeInstanceOf(OpenAiProvider)
+
+        const systemPrompt = "You are an AI assistant."
+        const userInput = "What is the most famous pokemon"
+        const zodSchema = z.object({
+            name: z.string(),
+            type: z.string(),
+            description: z.string()
+        })
+
+        const r = await openAi.prompt({systemPrompt, userInput, zodSchema})
+        console.log(r)
+        expect(r.output).toBeDefined
+        const pokemon = zodSchema.parse(JSON.parse(r.output))
+        expect(pokemon.name).toBe("Pikachu")
+    })
+
+    test('OpenAi prompt zodSchema and Knowledge base', async () => {
+        const openAi = OpenAiProviderFactory.instance()
+        expect(openAi).toBeInstanceOf(OpenAiProvider)
+
+        const systemPrompt = "You are an AI assistant."
+        const knowledgeBase = ["Charmander is a legendary and most famous Pokémon from the Pokémon series. It has a type of fire."]
+        const zodSchema = z.object({
+            name: z.string(),
+            type: z.string(),
+            description: z.string()
+        })
+
+        const userInput = "What is the most famous pokemon"
+        const r = await openAi.prompt({systemPrompt, userInput, zodSchema, knowledgeBase})
+        console.log(r)
+        expect(r.output).toBeDefined
+        const pokemon = zodSchema.parse(JSON.parse(r.output))
+        expect(pokemon.name).toBe("Charmander")
+    })
+
+    test('OpenAi prompt zodSchema, Knowledge base and history', async () => {
+        const openAi = OpenAiProviderFactory.instance()
+        expect(openAi).toBeInstanceOf(OpenAiProvider)
+
+        const systemPrompt = "You are an AI assistant."
+
+        const knowledgeBase = ["Charmander is a legendary and most famous Pokémon from the Pokémon series. It has a type of fire."]
+        const history: IPromptMessage[] = [
+            {role: 'assistant', content: 'what is your favourite pokemon'},
+            {role: 'user', content: 'My favourite pokemon is Bulbasaur and is a water type'},
+            {role: 'assistant', content: 'Do you like team Rocket?'},
+            {role: 'user', content: 'No, never!'},
+        ]
+
+        const zodSchema = z.object({
+            famous: z.object({
+                name: z.string(),
+                type: z.string(),
+                description: z.string()
+            }),
+            favorite: z.object({
+                name: z.string(),
+                type: z.string(),
+                description: z.string()
+            })
+
+        })
+
+        const userInput = "What is the most famous pokemon and what is my favorite?"
+
+        const r = await openAi.prompt({systemPrompt, userInput, zodSchema, knowledgeBase, history})
+        console.log(r)
+        expect(r.output).toBeDefined
+        const pokemons = zodSchema.parse(JSON.parse(r.output))
+        expect(pokemons.famous.name).toBe("Charmander")
+        expect(pokemons.famous.type).toBeOneOf(["fire", "Fire"])
+        expect(pokemons.favorite.name).toBe("Bulbasaur")
+        expect(pokemons.favorite.type).toBeOneOf(["water", "Water"]) //Lie in this test, but it's a part of the schema and history for demonstration purposes.'
+    })
+
+    test('OpenAi prompt zodSchema, Knowledge base and memory', async () => {
+        const openAi = OpenAiProviderFactory.instance()
+        expect(openAi).toBeInstanceOf(OpenAiProvider)
+
+        const systemPrompt = "You are an AI assistant."
+
+        const knowledgeBase = ["Charmander is a legendary and most famous Pokémon from the Pokémon series. It has a type of fire."]
+
+
+        const memory: IPromptMemory[] = [
+            {key: 'favorite_pokemon', value: 'Squirtle'},
+            {key: 'favourite_pokemon_type', value: 'water'},
+        ]
+
+        const zodSchema = z.object({
+            famous: z.object({
+                name: z.string(),
+                type: z.string(),
+                description: z.string()
+            }),
+            favorite: z.object({
+                name: z.string(),
+                type: z.string(),
+                description: z.string()
+            })
+
+        })
+
+        const userInput = "What is the most famous pokemon and what is my favorite?"
+
+        const r = await openAi.prompt({systemPrompt, userInput, zodSchema, knowledgeBase, memory})
+        console.log(r)
+        expect(r.output).toBeDefined
+        const pokemons = zodSchema.parse(JSON.parse(r.output))
+        expect(pokemons.famous.name).toBe("Charmander")
+        expect(pokemons.famous.type).toBe("fire")
+        expect(pokemons.favorite.name).toBe("Squirtle")
+        expect(pokemons.favorite.type).toBe("water") //Lie in this test, but it's a part of the schema and history for demonstration purposes.'
+    })
+
+})

package/tsconfig.json

@@ -0,0 +1,16 @@
+{
+  "extends": "../../../tsconfig.base.json",
+  "compilerOptions": {
+    "rootDir": "src",
+    "outDir": "dist",
+    "declarationDir": "./types",
+  },
+  "exclude": ["test", "types","dist"],
+  "ts-node": {
+    "esm": true
+  },
+  "tsc-alias": {
+    "resolveFullPaths": true,
+    "verbose": false
+  }
+}