Ability to update .env file for adding openai key

Author: jsajal

packages/flowtest-electron/src/ipc/collection.js

@@ -168,6 +168,24 @@ const registerRendererEventHandlers = (mainWindow, watcher) => {
     }
   });
 
+  ipcMain.handle('renderer:addOrUpdate-dotEnvironment', async (event, collectionPath, variables) => {
+    try {
+      const pathname = path.join(collectionPath, '.env');
+      // variables should be of format `k1=v1\nk2=v2`;
+
+      // Append to the .env file or create it if it doesn't exist
+      fs.appendFile(pathname, variables, (err) => {
+        if (err) {
+          console.error('Error writing to .env file:', err);
+          return Promise.reject(error);
+        }
+        console.log('.env file has been updated');
+      });
+    } catch (error) {
+      return Promise.reject(error);
+    }
+  });
+
   ipcMain.handle('renderer:create-flowtest', async (event, name, path, flowData) => {
     try {
       if (isDirectory(path)) {

src/components/molecules/flow/flowtestai.js

@@ -59,11 +59,9 @@ export const generateFlowData = async (instruction, modelName, collectionId) =>
         return Promise.reject(new Error(`model: ${modelName} not supported`));
       }
     } else {
-      return Promise.reject(new Error('A flowtest with this path does not exist'));
+      return Promise.reject(new Error('Collection not found'));
     }
   } catch (error) {
-    console.log(`Error generating flowData: ${error}`);
-    // TODO: show error in UI
-    return Promise.reject(error);
+    return Promise.reject(`Error generating flowData: ${error}`);
   }
 };

src/components/molecules/flow/index.js

@@ -310,6 +310,7 @@ const Flow = ({ tabId, collectionId, flowData }) => {
                   setEdges(result.edges);
                 })
                 .catch((error) => {
+                  // TODO: show error in UI
                   console.log(error);
                 });
             }}

src/service/collection.js

@@ -147,6 +147,23 @@ export const deleteEnvironmentFile = (name, collectionId) => {
   }
 };
 
+export const addOrUpdateDotEnvironmentFile = (collectionId, variables) => {
+  const { ipcRenderer } = window;
+
+  const collection = useCollectionStore.getState().collections.find((c) => c.id === collectionId);
+
+  if (collection) {
+    return new Promise((resolve, reject) => {
+      ipcRenderer
+        .invoke('renderer:addOrUpdate-dotEnvironment', collection.pathname, variables)
+        .then(resolve)
+        .catch(reject);
+    });
+  } else {
+    return Promise.reject(new Error('Collection not found'));
+  }
+};
+
 export const createFlowTest = (name, folderPath, collectionId) => {
   try {
     const { ipcRenderer } = window;