Anthropic: show model refresh button when missing key

enricoros · Nov 6, 2023 · a9d97b9 · a9d97b9 · vercel · Nov 6, 2023
1 parent 87f0bf1
commit a9d97b9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/modules/llms/transports/server/anthropic.router.ts b/src/modules/llms/transports/server/anthropic.router.ts
@@ -144,8 +144,11 @@ export function anthropicAccess(access: AnthropicAccessSchema, apiPath: string):
 
   // API key
   const anthropicKey = access.anthropicKey || process.env.ANTHROPIC_API_KEY || '';
+
+  // break for the missing key only on the default host
   if (!anthropicKey)
-    throw new Error('Missing Anthropic API Key. Add it on the UI (Models Setup) or server side (your deployment).');
+    if (!access.anthropicHost && !process.env.ANTHROPIC_API_HOST)
+      throw new Error('Missing Anthropic API Key. Add it on the UI (Models Setup) or server side (your deployment).');
 
   // API host
   let anthropicHost = fixupHost(access.anthropicHost || process.env.ANTHROPIC_API_HOST || DEFAULT_ANTHROPIC_HOST, apiPath);

diff --git a/src/modules/llms/vendors/anthropic/AnthropicSourceSetup.tsx b/src/modules/llms/vendors/anthropic/AnthropicSourceSetup.tsx
@@ -32,7 +32,7 @@ export function AnthropicSourceSetup(props: { sourceId: DModelSourceId }) {
   const needsUserKey = !ModelVendorAnthropic.hasServerKey;
   const keyValid = isValidAnthropicApiKey(anthropicKey);
   const keyError = (/*needsUserKey ||*/ !!anthropicKey) && !keyValid;
-  const shallFetchSucceed = anthropicKey ? keyValid : !needsUserKey;
+  const shallFetchSucceed = anthropicKey ? keyValid : (!needsUserKey || !!anthropicHost);
 
   // fetch models
   const { isFetching, refetch, isError, error } = apiQuery.llmAnthropic.listModels.useQuery({