fix: litellm issue

f96beeee · Timothy J. Baek · 0c367412 · f96beeee · f96beeee
Commit f96beeee authored Mar 29, 2024 by Timothy J. Baek
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 2 deletions

src/routes/(app)/+page.svelte src/routes/(app)/+page.svelte +5 -1

src/routes/(app)/c/[id]/+page.svelte src/routes/(app)/c/[id]/+page.svelte +3 -1

No files found.
--- a/src/routes/(app)/+page.svelte
+++ b/src/routes/(app)/+page.svelte
@@ -534,6 +534,8 @@

 		console.log(docs);

+		console.log(model);
+
 		const res = await generateOpenAIChatCompletion(
 			localStorage.token,
 			{
@@ -586,7 +588,9 @@
 				max_tokens: $settings?.options?.num_predict ?? undefined,
 				docs: docs.length > 0 ? docs : undefined
 			},
-			model.source === 'litellm' ? `${LITELLM_API_BASE_URL}/v1` : `${OPENAI_API_BASE_URL}`
+			model.source.toLowerCase() === 'litellm'
+				? `${LITELLM_API_BASE_URL}/v1`
+				: `${OPENAI_API_BASE_URL}`
 		);

 		if (res && res.ok) {

--- a/src/routes/(app)/c/[id]/+page.svelte
+++ b/src/routes/(app)/c/[id]/+page.svelte
@@ -602,7 +602,9 @@
 				max_tokens: $settings?.options?.num_predict ?? undefined,
 				docs: docs.length > 0 ? docs : undefined
 			},
-			model.source === 'litellm' ? `${LITELLM_API_BASE_URL}/v1` : `${OPENAI_API_BASE_URL}`
+			model.source.toLowerCase() === 'litellm'
+				? `${LITELLM_API_BASE_URL}/v1`
+				: `${OPENAI_API_BASE_URL}`
 		);

 		if (res && res.ok) {