File size: 2,459 Bytes
d9ce58c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
b4fbe2e
 
 
 
 
d9ce58c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
import React, { createContext, useContext, useState, useEffect } from "react";

const ServiceStatusContext = createContext();

export function ServiceStatusProvider({ children }) {
  const [services, setServices] = useState({
    mistral: { status: "loading", latency: null },
    flux: { status: "loading", latency: null },
  });

  const areServicesHealthy = () => {
    return Object.values(services).every(
      (service) => service.status === "healthy"
    );
  };

  const fetchServiceHealth = async (service) => {
    console.log(`Checking health for ${service} service...`);
    try {
      const response = await fetch(`/api/health/${service}`, {
        method: "GET",
        headers: {
          Accept: "application/json",
        },
      });
      console.log(`Response status for ${service}:`, response.status);

      const data = await response.json();
      console.log(`Health data for ${service}:`, data);

      if (response.ok) {
        setServices((prev) => ({
          ...prev,
          [service]: {
            status: data.status,
            latency: data.latency,
            error: data.error,
          },
        }));
      } else {
        const errorData = data?.detail || data;
        console.error(`Error checking ${service} health:`, errorData);
        setServices((prev) => ({
          ...prev,
          [service]: {
            status: "unhealthy",
            latency: null,
            error: errorData?.error || "Service unavailable",
          },
        }));
      }
    } catch (error) {
      console.error(`Failed to check ${service} health:`, error);
      setServices((prev) => ({
        ...prev,
        [service]: {
          status: "unhealthy",
          latency: null,
          error: "Connection error",
        },
      }));
    }
  };

  useEffect(() => {
    console.log("ServiceStatusProvider mounted, checking services health...");
    // Un seul check au montage du composant
    fetchServiceHealth("mistral");
    fetchServiceHealth("flux");
  }, []); // Pas d'intervalle, juste une vérification au montage

  return (
    <ServiceStatusContext.Provider value={{ services, areServicesHealthy }}>
      {children}
    </ServiceStatusContext.Provider>
  );
}

export function useServiceStatus() {
  const context = useContext(ServiceStatusContext);
  if (!context) {
    throw new Error(
      "useServiceStatus must be used within a ServiceStatusProvider"
    );
  }
  return context;
}