fix(playground): handle experiment id's for non streaming (#5601)

Arize-ai · Dec 3, 2024 · 1a51350 · 1a51350
1 parent 37c2092
commit 1a51350
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/app/src/pages/playground/PlaygroundDatasetExamplesTable.tsx b/app/src/pages/playground/PlaygroundDatasetExamplesTable.tsx
@@ -507,14 +507,25 @@ export function PlaygroundDatasetExamplesTable({
           });
           return;
         }
+        updateInstance({
+          instanceId,
+          patch: {
+            experimentId: response.chatCompletionOverDataset.experimentId,
+          },
+        });
         setExampleDataForInstance({
           instanceId,
           data: createExampleResponsesForInstance(
             response.chatCompletionOverDataset
           ),
         });
       },
-    [markPlaygroundInstanceComplete, notifyError, setExampleDataForInstance]
+    [
+      markPlaygroundInstanceComplete,
+      notifyError,
+      setExampleDataForInstance,
+      updateInstance,
+    ]
   );
 
   useEffect(() => {
@@ -584,6 +595,10 @@ export function PlaygroundDatasetExamplesTable({
         if (activeRunId === null) {
           continue;
         }
+        updateInstance({
+          instanceId: instance.id,
+          patch: { experimentId: null },
+        });
         const variables = {
           input: getChatCompletionOverDatasetInput({
             credentials,