mishig HF staff commited on
Commit
0c2fe0d
1 Parent(s): 58fa8dd

Improve error handling

Browse files
src/lib/components/InferencePlayground/InferencePlayground.svelte CHANGED
@@ -76,6 +76,7 @@
76
  }
77
  (document.activeElement as HTMLElement).blur();
78
  loading = true;
 
79
 
80
  try {
81
  const startTime = performance.now();
@@ -84,6 +85,7 @@
84
  if (conversation.streaming) {
85
  const streamingMessage = { role: "assistant", content: "" };
86
  conversation.messages = [...conversation.messages, streamingMessage];
 
87
  abortController = new AbortController();
88
 
89
  await handleStreamingResponse(
@@ -109,6 +111,10 @@
109
  const endTime = performance.now();
110
  latency = Math.round(endTime - startTime);
111
  } catch (error) {
 
 
 
 
112
  if (error.name !== "AbortError") {
113
  alert("error: " + (error as Error).message);
114
  }
@@ -127,7 +133,7 @@
127
  function handleTokenSubmit(e: Event) {
128
  const form = e.target as HTMLFormElement;
129
  const formData = new FormData(form);
130
- hfToken = (formData.get("hf-token") as string) ?? "";
131
  submit();
132
  showTokenModal = false;
133
  }
 
76
  }
77
  (document.activeElement as HTMLElement).blur();
78
  loading = true;
79
+ let streamingMsgAdded = false;
80
 
81
  try {
82
  const startTime = performance.now();
 
85
  if (conversation.streaming) {
86
  const streamingMessage = { role: "assistant", content: "" };
87
  conversation.messages = [...conversation.messages, streamingMessage];
88
+ streamingMsgAdded = true;
89
  abortController = new AbortController();
90
 
91
  await handleStreamingResponse(
 
111
  const endTime = performance.now();
112
  latency = Math.round(endTime - startTime);
113
  } catch (error) {
114
+ if (streamingMsgAdded) {
115
+ conversation.messages = conversation.messages.slice(0, -1);
116
+ }
117
+ hfToken = "";
118
  if (error.name !== "AbortError") {
119
  alert("error: " + (error as Error).message);
120
  }
 
133
  function handleTokenSubmit(e: Event) {
134
  const form = e.target as HTMLFormElement;
135
  const formData = new FormData(form);
136
+ hfToken = (formData.get("hf-token") as string).trim() ?? "";
137
  submit();
138
  showTokenModal = false;
139
  }