Salih/post streaming cleanup (#513)

This commit is contained in:
Kerem Yilmaz
2024-06-25 12:39:25 -07:00
committed by GitHub
parent a0a3aa6f83
commit e056c491a1
2 changed files with 9 additions and 106 deletions

View File

@@ -14,6 +14,7 @@ import { getClient } from "@/api/AxiosClient";
import { useCredentialGetter } from "@/hooks/useCredentialGetter";
import { Skeleton } from "@/components/ui/skeleton";
import { toast } from "@/components/ui/use-toast";
import { envCredential } from "@/util/env";
type StreamMessage = {
task_id: string;
@@ -70,12 +71,18 @@ function TaskActions() {
async function run() {
// Create WebSocket connection.
const credential = await credentialGetter!();
let credential = null;
if (credentialGetter) {
const token = await credentialGetter();
credential = `?token=Bearer ${token}`;
} else {
credential = `?apikey=${envCredential}`;
}
if (socket) {
socket.close();
}
socket = new WebSocket(
`${wssBaseUrl}/stream/tasks/${taskId}?token=Bearer ${credential}`,
`${wssBaseUrl}/stream/tasks/${taskId}${credential}`,
);
// Listen for messages
socket.addEventListener("message", (event) => {