tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-20 06:19:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

workign run test connection many times

Browse Source
This commit is contained in:
Ishaan Jaff
2025-03-14 17:27:30 -07:00
parent 9a01602979
commit 999ffd1bdb
2 changed files with 34 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ui/litellm-dashboard/src/components/add_model/ConnectionErrorDisplay.tsx
+10 -1
View File
@@ -12,6 +12,7 @@ interface ConnectionErrorDisplayProps {
testMode: string;
modelName?: string;
onClose?: () => void;
onTestComplete?: () => void;
}
const ConnectionErrorDisplay: React.FC<ConnectionErrorDisplayProps> = ({
@@ -19,7 +20,8 @@ const ConnectionErrorDisplay: React.FC<ConnectionErrorDisplayProps> = ({
accessToken,
testMode,
modelName = "this model",
onClose
onClose,
onTestComplete
}) => {
const [error, setError] = React.useState<Error | string | null>(null);
const [rawRequest, setRawRequest] = React.useState<any>(null);
@@ -30,6 +32,12 @@ const ConnectionErrorDisplay: React.FC<ConnectionErrorDisplayProps> = ({
const testModelConnection = async () => {
setIsLoading(true);
setShowDetails(false);
setError(null);
setRawRequest(null);
setRawResponse(null);
setIsSuccess(false);
try {
const result = await prepareModelAddRequest(formValues, accessToken, null);
if (!result) throw new Error("Failed to prepare model data");
@@ -55,6 +63,7 @@ const ConnectionErrorDisplay: React.FC<ConnectionErrorDisplayProps> = ({
setIsSuccess(false);
} finally {
setIsLoading(false);
if (onTestComplete) onTestComplete();
}
};
ui/litellm-dashboard/src/components/add_model/add_model_tab.tsx
+24 -13
View File
@@ -45,16 +45,21 @@ const AddModelTab: React.FC<AddModelTabProps> = ({
credentials,
accessToken,
}) => {
// Add state for test mode and connection error
// State for test mode and connection testing
const [testMode, setTestMode] = useState<string>("chat");
const [isResultModalVisible, setIsResultModalVisible] = useState<boolean>(false);
const [isTestingConnection, setIsTestingConnection] = useState<boolean>(false);
// Using a unique ID to force the ConnectionErrorDisplay to remount and run a fresh test
const [connectionTestId, setConnectionTestId] = useState<string>("");
// Test connection directly when button is clicked
// Test connection when button is clicked
const handleTestConnection = async () => {
setIsTestingConnection(true);
// Generate a new test ID (using timestamp for uniqueness)
// This forces React to create a new instance of ConnectionErrorDisplay
setConnectionTestId(`test-${Date.now()}`);
// Show the modal with the fresh test
setIsResultModalVisible(true);
// The actual testing is handled in ConnectionErrorDisplay component
};
return (
@@ -241,16 +246,22 @@ const AddModelTab: React.FC<AddModelTabProps> = ({
]}
width={700}
>
<ConnectionErrorDisplay
formValues={form.getFieldsValue()}
accessToken={accessToken}
testMode={testMode}
modelName={form.getFieldValue('model_name') || form.getFieldValue('model')}
onClose={() => {
setIsResultModalVisible(false);
setIsTestingConnection(false);
}}
/>
{/* Only render the ConnectionErrorDisplay when modal is visible and we have a test ID */}
{isResultModalVisible && (
<ConnectionErrorDisplay
// The key prop tells React to create a fresh component instance when it changes
key={connectionTestId}
formValues={form.getFieldsValue()}
accessToken={accessToken}
testMode={testMode}
modelName={form.getFieldValue('model_name') || form.getFieldValue('model')}
onClose={() => {
setIsResultModalVisible(false);
setIsTestingConnection(false);
}}
onTestComplete={() => setIsTestingConnection(false)}
/>
)}
</Modal>
</>
);