openreplay/frontend/app/mstore/aiSummaryStore.ts

import { makeAutoObservable } from 'mobx';

import { aiService } from 'App/services';

export default class AiSummaryStore {
  text = '';
  toggleSummary = false;
  isLoading = false;

  constructor() {
    makeAutoObservable(this);
  }

  setText(text: string) {
    this.text = text;
  }

  setToggleSummary(toggleSummary: boolean) {
    this.toggleSummary = toggleSummary;
  }

  setLoading(loading: boolean) {
    this.isLoading = loading;
  }

  getSummary = async (sessionId: string) => {
    if (this.isLoading) return;

    this.setLoading(true);
    this.setText('');
    try {
      const respText = await aiService.getSummary(sessionId);
      if (!respText) return;

      this.setText(respText);
    } catch (e) {
      console.error(e);
    } finally {
      this.setLoading(false);
    }
  };

  getDetailedSummary = async (sessionId: string, networkEvents: any[], feat: 'errors' | 'issues' | 'journey', startTs: number, endTs: number) => {
    if (this.isLoading) return;

    this.setLoading(true);
    this.setText('');
    try {
      const respText = await aiService.getDetailedSummary(sessionId, networkEvents,feat, startTs, endTs);
      if (!respText) return;

      this.setText(respText);
    } catch (e) {
      console.error(e);
    } finally {
      this.setLoading(false);
    }
  }
}