Apply review suggestions

Author: Nuvindu

workspaces/ballerina/ballerina-extension/e2e-test/e2e-playwright-tests/file-integration/smb.spec.ts

@@ -76,7 +76,7 @@ export default function createTests() {
 
             const saveChangesBtn = artifactWebView.locator('#save-changes-btn vscode-button[appearance="primary"]');
             await saveChangesBtn.waitFor({ state: 'visible' });
-            await expect(saveChangesBtn).toHaveClass('disabled', { timeout: 5000 });
+            await expect(saveChangesBtn).toHaveClass(/disabled/, { timeout: 5000 });
             await expect(saveChangesBtn).toHaveText('Save Changes');
 
             const backBtn = artifactWebView.locator('[data-testid="back-button"]');

workspaces/ballerina/ballerina-visualizer/src/views/BI/ServiceDesigner/Forms/SMBForm/index.tsx

@@ -102,7 +102,7 @@ export function SMBForm(props: SMBFormProps) {
             setFunctionModel(initialFunction);
             setSelectedFileFormat(initialFunction.name?.metadata?.label || '');
         }
-    }, [selectedHandler]);
+    }, [selectedHandler, isNew]);
 
     const handleParamChange = (params: ParameterModel[]) => {
         if (functionModel) {
@@ -149,7 +149,7 @@ export function SMBForm(props: SMBFormProps) {
 
         if (selectedFileFormat === 'RAW') {
             if (isStreamEnabled) {
-                return `stream<byte[], error>`;
+                return `stream<byte[], error?>`;
             } else {
                 return `byte[]`;
             }
@@ -158,15 +158,17 @@ export function SMBForm(props: SMBFormProps) {
         if (baseType.endsWith("[]") && baseType !== "string[]") {
             baseType = baseType.slice(0, -2);
         } else if (baseType.startsWith("stream<")) {
-            if (baseType.endsWith(", error>")) {
+            if (baseType.endsWith(", error?>")) {
+                baseType = baseType.slice(7, -9);
+            } else if (baseType.endsWith(", error>")) {
                 baseType = baseType.slice(7, -8);
             } else if (baseType.endsWith(">")) {
                 baseType = baseType.slice(7, -1);
             }
         }
 
         if (isStreamEnabled) {
-            return `stream<${baseType}, error>`;
+            return `stream<${baseType}, error?>`;
         } else {
             return `${baseType}[]`;
         }
@@ -183,7 +185,9 @@ export function SMBForm(props: SMBFormProps) {
         let baseType = typeValue;
 
         if (baseType.startsWith("stream<")) {
-            if (baseType.endsWith(", error>")) {
+            if (baseType.endsWith(", error?>")) {
+                baseType = baseType.slice(7, -9);
+            } else if (baseType.endsWith(", error>")) {
                 baseType = baseType.slice(7, -8);
             } else if (baseType.endsWith(">")) {
                 baseType = baseType.slice(7, -1);