update ui

Arize-ai · Oct 31, 2024 · 78f0b0a · 78f0b0a
1 parent a29f4d2
commit 78f0b0a
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 39 deletions.
diff --git a/app/schema.graphql b/app/schema.graphql
@@ -96,11 +96,7 @@ enum ChatCompletionMessageRole {
   AI
 }
 
-type ChatCompletionSubscriptionError {
-  message: String!
-}
-
-union ChatCompletionSubscriptionPayload = TextChunk | ToolCallChunk | FinishedChatCompletion | ChatCompletionSubscriptionError
+union ChatCompletionSubscriptionPayload = TextChunk | ToolCallChunk | FinishedChatCompletion
 
 input ClearProjectInput {
   id: GlobalID!
@@ -835,6 +831,7 @@ type ExportedFile {
 
 type FinishedChatCompletion {
   span: Span!
+  errorMessage: String
 }
 
 type FunctionCallChunk {

diff --git a/app/src/pages/playground/PlaygroundOutput.tsx b/app/src/pages/playground/PlaygroundOutput.tsx
@@ -200,9 +200,7 @@ function useChatCompletionSubscription({
               span {
                 id
               }
-            }
-            ... on ChatCompletionSubscriptionError {
-              message
+              errorMessage
             }
           }
         }
@@ -378,20 +376,12 @@ function PlaygroundOutputText(props: PlaygroundInstanceProps) {
             spanId: chatCompletion.span.id,
           },
         });
-      } else if (
-        chatCompletion.__typename === "ChatCompletionSubscriptionError"
-      ) {
-        markPlaygroundInstanceComplete(props.playgroundInstanceId);
-        updateInstance({
-          instanceId: props.playgroundInstanceId,
-          patch: {
-            isRunning: false,
-          },
-        });
-        notifyError({
-          title: "Chat completion failed",
-          message: chatCompletion.message,
-        });
+        if (chatCompletion.errorMessage != null) {
+          notifyError({
+            title: "Chat completion failed",
+            message: chatCompletion.errorMessage,
+          });
+        }
       }
     },
     onCompleted: () => {

diff --git a/app/src/pages/playground/__generated__/PlaygroundOutputSubscription.graphql.ts b/app/src/pages/playground/__generated__/PlaygroundOutputSubscription.graphql.ts