use_fetch_agentframework_traces.ts

/*
 * Copyright OpenSearch Contributors
 * SPDX-License-Identifier: Apache-2.0
 */

import { useEffect, useReducer } from 'react';
import { ASSISTANT_API } from '../../common/constants/llm';
import { AgentFrameworkTrace } from '../../common/utils/llm_chat/traces';
import { useCore } from '../contexts/core_context';
import { GenericReducer, genericReducer } from './fetch_reducer';

export const useFetchAgentFrameworkTraces = (interactionId: string) => {
  const core = useCore();
  const reducer: GenericReducer<AgentFrameworkTrace[]> = genericReducer;
  const [state, dispatch] = useReducer(reducer, { loading: false });

  useEffect(() => {
    const abortController = new AbortController();
    dispatch({ type: 'request' });
    if (!interactionId) {
      dispatch({ type: 'success', payload: undefined });
      return;
    }

    core.services.http
      .get<AgentFrameworkTrace[]>(`${ASSISTANT_API.TRACE}/${interactionId}`, {
        signal: abortController.signal,
        query: core.services.dataSource.getDataSourceQuery(),
      })
      .then((payload) =>
        dispatch({
          type: 'success',
          payload,
        })
      )
      .catch((error) => {
        if (error.name === 'AbortError') return;
        dispatch({ type: 'failure', error });
      });

    return () => abortController.abort();
  }, [core.services.http, interactionId, core.services.dataSource]);

  return { ...state };
};