fix: add record after yocsef

JingYiJun · May 24, 2024 · 2956a9a · 2956a9a
1 parent 6767233
commit 2956a9a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/apis/record/yocsef.go b/apis/record/yocsef.go
@@ -74,7 +74,7 @@ func InferYocsefAsyncAPI(c *websocket.Conn) {
 
 		DB.Create(&record)
 
-		_ = c.WriteJSON(record)
+		_ = c.WriteJSON(InferenceResponse{Response: record.Response})
 
 		return nil
 	}