added ai service, some refactorng

Author: slaveeks

src/vercel-ai/index.ts src/integrations/vercel-ai/index.tssrc/vercel-ai/index.ts renamed to src/integrations/vercel-ai/index.ts

@@ -7,7 +7,6 @@ const mongo = require('../mongo');
 const Event = require('../models/event');
 const { ObjectID } = require('mongodb');
 const { composeEventPayloadByRepetition } = require('../utils/merge');
-const { vercelAIApi } = require('../vercel-ai');
 
 const MAX_DB_READ_BATCH_SIZE = Number(process.env.MAX_DB_READ_BATCH_SIZE);
 
@@ -179,27 +178,6 @@ class EventsFactory extends Factory {
     return new Event(searchResult);
   }
 
-  /**
-   * Generate 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(eventId, originalEventId) {
-    const repetition = await this.getEventRepetition(eventId, originalEventId);
-
-    if (!repetition) {
-      throw new Error('Repetition not found');
-    }
-
-    const payload = repetition.payload;
-    const solution = await vercelAIApi.generateSuggestion(payload);
-
-    return solution;
-  }
-
   /**
    * Returns events that grouped by day
    *

src/vercel-ai/inputs/eventSolving.ts …rations/vercel-ai/inputs/eventSolving.tssrc/vercel-ai/inputs/eventSolving.ts renamed to src/integrations/vercel-ai/inputs/eventSolving.ts src/vercel-ai/inputs/eventSolving.ts renamed to src/integrations/vercel-ai/inputs/eventSolving.ts

@@ -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}
@@ -100,7 +101,7 @@ module.exports = {
     async aiSuggestion({ projectId, _id: eventId, originalEventId }, _args, context) {
       const factory = getEventsFactory(context, projectId);
 
-      return factory.aiSuggestion(eventId, originalEventId);
+      return aiService.generateSuggestion(factory, eventId, originalEventId);
     },
 
     /**

src/vercel-ai/instructions/cto.ts …tegrations/vercel-ai/instructions/cto.tssrc/vercel-ai/instructions/cto.ts renamed to src/integrations/vercel-ai/instructions/cto.ts src/vercel-ai/instructions/cto.ts renamed to src/integrations/vercel-ai/instructions/cto.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/models/eventsFactory.js

@@ -0,0 +1,20 @@
+import { EventAddons, EventData } from '@hawk.so/types';
+
+type Event = {
+    _id: string;
+    payload: EventData<EventAddons>;
+};
+
+/**
+ * Interface for 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>;
+}