| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
|
|
| import { useCallback, useEffect, useState } from 'react'; |
| import { API, toBoolean } from '../../helpers'; |
|
|
| export const useModelDeploymentSettings = () => { |
| const [loading, setLoading] = useState(true); |
| const [settings, setSettings] = useState({ |
| 'model_deployment.ionet.enabled': false, |
| 'model_deployment.ionet.api_key': '', |
| }); |
| const [connectionState, setConnectionState] = useState({ |
| loading: false, |
| ok: null, |
| error: null, |
| }); |
|
|
| const getSettings = async () => { |
| try { |
| setLoading(true); |
| const res = await API.get('/api/option/'); |
| const { success, data } = res.data; |
| |
| if (success) { |
| const newSettings = { |
| 'model_deployment.ionet.enabled': false, |
| 'model_deployment.ionet.api_key': '', |
| }; |
| |
| data.forEach((item) => { |
| if (item.key.endsWith('enabled')) { |
| newSettings[item.key] = toBoolean(item.value); |
| } else if (newSettings.hasOwnProperty(item.key)) { |
| newSettings[item.key] = item.value || ''; |
| } |
| }); |
| |
| setSettings(newSettings); |
| } |
| } catch (error) { |
| console.error('Failed to get model deployment settings:', error); |
| } finally { |
| setLoading(false); |
| } |
| }; |
|
|
| useEffect(() => { |
| getSettings(); |
| }, []); |
|
|
| const apiKey = settings['model_deployment.ionet.api_key']; |
| const isIoNetEnabled = settings['model_deployment.ionet.enabled'] && |
| apiKey && |
| apiKey.trim() !== ''; |
|
|
| const buildConnectionError = (rawMessage, fallbackMessage = 'Connection failed') => { |
| const message = (rawMessage || fallbackMessage).trim(); |
| const normalized = message.toLowerCase(); |
| if (normalized.includes('expired') || normalized.includes('expire')) { |
| return { type: 'expired', message }; |
| } |
| if (normalized.includes('invalid') || normalized.includes('unauthorized') || normalized.includes('api key')) { |
| return { type: 'invalid', message }; |
| } |
| if (normalized.includes('network') || normalized.includes('timeout')) { |
| return { type: 'network', message }; |
| } |
| return { type: 'unknown', message }; |
| }; |
|
|
| const testConnection = useCallback(async (apiKey) => { |
| const key = (apiKey || '').trim(); |
| if (key === '') { |
| setConnectionState({ loading: false, ok: null, error: null }); |
| return; |
| } |
|
|
| setConnectionState({ loading: true, ok: null, error: null }); |
| try { |
| const response = await API.post( |
| '/api/deployments/test-connection', |
| { api_key: key }, |
| { skipErrorHandler: true }, |
| ); |
|
|
| if (response?.data?.success) { |
| setConnectionState({ loading: false, ok: true, error: null }); |
| return; |
| } |
|
|
| const message = response?.data?.message || 'Connection failed'; |
| setConnectionState({ loading: false, ok: false, error: buildConnectionError(message) }); |
| } catch (error) { |
| if (error?.code === 'ERR_NETWORK') { |
| setConnectionState({ |
| loading: false, |
| ok: false, |
| error: { type: 'network', message: 'Network connection failed' }, |
| }); |
| return; |
| } |
| const rawMessage = error?.response?.data?.message || error?.message || 'Unknown error'; |
| setConnectionState({ loading: false, ok: false, error: buildConnectionError(rawMessage, 'Connection failed') }); |
| } |
| }, []); |
|
|
| useEffect(() => { |
| if (!loading && isIoNetEnabled) { |
| testConnection(apiKey); |
| return; |
| } |
| setConnectionState({ loading: false, ok: null, error: null }); |
| }, [loading, isIoNetEnabled, apiKey, testConnection]); |
|
|
| return { |
| loading, |
| settings, |
| apiKey, |
| isIoNetEnabled, |
| refresh: getSettings, |
| connectionLoading: connectionState.loading, |
| connectionOk: connectionState.ok, |
| connectionError: connectionState.error, |
| testConnection, |
| }; |
| }; |
|
|
