feat(ai): added base for ai suggestions

Author: slaveeks

package.json

@@ -11,7 +11,7 @@
     "dev:up": "docker-compose -f docker-compose.dev.yml up -d",
     "dev:down": "docker-compose -f docker-compose.dev.yml down",
     "build": "tsc",
-    "convert": "node ./convertors/set-user-project-last-visit.js",
+    "convert": "node ./convertors/set-user-workspaces-membership.js",
     "migrations:create": "docker-compose exec api yarn migrate-mongo create",
     "migrations:up": "docker-compose exec api yarn migrate-mongo up",
     "migrations:down": "docker-compose exec api yarn migrate-mongo down",
@@ -33,6 +33,7 @@
     "typescript": "^4.7.4"
   },
   "dependencies": {
+    "@ai-sdk/openai": "^2.0.64",
     "@amplitude/node": "^1.10.0",
     "@graphql-tools/merge": "^8.3.1",
     "@graphql-tools/schema": "^8.5.1",
@@ -55,6 +56,7 @@
     "@types/node-fetch": "^2.5.4",
     "@types/safe-regex": "^1.1.6",
     "@types/uuid": "^8.3.4",
+    "ai": "^5.0.89",
     "amqp-connection-manager": "^3.1.0",
     "amqplib": "^0.5.5",
     "apollo-server-express": "^3.10.0",
@@ -86,6 +88,8 @@
     "prom-client": "^15.1.3",
     "safe-regex": "^2.1.0",
     "ts-node-dev": "^2.0.0",
-    "uuid": "^8.3.2"
-  }
-}
+    "uuid": "^8.3.2",
+    "zod": "^4.1.12"
+  },
+  "packageManager": "yarn@1.22.22+sha512.a6b2f7906b721bba3d67d4aff083df04dad64c399707841b7acf00f6b133b7ac24255f2652fa22ae3534329dc6180534e98d17432037ff6fd140556e2bb3137e"
+}

src/models/eventsFactory.js

@@ -7,7 +7,7 @@ const mongo = require('../mongo');
 const Event = require('../models/event');
 const { ObjectID } = require('mongodb');
 const { composeEventPayloadByRepetition } = require('../utils/merge');
-const { openAIApi } = require('../openai');
+const { vercelAIApi } = require('../resolvers/vercel-ai');
 
 const MAX_DB_READ_BATCH_SIZE = Number(process.env.MAX_DB_READ_BATCH_SIZE);
 
@@ -180,22 +180,22 @@ class EventsFactory extends Factory {
   }
 
   /**
-   * Ask AI about solving repetition
+   * Generate AI suggestion for the event
    *
-   * @param {string} repetitionId - repetition ID
-   * @param {string} eventId - event ID
-   *
-   * @return {string} - AI answer
+   * @param {string} projectId - event's project
+   * @param {string} eventId - event id
+   * @param {string} originalEventId - original event id
+   * @returns {Promise<string>} AI suggestion for the event
    */
-  async askAi(repetitionId, eventId) {
-    const repetition = await this.getEventRepetition(repetitionId, eventId);
+  async aiSuggestion(eventId, originalEventId) {
+    const repetition = await this.getEventRepetition(eventId, originalEventId);
 
     if (!repetition) {
       throw new Error('Repetition not found');
     }
 
-    const payload = repetition.event.payload;
-    const solution = await openAIApi.solveEvent(payload);
+    const payload = repetition.payload;
+    const solution = await vercelAIApi.generateSuggestion(payload);
 
     return solution;
   }

src/openai/index.ts

@@ -89,11 +89,18 @@ module.exports = {
       return factory.findChartData(days, timezoneOffset, groupHash);
     },
 
-    async askAi({ projectId, id: eventId, originalEventId }, _args, context) {
+    /**
+     * Return AI suggestion for the event
+     *
+     * @param {string} projectId - event's project
+     * @param {string} eventId - event id
+     * @param {string} originalEventId - original event id
+     * @returns {Promise<string>} AI suggestion for the event
+     */
+    async aiSuggestion({ projectId, id: eventId, originalEventId }, _args, context) {
       const factory = getEventsFactory(context, projectId);
-      const aiAnswer = await factory.askAi(eventId, originalEventId);
 
-      return aiAnswer;
+      return factory.aiSuggestion(eventId, originalEventId);
     },
 
     /**

src/openai/instructions/cto.ts

@@ -0,0 +1,40 @@
+import { EventAddons, EventData } from '@hawk.so/types';
+import { generateText } from 'ai';
+import { openai } from '@ai-sdk/openai';
+import { eventSolvingInput } from './inputs/eventSolving';
+import { ctoInstruction } from './instructions/cto';
+
+/**
+ * Vercel AI API
+ */
+class VercelAIApi {
+    /**
+     * Model ID
+     */
+    private readonly modelId: string;
+
+    constructor() {
+      /**
+       * @todo make it dynamic, get from project settings
+       */
+      this.modelId = 'gpt-4o';
+    }
+
+    /**
+     * Generate AI suggestion for the event
+     *
+     * @param {EventData<EventAddons>} payload - event data
+     * @returns {Promise<string>} AI suggestion for the event
+     */
+    public async generateSuggestion(payload: EventData<EventAddons>) {
+      const { text } = await generateText({
+        model: openai(this.modelId),
+        system: ctoInstruction,
+        prompt: eventSolvingInput(payload),
+      });
+
+      return text;
+    }
+}
+
+export const vercelAIApi = new VercelAIApi();

src/openai/types.ts

@@ -1,4 +1,4 @@
-import { EventData, EventAddons } from "@hawk.so/types"
+import { EventData, EventAddons } from '@hawk.so/types';
 
 export const eventSolvingInput = (payload: EventData<EventAddons>) => `
 
@@ -12,4 +12,4 @@ Response:
   "solution": "...",
   "explanation": "..."
 }
-`
+`;

src/resolvers/event.js

@@ -0,0 +1 @@
+export const ctoInstruction = 'Ты технический директор ИТ компании, тебе нужно пояснить ошибку и предложить решение';

src/resolvers/vercel-ai/index.ts

@@ -263,9 +263,9 @@ type Event {
   repetitionsPortion(cursor: String = null, limit: Int = 10): RepetitionsPortion!
 
   """
-  AI solution for the event
+  AI suggestion for the event
   """
-  askAi: String!
+  aiSuggestion: String!
 
   """
   Array of users who visited event