update initial instance id

Arize-ai · Oct 7, 2024 · fabb2e3 · fabb2e3
1 parent bf9bc78
commit fabb2e3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/app/src/pages/playground/utils.ts b/app/src/pages/playground/utils.ts
@@ -1,4 +1,7 @@
-import { PlaygroundInstance } from "@phoenix/store";
+import {
+  INITIAL_PLAYGROUND_INSTANCE_ID,
+  PlaygroundInstance,
+} from "@phoenix/store";
 import { spanPlaygroundPageLoaderQuery$data } from "./__generated__/spanPlaygroundPageLoaderQuery.graphql";
 import { safelyParseJSON } from "@phoenix/utils/jsonUtils";
 import { llmAttributesSchema } from "./schemas";
@@ -22,7 +25,7 @@ export function transformSpanAttributesToPlaygroundInstance(
     throw new Error("Invalid data");
   }
   return {
-    id: 1,
+    id: INITIAL_PLAYGROUND_INSTANCE_ID,
     activeRunId: null,
     isRunning: false,
     input: {

diff --git a/app/src/store/playgroundStore.tsx b/app/src/store/playgroundStore.tsx
@@ -3,7 +3,8 @@ import { devtools } from "zustand/middleware";
 
 export type GenAIOperationType = "chat" | "text_completion";
 
-let playgroundInstanceIdIndex = 0;
+export const INITIAL_PLAYGROUND_INSTANCE_ID = 0;
+let playgroundInstanceIdIndex = INITIAL_PLAYGROUND_INSTANCE_ID;
 let playgroundRunIdIndex = 0;
 
 /**
@@ -182,6 +183,9 @@ const DEFAULT_TEXT_COMPLETION_TEMPLATE: PlaygroundTextCompletionTemplate = {
 export const createPlaygroundStore = (
   initialProps?: Partial<PlaygroundProps>
 ) => {
+  if (initialProps?.instances != null && initialProps.instances.length > 0) {
+    playgroundInstanceIdIndex += initialProps.instances.length;
+  }
   const playgroundStore: StateCreator<PlaygroundState> = (set, get) => ({
     operationType: "chat",
     inputMode: "manual",