DB-GPT/web/components/models_evaluation/context/EvaluationContext.tsx

import React, { createContext, useContext } from 'react';
import { useEvaluationList } from '../hooks/useEvaluationList';

interface EvaluationContextType {
  refresh: () => void;
  data: any;
  loading: boolean;
  getModelsEvaluation: (page?: number, pageSize?: number) => void;
}

const EvaluationContext = createContext<EvaluationContextType | undefined>(undefined);

interface EvaluationProviderProps {
  children: React.ReactNode;
  filterValue?: string;
  type?: string;
}

export const EvaluationProvider: React.FC<EvaluationProviderProps> = ({ children, filterValue = '', type = 'all' }) => {
  const { data, loading, getModelsEvaluation, refresh } = useEvaluationList({
    filterValue,
    type,
  });

  return (
    <EvaluationContext.Provider
      value={{
        refresh,
        data,
        loading,
        getModelsEvaluation,
      }}
    >
      {children}
    </EvaluationContext.Provider>
  );
};

export const useEvaluation = () => {
  const context = useContext(EvaluationContext);
  if (context === undefined) {
    throw new Error('useEvaluation must be used within an EvaluationProvider');
  }
  return context;
};