From 51a57d4a56724595d7740ebee07cd0d8cb33a8ba Mon Sep 17 00:00:00 2001
From: Andy Brenneke <andy.brenneke@ironcladhq.com>
Date: Fri, 16 Feb 2024 10:29:50 -0800
Subject: [PATCH] Fix "fine-tune-results" being an file upload type option

---
 packages/core/src/plugins/openai/nodes/UploadFileNode.ts | 8 ++++----
 packages/core/src/utils/openai.ts                        | 6 ++++++
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/packages/core/src/plugins/openai/nodes/UploadFileNode.ts b/packages/core/src/plugins/openai/nodes/UploadFileNode.ts
index 4163af656..22abdbb2e 100644
--- a/packages/core/src/plugins/openai/nodes/UploadFileNode.ts
+++ b/packages/core/src/plugins/openai/nodes/UploadFileNode.ts
@@ -9,12 +9,12 @@ import {
 import { newId, dedent, getInputOrData, coerceType, coerceTypeOptional } from '../../../utils/index.js';
 import { pluginNodeDefinition } from '../../../model/NodeDefinition.js';
 import { handleOpenAIError } from '../handleOpenaiError.js';
-import { openAIFilePurposeOptions } from '../../../utils/openai.js';
+import { openAIFileUploadPurposeOptions } from '../../../utils/openai.js';
 
 export type UploadFileNode = ChartNode<'openaiUploadFile', UploadFileNodeData>;
 
 export type UploadFileNodeData = {
-  purpose: 'fine-tune' | 'fine-tune-results' | 'assistants' | 'assistants_output';
+  purpose: 'fine-tune' | 'assistants' | 'assistants_output';
 };
 
 export const UploadFileNodeImpl: PluginNodeImpl<UploadFileNode> = {
@@ -91,7 +91,7 @@ export const UploadFileNodeImpl: PluginNodeImpl<UploadFileNode> = {
         type: 'dropdown',
         dataKey: 'purpose',
         label: 'Purpose',
-        options: openAIFilePurposeOptions,
+        options: openAIFileUploadPurposeOptions,
         defaultValue: 'assistants',
       },
     ];
@@ -99,7 +99,7 @@ export const UploadFileNodeImpl: PluginNodeImpl<UploadFileNode> = {
 
   getBody(data) {
     return dedent`
-      Purpose: ${openAIFilePurposeOptions.find(({ value }) => value === data.purpose)?.label ?? 'Unknown'}
+      Purpose: ${openAIFileUploadPurposeOptions.find(({ value }) => value === data.purpose)?.label ?? 'Unknown'}
     `;
   },
 
diff --git a/packages/core/src/utils/openai.ts b/packages/core/src/utils/openai.ts
index 4c16c677f..762377e30 100644
--- a/packages/core/src/utils/openai.ts
+++ b/packages/core/src/utils/openai.ts
@@ -568,6 +568,12 @@ export const openAIFilePurposeOptions = [
   { value: 'assistants_output', label: 'Assistants Output' },
 ];
 
+export const openAIFileUploadPurposeOptions = [
+  { value: 'fine-tune', label: 'Fine-tuning' },
+  { value: 'assistants', label: 'Assistants' },
+  { value: 'assistants_output', label: 'Assistants Output' },
+];
+
 export type OpenAIThread = {
   /** The identifier, which can be referenced in API endpoints. */
   id: string;