init openai

Author: slaveeks

package.json

@@ -81,9 +81,11 @@
     "mime-types": "^2.1.25",
     "mongodb": "^3.7.3",
     "morgan": "^1.10.1",
+    "node-fetch": "^3.3.2",
+    "openai": "^6.8.1",
     "prom-client": "^15.1.3",
     "safe-regex": "^2.1.0",
     "ts-node-dev": "^2.0.0",
     "uuid": "^8.3.2"
   }
-}
+}

src/models/eventsFactory.js

@@ -7,6 +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 MAX_DB_READ_BATCH_SIZE = Number(process.env.MAX_DB_READ_BATCH_SIZE);
 
@@ -178,6 +179,27 @@ class EventsFactory extends Factory {
     return new Event(searchResult);
   }
 
+  /**
+   * Ask AI about solving repetition
+   *
+   * @param {string} repetitionId - repetition ID
+   * @param {string} eventId - event ID
+   *
+   * @return {string} - AI answer
+   */
+  async askAi(repetitionId, eventId) {
+    const repetition = await this.getEventRepetition(repetitionId, eventId);
+
+    if (!repetition) {
+      throw new Error('Repetition not found');
+    }
+
+    const payload = repetition.event.payload;
+    const solution = await openAIApi.solveEvent(payload);
+
+    return solution;
+  }
+
   /**
    * Returns events that grouped by day
    *

src/openai/index.ts

@@ -0,0 +1,27 @@
+import { EventAddons, EventData } from "@hawk.so/types";
+import OpenAI from "openai";
+import { eventSolvingInput } from "./inputs/eventSolving";
+import { ctoInstruction } from "./instructions/cto";
+
+class OpenAIApi {
+    private readonly client: OpenAI;
+
+    constructor() {
+        this.client = new OpenAI({
+            apiKey: process.env.OPENAI_API_KEY,
+            // fetch,
+        });
+    }
+
+    async solveEvent(payload: EventData<EventAddons>) {
+        const response = await this.client.responses.create({
+            model: "gpt-4o",
+            instructions: ctoInstruction,
+            input: eventSolvingInput(payload),
+        });
+
+        return response.output;
+    }
+}
+
+export const openAIApi = new OpenAIApi();

src/openai/inputs/eventSolving.ts

@@ -0,0 +1,15 @@
+import { EventData, EventAddons } from "@hawk.so/types"
+
+export const eventSolvingInput = (payload: EventData<EventAddons>) => `
+
+Проанализируй ошибку и предложи решение
+
+Payload: ${JSON.stringify(payload)}
+
+Response:
+
+{
+  "solution": "...",
+  "explanation": "..."
+}
+`

src/openai/instructions/cto.ts

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

src/openai/types.ts

@@ -89,6 +89,13 @@ module.exports = {
       return factory.findChartData(days, timezoneOffset, groupHash);
     },
 
+    async askAi({ projectId, id: eventId, originalEventId }, _args, context) {
+      const factory = getEventsFactory(context, projectId);
+      const aiAnswer = await factory.askAi(eventId, originalEventId);
+
+      return aiAnswer;
+    },
+
     /**
      * Return release data for the event
      *

src/resolvers/event.js

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

src/typeDefs/event.ts

@@ -2343,6 +2343,11 @@ cssstyle@^2.3.0:
   dependencies:
     cssom "~0.3.6"
 
+data-uri-to-buffer@^4.0.0:
+  version "4.0.1"
+  resolved "https://registry.yarnpkg.com/data-uri-to-buffer/-/data-uri-to-buffer-4.0.1.tgz#d8feb2b2881e6a4f58c2e08acfd0e2834e26222e"
+  integrity sha512-0R9ikRb668HB7QDxT1vkpuUBtqc53YyAwMwGeUFKRojY/NWKvdZ+9UYtRfGmhqNbRkTSVpMbmyhXipFFv2cb/A==
+
 data-urls@^2.0.0:
   version "2.0.0"
   resolved "https://registry.yarnpkg.com/data-urls/-/data-urls-2.0.0.tgz#156485a72963a970f5d5821aaf642bef2bf2db9b"
@@ -3095,6 +3100,14 @@ fd-slicer@~1.1.0:
   dependencies:
     pend "~1.2.0"
 
+fetch-blob@^3.1.2, fetch-blob@^3.1.4:
+  version "3.2.0"
+  resolved "https://registry.yarnpkg.com/fetch-blob/-/fetch-blob-3.2.0.tgz#f09b8d4bbd45adc6f0c20b7e787e793e309dcce9"
+  integrity sha512-7yAQpD2UMJzLi1Dqv7qFYnPbaPx7ZfFK6PiIxQ4PfkGPyNyl2Ugx+a/umUonmKqjhM4DnfbMvdX6otXq83soQQ==
+  dependencies:
+    node-domexception "^1.0.0"
+    web-streams-polyfill "^3.0.3"
+
 figures@^3.0.0:
   version "3.2.0"
   resolved "https://registry.yarnpkg.com/figures/-/figures-3.2.0.tgz#625c18bd293c604dc4a8ddb2febf0c88341746af"
@@ -3227,6 +3240,13 @@ form-data@^4.0.0:
     combined-stream "^1.0.8"
     mime-types "^2.1.12"
 
+formdata-polyfill@^4.0.10:
+  version "4.0.10"
+  resolved "https://registry.yarnpkg.com/formdata-polyfill/-/formdata-polyfill-4.0.10.tgz#24807c31c9d402e002ab3d8c720144ceb8848423"
+  integrity sha512-buewHzMvYL29jdeQTVILecSaZKnt/RJWjoZCF5OW60Z67/GmSLBkOFM7qh1PI3zFNtJbaZL5eQu1vLfazOwj4g==
+  dependencies:
+    fetch-blob "^3.1.2"
+
 forwarded@0.2.0:
   version "0.2.0"
   resolved "https://registry.yarnpkg.com/forwarded/-/forwarded-0.2.0.tgz#2269936428aad4c15c7ebe9779a84bf0b2a81811"
@@ -5091,6 +5111,11 @@ node-addon-api@^5.0.0:
   resolved "https://registry.yarnpkg.com/node-addon-api/-/node-addon-api-5.0.0.tgz#7d7e6f9ef89043befdb20c1989c905ebde18c501"
   integrity sha512-CvkDw2OEnme7ybCykJpVcKH+uAOLV2qLqiyla128dN9TkEWfrYmxG6C2boDe5KcNQqZF3orkqzGgOMvZ/JNekA==
 
+node-domexception@^1.0.0:
+  version "1.0.0"
+  resolved "https://registry.yarnpkg.com/node-domexception/-/node-domexception-1.0.0.tgz#6888db46a1f71c0b76b3f7555016b63fe64766e5"
+  integrity sha512-/jKZoMpw0F8GRwl4/eLROPA3cfcXtLApP0QzLmUT/HuPCZWyB7IY9ZrMeKw2O/nFIqPQB3PVM9aYm0F312AXDQ==
+
 node-fetch@^2.6.0:
   version "2.7.0"
   resolved "https://registry.yarnpkg.com/node-fetch/-/node-fetch-2.7.0.tgz#d0f0fa6e3e2dc1d27efcd8ad99d550bda94d187d"
@@ -5105,6 +5130,15 @@ node-fetch@^2.6.7:
   dependencies:
     whatwg-url "^5.0.0"
 
+node-fetch@^3.3.2:
+  version "3.3.2"
+  resolved "https://registry.yarnpkg.com/node-fetch/-/node-fetch-3.3.2.tgz#d1e889bacdf733b4ff3b2b243eb7a12866a0b78b"
+  integrity sha512-dRB78srN/l6gqWulah9SrxeYnxeddIG30+GOqK/9OlLVyLg3HPnr6SqOWTWOXKRwC2eGYCkZ59NNuSgvSrpgOA==
+  dependencies:
+    data-uri-to-buffer "^4.0.0"
+    fetch-blob "^3.1.4"
+    formdata-polyfill "^4.0.10"
+
 node-int64@^0.4.0:
   version "0.4.0"
   resolved "https://registry.yarnpkg.com/node-int64/-/node-int64-0.4.0.tgz#87a9065cdb355d3182d8f94ce11188b825c68a3b"
@@ -5313,6 +5347,11 @@ onetime@^5.1.0:
   dependencies:
     mimic-fn "^2.1.0"
 
+openai@^6.8.1:
+  version "6.8.1"
+  resolved "https://registry.yarnpkg.com/openai/-/openai-6.8.1.tgz#72610890aa6f67b3473c7be2e2d6ff25ebb9846a"
+  integrity sha512-ACifslrVgf+maMz9vqwMP4+v9qvx5Yzssydizks8n+YUJ6YwUoxj51sKRQ8HYMfR6wgKLSIlaI108ZwCk+8yig==
+
 optional-require@^1.1.8:
   version "1.1.8"
   resolved "https://registry.yarnpkg.com/optional-require/-/optional-require-1.1.8.tgz#16364d76261b75d964c482b2406cb824d8ec44b7"
@@ -6927,6 +6966,11 @@ walker@^1.0.7, walker@~1.0.5:
   dependencies:
     makeerror "1.0.12"
 
+web-streams-polyfill@^3.0.3:
+  version "3.3.3"
+  resolved "https://registry.yarnpkg.com/web-streams-polyfill/-/web-streams-polyfill-3.3.3.tgz#2073b91a2fdb1fbfbd401e7de0ac9f8214cecb4b"
+  integrity sha512-d2JWLCivmZYTSIoge9MsgFCZrt571BikcWGYkjC1khllbTeDlGqZ2D8vD8E/lJa8WGWbb7Plm8/XJYV7IJHZZw==
+
 webidl-conversions@^3.0.0:
   version "3.0.1"
   resolved "https://registry.yarnpkg.com/webidl-conversions/-/webidl-conversions-3.0.1.tgz#24534275e2a7bc6be7bc86611cc16ae0a5654871"