better handling of starting the prmpt

microsoft · Apr 29, 2024 · 47782ab · 47782ab
1 parent 7412363
commit 47782ab
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 13 deletions.
diff --git a/packages/core/src/test.ts b/packages/core/src/test.ts
@@ -44,13 +44,23 @@ export function generatePromptFooConfiguration(
                                 text: {
                                     id: "azureopenai:chat:gpt-4",
                                     config: {
-                                        apiHost: testProvider,
+                                        apiHost: testProvider
+                                            .replace(/^https:\/\//i, "")
+                                            .replace(
+                                                /\/openai\/deployments$/i,
+                                                ""
+                                            ),
                                     },
                                 },
                                 embedding: {
                                     id: "azureopenai:embeddings:text-embedding-ada-002",
                                     config: {
-                                        apiHost: testProvider,
+                                        apiHost: testProvider
+                                            .replace(/^https:\/\//i, "")
+                                            .replace(
+                                                /\/openai\/deployments$/i,
+                                                ""
+                                            ),
                                     },
                                 },
                             }

diff --git a/packages/vscode/src/fragmentcommands.ts b/packages/vscode/src/fragmentcommands.ts
@@ -73,17 +73,23 @@ async function showPromptParametersQuickPicks(
 }
 
 async function startTestViewer() {
-    // show results
-    const url = "http://localhost:15500"
-    const terminal = vscode.window.createTerminal({
-        name: "Promptfoo viewer",
-        isTransient: true,
-        env: {
-            PROMPTFOO_DISABLE_TELEMETRY: "1",
-            PROMPTFOO_DISABLE_UPDATE: "1",
-        },
-    })
-    terminal.sendText(`npx --yes promptfoo@latest view -y`)
+    const name = "Promptfoo View"
+    if (vscode.window.terminals.find((t) => t.name === name)) {
+        await vscode.env.openExternal(
+            vscode.Uri.parse("http://localhost:15500")
+        )
+    } else {
+        // show results
+        const terminal = vscode.window.createTerminal({
+            name,
+            isTransient: true,
+            env: {
+                PROMPTFOO_DISABLE_TELEMETRY: "1",
+                PROMPTFOO_DISABLE_UPDATE: "1",
+            },
+        })
+        terminal.sendText(`npx --yes promptfoo@latest view -y`)
+    }
 }
 
 export function activateFragmentCommands(state: ExtensionState) {