Merge pull request #590 from codex-team/feat/openai

feat(ai): added ai integration

Author: slaveeks

package.json

@@ -1,6 +1,6 @@
 {
   "name": "hawk.api",
-  "version": "1.2.25",
+  "version": "1.2.26",
   "main": "index.ts",
   "license": "BUSL-1.1",
   "scripts": {
@@ -34,6 +34,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",
@@ -53,9 +54,9 @@
     "@types/mongodb": "^3.6.20",
     "@types/morgan": "^1.9.10",
     "@types/node": "^16.11.46",
-    "@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 +87,7 @@
     "redis": "^4.7.0",
     "safe-regex": "^2.1.0",
     "ts-node-dev": "^2.0.0",
-    "uuid": "^8.3.2"
+    "uuid": "^8.3.2",
+    "zod": "^3.25.76"
   }
 }

src/integrations/vercel-ai/index.ts

@@ -0,0 +1,41 @@
+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';
+
+/**
+ * Interface for interacting with Vercel AI Gateway
+ */
+class VercelAIApi {
+    /**
+     * Model ID to use for generating suggestions
+     */
+    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
+     * @todo add defence against invalid prompt injection
+     */
+    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/integrations/vercel-ai/inputs/eventSolving.ts

@@ -0,0 +1,5 @@
+import { EventData, EventAddons } from '@hawk.so/types';
+
+export const eventSolvingInput = (payload: EventData<EventAddons>) => `
+Payload: ${JSON.stringify(payload)}
+`;

src/integrations/vercel-ai/instructions/cto.ts

@@ -0,0 +1,9 @@
+export const ctoInstruction = `Ты технический директор ИТ компании, тебе нужно пояснить ошибку и предложить решение.
+
+Предоставь ответ в следующем формате:
+
+1. Описание проблемы
+2. Решение проблемы
+3. Описание того, как можно предотвратить подобную ошибку в будущем
+
+Ответь на русском языке.`;

src/models/eventsFactory.js

@@ -691,7 +691,7 @@ class EventsFactory extends Factory {
     /**
      * If originalEventId equals repetitionId than user wants to get first repetition which is original event
      */
-    if (repetitionId === originalEventId) {
+    if (repetitionId.toString() === originalEventId.toString()) {
       const originalEvent = await this.eventsDataLoader.load(originalEventId);
 
       /**

src/resolvers/event.js

@@ -1,5 +1,6 @@
 const getEventsFactory = require('./helpers/eventsFactory').default;
 const sendPersonalNotification = require('../utils/personalNotifications').default;
+const { aiService } = require('../services/ai');
 
 /**
  * See all types and fields here {@see ../typeDefs/event.graphql}
@@ -89,6 +90,20 @@ module.exports = {
       return factory.getEventDailyChart(groupHash, days, timezoneOffset);
     },
 
+    /**
+     * 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);
+
+      return aiService.generateSuggestion(factory, eventId, originalEventId);
+    },
+
     /**
      * Return release data for the event
      *

src/services/ai.ts

@@ -0,0 +1,27 @@
+import { vercelAIApi } from '../integrations/vercel-ai/';
+import { EventsFactoryInterface } from './types';
+
+/**
+ * Service for interacting with AI
+ */
+export class AIService {
+  /**
+   * Generate suggestion for the event
+   *
+   * @param eventsFactory - events factory
+   * @param eventId - event id
+   * @param originalEventId - original event id
+   * @returns {Promise<string>} - suggestion
+   */
+  public async generateSuggestion(eventsFactory: EventsFactoryInterface, eventId: string, originalEventId: string): Promise<string> {
+    const event = await eventsFactory.getEventRepetition(eventId, originalEventId);
+
+    if (!event) {
+      throw new Error('Event not found');
+    }
+
+    return vercelAIApi.generateSuggestion(event.payload);
+  }
+}
+
+export const aiService = new AIService();

src/services/types.ts

@@ -0,0 +1,23 @@
+import { EventAddons, EventData } from '@hawk.so/types';
+
+/**
+ * Event type which is returned by events factory
+ */
+type Event = {
+    _id: string;
+    payload: EventData<EventAddons>;
+};
+
+/**
+ * Interface for interacting with events factory
+ */
+export interface EventsFactoryInterface {
+  /**
+   * Get event repetition
+   *
+   * @param repetitionId - repetition id
+   * @param originalEventId - original event id
+   * @returns {Promise<EventData<EventAddons>>} - event repetition
+   */
+  getEventRepetition(repetitionId: string, originalEventId: string): Promise<Event>;
+}

src/typeDefs/event.ts

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

src/types/vercel-ai.d.ts

@@ -0,0 +1,18 @@
+declare module 'ai' {
+    /**
+     * Minimal type for generateText used in server-side integration.
+     */
+    export function generateText(input: {
+        model: any;
+        system?: string;
+        prompt: string;
+    }): Promise<{ text: string }>;
+}
+
+declare module '@ai-sdk/openai' {
+    /**
+     * Minimal types for OpenAI provider.
+     */
+    export function createOpenAI(config?: { apiKey?: string }): (model: string) => any;
+    export const openai: (model: string) => any;
+}