update prop type to be InitialPlaygroundState

Arize-ai · Oct 8, 2024 · 80e35d7 · 80e35d7
1 parent 95fc185
commit 80e35d7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/app/src/pages/playground/Playground.tsx b/app/src/pages/playground/Playground.tsx
@@ -8,12 +8,12 @@ import {
   PlaygroundProvider,
   usePlaygroundContext,
 } from "@phoenix/contexts/PlaygroundContext";
-import { PlaygroundProps } from "@phoenix/store";
+import { InitialPlaygroundState } from "@phoenix/store";
 
 import { PlaygroundInstance } from "./PlaygroundInstance";
 import { PlaygroundOperationTypeRadioGroup } from "./PlaygroundOperationTypeRadioGroup";
 
-export function Playground(props: Partial<PlaygroundProps>) {
+export function Playground(props: InitialPlaygroundState) {
   return (
     <PlaygroundProvider {...props}>
       <View

diff --git a/app/src/store/playgroundStore.tsx b/app/src/store/playgroundStore.tsx
@@ -87,6 +87,8 @@ export interface PlaygroundProps {
   instances: Array<PlaygroundInstance>;
 }
 
+export type InitialPlaygroundState = Partial<PlaygroundProps>;
+
 type DatasetInput = {
   datasetId: string;
 };
@@ -192,7 +194,7 @@ const DEFAULT_TEXT_COMPLETION_TEMPLATE: PlaygroundTextCompletionTemplate = {
 };
 
 export const createPlaygroundStore = (
-  initialProps?: Partial<PlaygroundProps>
+  initialProps?: InitialPlaygroundState
 ) => {
   const playgroundStore: StateCreator<PlaygroundState> = (set, get) => ({
     operationType: "chat",