Feature: Search in repetition delta

package.json

@@ -1,6 +1,6 @@
 {
   "name": "hawk.api",
-  "version": "1.2.23",
+  "version": "1.2.24",
   "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

@@ -292,6 +292,12 @@ class EventsFactory extends Factory {
     const searchFilter = search.trim().length > 0
       ? {
         $or: [
+          {
+            'repetition.delta': {
+              $regex: escapedSearch,
+              $options: 'i',
+            },
+          },
           {
             'event.payload.title': {
               $regex: escapedSearch,
@@ -443,25 +449,50 @@ class EventsFactory extends Factory {
     const days = Math.ceil((end - start) / (24 * 60 * 60 * 1000));
 
     try {
-      const redisData = await this.chartDataService.getProjectChartData(
-        projectId,
-        startDate,
-        endDate,
-        groupBy,
-        timezoneOffset
-      );
-
-      if (redisData && redisData.length > 0) {
-        return redisData;
-      }
-
-      // Fallback to Mongo (empty groupHash for project-level data)
-      return this.findChartData(days, timezoneOffset, '');
+      const [acceptedSeries, rateLimitedSeries] = await Promise.all([
+        this.chartDataService.getProjectChartData(
+          projectId,
+          startDate,
+          endDate,
+          groupBy,
+          timezoneOffset,
+          'events-accepted'
+        ),
+        this.chartDataService.getProjectChartData(
+          projectId,
+          startDate,
+          endDate,
+          groupBy,
+          timezoneOffset,
+          'events-rate-limited'
+        ),
+      ]);
+
+      return [
+        {
+          label: 'accepted',
+          data: acceptedSeries,
+        },
+        {
+          label: 'rate-limited',
+          data: rateLimitedSeries,
+        },
+      ];
     } catch (err) {
       console.error('[EventsFactory] getProjectChartData error:', err);
 
-      // Fallback to Mongo on error (empty groupHash for project-level data)
-      return this.findChartData(days, timezoneOffset, '');
+      const fallbackAccepted = await this.findChartData(days, timezoneOffset, '');
+
+      return [
+        {
+          label: 'accepted',
+          data: fallbackAccepted,
+        },
+        {
+          label: 'rate-limited',
+          data: this._composeZeroSeries(fallbackAccepted),
+        },
+      ];
     }
   }
 
@@ -474,7 +505,14 @@ class EventsFactory extends Factory {
    * @returns {Promise<Array>}
    */
   async getEventDailyChart(groupHash, days, timezoneOffset = 0) {
-    return this.findChartData(days, timezoneOffset, groupHash);
+    const data = await this.findChartData(days, timezoneOffset, groupHash);
+
+    return [
+      {
+        label: 'accepted',
+        data,
+      },
+    ];
   }
 
   /**
@@ -568,6 +606,23 @@ class EventsFactory extends Factory {
     return result;
   }
 
+  /**
+   * Compose zero-filled chart series using timestamps from the provided template
+   *
+   * @param {Array<{timestamp: number, count: number}>} template - reference series for timestamps
+   * @returns {Array<{timestamp: number, count: number}>}
+   */
+  _composeZeroSeries(template = []) {
+    if (!Array.isArray(template) || template.length === 0) {
+      return [];
+    }
+
+    return template.map((point) => ({
+      timestamp: point.timestamp,
+      count: 0,
+    }));
+  }
+
   /**
    * Returns number of documents that occurred after the last visit time
    *
@@ -686,7 +741,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/redisHelper.ts

@@ -36,6 +36,7 @@ export default class RedisHelper {
   constructor() {
     if (!process.env.REDIS_URL) {
       console.warn('[Redis] REDIS_URL not set, Redis features will be disabled');
+
       return;
     }
 
@@ -49,7 +50,9 @@ export default class RedisHelper {
              * Max wait time: 30 seconds
              */
             const delay = Math.min(retries * 1000, 30000);
+
             console.log(`[Redis] Reconnecting... attempt ${retries}, waiting ${delay}ms`);
+
             return delay;
           },
         },
@@ -93,6 +96,7 @@ export default class RedisHelper {
     if (!RedisHelper.instance) {
       RedisHelper.instance = new RedisHelper();
     }
+
     return RedisHelper.instance;
   }
 
@@ -102,6 +106,7 @@ export default class RedisHelper {
   public async initialize(): Promise<void> {
     if (!this.redisClient) {
       console.warn('[Redis] Client not initialized, skipping connection');
+
       return;
     }
 

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/chartDataService.ts

@@ -19,6 +19,7 @@ export default class ChartDataService {
    * @param endDate - end date as ISO string (e.g., '2025-01-31T23:59:59Z')
    * @param groupBy - grouping interval in minutes (1=minute, 60=hour, 1440=day)
    * @param timezoneOffset - user's local timezone offset in minutes (default: 0)
+   * @param metricType - Redis metric type suffix (e.g., 'events-accepted', 'events-rate-limited')
    * @returns Array of data points with timestamp and count
    * @throws Error if Redis is not connected (caller should fallback to MongoDB)
    */
@@ -27,7 +28,8 @@ export default class ChartDataService {
     startDate: string,
     endDate: string,
     groupBy: number,
-    timezoneOffset = 0
+    timezoneOffset = 0,
+    metricType = 'events-accepted'
   ): Promise<{ timestamp: number; count: number }[]> {
     // Check if Redis is connected
     if (!this.redisHelper.isConnected()) {
@@ -37,7 +39,7 @@ export default class ChartDataService {
 
     // Determine granularity and compose key
     const granularity = getTimeSeriesSuffix(groupBy);
-    const key = composeProjectMetricsKey(granularity, projectId);
+    const key = composeProjectMetricsKey(granularity, projectId, metricType);
 
     // Parse ISO date strings to milliseconds
     const start = new Date(startDate).getTime();
@@ -46,6 +48,7 @@ export default class ChartDataService {
 
     // Fetch data from Redis
     let result: TsRangeResult[] = [];
+
     try {
       result = await this.redisHelper.tsRange(
         key,
@@ -65,8 +68,10 @@ export default class ChartDataService {
 
     // Transform data from Redis
     const dataPoints: { [ts: number]: number } = {};
+
     for (const [tsStr, valStr] of result) {
       const tsMs = Number(tsStr);
+
       dataPoints[tsMs] = Number(valStr) || 0;
     }
 
@@ -79,6 +84,7 @@ export default class ChartDataService {
 
     while (current <= end) {
       const count = dataPoints[current] || 0;
+
       filled.push({
         timestamp: Math.floor((current + timezoneOffset * 60 * 1000) / 1000),
         count,

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>;
+}