highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-04-03 09:28:25 +02:00
Code Issues Packages Projects Releases Wiki Activity

Custom LLM provider fix

Browse Source
This commit is contained in:
Weves 2024-05-07 17:09:09 -07:00 committed by Chris Weaver
parent 76a5f26fe1
commit 8cbf7c8097
2 changed files with 25 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
web/src/app/admin/models/llm/ConfiguredLLMProviderDisplay.tsx
View File

@ -15,7 +15,7 @@ function LLMProviderUpdateModal({
shouldMarkAsDefault,
setPopup,
}: {
llmProviderDescriptor: WellKnownLLMProviderDescriptor | null;
llmProviderDescriptor: WellKnownLLMProviderDescriptor | null | undefined;
onClose: () => void;
existingLlmProvider?: FullLLMProvider;
shouldMarkAsDefault?: boolean;
@ -55,7 +55,7 @@ function LLMProviderDisplay({
existingLlmProvider,
shouldMarkAsDefault,
}: {
llmProviderDescriptor: WellKnownLLMProviderDescriptor | null;
llmProviderDescriptor: WellKnownLLMProviderDescriptor | null | undefined;
existingLlmProvider: FullLLMProvider;
shouldMarkAsDefault?: boolean;
}) {
@ -143,8 +143,10 @@ function LLMProviderDisplay({
export function ConfiguredLLMProviderDisplay({
existingLlmProviders,
llmProviderDescriptors,
}: {
existingLlmProviders: FullLLMProvider[];
llmProviderDescriptors: WellKnownLLMProviderDescriptor[];
}) {
existingLlmProviders = existingLlmProviders.sort((a, b) => {
if (a.is_default_provider && !b.is_default_provider) {
@ -158,13 +160,26 @@ export function ConfiguredLLMProviderDisplay({
return (
<div className="gap-y-4 flex flex-col">
{existingLlmProviders.map((provider) => (
<LLMProviderDisplay
key={provider.id}
llmProviderDescriptor={null}
existingLlmProvider={provider}
/>
))}
{existingLlmProviders.map((provider) => {
const defaultProviderDesciptor = llmProviderDescriptors.find(
(llmProviderDescriptors) =>
llmProviderDescriptors.name === provider.provider
);
console.log(provider.model_names.length);
return (
<LLMProviderDisplay
key={provider.id}
// if the user has specified custom model names,
// then the provider is custom - don't use the default
// provider descriptor
llmProviderDescriptor={
provider.model_names.length > 0 ? null : defaultProviderDesciptor
}
existingLlmProvider={provider}
/>
);
})}
</div>
);
}

5
web/src/app/admin/models/llm/LLMConfiguration.tsx
View File

@ -157,6 +157,7 @@ export function LLMConfiguration() {
</Text>
<ConfiguredLLMProviderDisplay
existingLlmProviders={existingLlmProviders}
llmProviderDescriptors={llmProviderDescriptors}
/>
</>
) : (
@ -173,10 +174,6 @@ export function LLMConfiguration() {
<div className="gap-y-4 flex flex-col">
{llmProviderDescriptors.map((llmProviderDescriptor) => {
const existingLlmProvider = existingLlmProviders.find(
(llmProvider) => llmProvider.name === llmProviderDescriptor.name
);
return (
<DefaultLLMProviderDisplay
key={llmProviderDescriptor.name}