浏览代码

feat: gpt-* model support added

Timothy J. Baek 1 年之前
父节点
当前提交
0801ee7dcc
共有 2 个文件被更改,包括 9 次插入7 次删除
  1. 1 1
      src/lib/components/chat/SettingsModal.svelte
  2. 8 6
      src/routes/+page.svelte

+ 1 - 1
src/lib/components/chat/SettingsModal.svelte

@@ -334,7 +334,7 @@
 								</div>
 							</div>
 							<div class="mt-2 text-xs text-gray-500">
-								Adds optional support for 'gpt-3.5-turbo'.
+								Adds optional support for 'gpt-*' models available.
 							</div>
 						</div>
 

+ 8 - 6
src/routes/+page.svelte

@@ -365,7 +365,7 @@
 
 		if (settings.OPENAI_API_KEY) {
 			// Validate OPENAI_API_KEY
-			const openaiModels = await fetch(`https://api.openai.com/v1/models`, {
+			const openaiModelRes = await fetch(`https://api.openai.com/v1/models`, {
 				method: 'GET',
 				headers: {
 					'Content-Type': 'application/json',
@@ -381,14 +381,16 @@
 					toast.error(`OpenAI: ${error.error.message}`);
 					return null;
 				});
-
-			console.log(openaiModels);
+			const openaiModels = openaiModelRes.data;
 
 			if (openaiModels) {
 				models = [
 					...(res?.models ?? []),
 					{ name: 'hr' },
-					{ name: 'gpt-3.5-turbo', label: '(OpenAI)' }
+
+					...openaiModels
+						.map((model) => ({ name: model.id, label: 'OpenAI' }))
+						.filter((model) => model.name.includes('gpt'))
 				];
 			} else {
 				models = res?.models ?? [];
@@ -401,7 +403,7 @@
 	};
 
 	const sendPrompt = async (userPrompt) => {
-		if (selectedModel === 'gpt-3.5-turbo') {
+		if (selectedModel.includes('gpt-')) {
 			await sendPromptOpenAI(userPrompt);
 		} else {
 			await sendPromptOllama(userPrompt);
@@ -532,7 +534,7 @@
 						Authorization: `Bearer ${settings.OPENAI_API_KEY}`
 					},
 					body: JSON.stringify({
-						model: 'gpt-3.5-turbo',
+						model: selectedModel,
 						stream: true,
 						messages: messages.map((message) => ({ ...message, done: undefined }))
 					})