Fix model names for enabled LLM providers

This commit is contained in:
Weves
2024-05-14 13:13:25 -07:00
committed by Chris Weaver
parent e573ba80b9
commit d0850a0288
3 changed files with 5 additions and 10 deletions

View File

@@ -87,7 +87,7 @@ class FullLLMProvider(LLMProvider):
is_default_provider=llm_provider_model.is_default_provider,
model_names=(
llm_provider_model.model_names
or fetch_models_for_provider(llm_provider_model.name)
or fetch_models_for_provider(llm_provider_model.provider)
or [llm_provider_model.default_model_name]
),
)

View File

@@ -7,6 +7,7 @@ import { useState } from "react";
import { LLM_PROVIDERS_ADMIN_URL } from "./constants";
import { mutate } from "swr";
import { Badge, Button } from "@tremor/react";
import isEqual from "lodash/isEqual";
function LLMProviderUpdateModal({
llmProviderDescriptor,
@@ -165,7 +166,6 @@ export function ConfiguredLLMProviderDisplay({
(llmProviderDescriptors) =>
llmProviderDescriptors.name === provider.provider
);
console.log(provider.model_names.length);
return (
<LLMProviderDisplay
@@ -174,7 +174,9 @@ export function ConfiguredLLMProviderDisplay({
// then the provider is custom - don't use the default
// provider descriptor
llmProviderDescriptor={
provider.model_names.length > 0 ? null : defaultProviderDesciptor
isEqual(provider.model_names, defaultProviderDesciptor?.llm_names)
? defaultProviderDesciptor
: null
}
existingLlmProvider={provider}
/>

View File

@@ -136,13 +136,6 @@ export function LLMConfiguration() {
return <ThreeDotsLoader />;
}
const wellKnownLLMProviderNames = llmProviderDescriptors.map(
(llmProviderDescriptor) => llmProviderDescriptor.name
);
const customLLMProviders = existingLlmProviders.filter(
(llmProvider) => !wellKnownLLMProviderNames.includes(llmProvider.name)
);
return (
<>
<Title className="mb-2">Enabled LLM Providers</Title>