fix(OpenAI Node): Update models to only show those supported (n8n-io#…

…5805)
fl-g6 · Apr 24, 2023 · faf58d2 · faf58d2
1 parent 04345a8
commit faf58d2
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 27 deletions.
diff --git a/packages/nodes-base/nodes/OpenAi/ChatDescription.ts b/packages/nodes-base/nodes/OpenAi/ChatDescription.ts
@@ -35,39 +35,53 @@ const completeOperations: INodeProperties[] = [
 		name: 'model',
 		type: 'options',
 		description:
-			'The model to use. Currently, only gpt-3.5-turbo and gpt-3.5-turbo-0301 are supported.',
+			'The model which will generate the completion. <a href="https://beta.openai.com/docs/models/overview">Learn more</a>.',
 		displayOptions: {
 			show: {
 				operation: ['complete'],
 				resource: ['chat'],
 			},
 		},
-		options: [
-			{
-				name: 'gpt-3.5-turbo',
-				value: 'gpt-3.5-turbo',
-			},
-			{
-				name: 'gpt-3.5-turbo-0301',
-				value: 'gpt-3.5-turbo-0301',
-			},
-			{
-				name: 'gpt-4',
-				value: 'gpt-4',
-			},
-			{
-				name: 'gpt-4-0314',
-				value: 'gpt-4-0314',
-			},
-			{
-				name: 'gpt-4-32k',
-				value: 'gpt-4-32k',
-			},
-			{
-				name: 'gpt-4-32k-0314',
-				value: 'gpt-4-32k-0314',
+		typeOptions: {
+			loadOptions: {
+				routing: {
+					request: {
+						method: 'GET',
+						url: '/v1/models',
+					},
+					output: {
+						postReceive: [
+							{
+								type: 'rootProperty',
+								properties: {
+									property: 'data',
+								},
+							},
+							{
+								type: 'filter',
+								properties: {
+									pass: "={{ $responseItem.id.startsWith('gpt-') }}",
+								},
+							},
+							{
+								type: 'setKeyValue',
+								properties: {
+									// eslint-disable-next-line n8n-nodes-base/node-param-display-name-miscased-id
+									name: '={{$responseItem.id}}',
+									value: '={{$responseItem.id}}',
+								},
+							},
+							{
+								type: 'sort',
+								properties: {
+									key: 'name',
+								},
+							},
+						],
+					},
+				},
 			},
-		],
+		},
 		routing: {
 			send: {
 				type: 'body',

diff --git a/packages/nodes-base/nodes/OpenAi/TextDescription.ts b/packages/nodes-base/nodes/OpenAi/TextDescription.ts
@@ -84,7 +84,7 @@ const completeOperations: INodeProperties[] = [
 							{
 								type: 'filter',
 								properties: {
-									pass: "={{ !$responseItem.id.startsWith('audio-') && !['cushman:2020-05-03', 'davinci-if:3.0.0', 'davinci-instruct-beta:2.0.0', 'if'].includes($responseItem.id) && !$responseItem.id.includes('-edit-') && !$responseItem.id.endsWith(':001') }}",
+									pass: "={{ !$responseItem.id.startsWith('audio-') && !$responseItem.id.startsWith('gpt-') && !$responseItem.id.startsWith('whisper-') && !['cushman:2020-05-03', 'davinci-if:3.0.0', 'davinci-instruct-beta:2.0.0', 'if'].includes($responseItem.id) && !$responseItem.id.includes('-edit-') && !$responseItem.id.endsWith(':001') }}",
 								},
 							},
 							{