chore: add panic recover for openai component

namwoam · Aug 31, 2024 · ae8db60 · ae8db60
1 parent 6f1faa8
commit ae8db60
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/ai/openai/v0/main.go b/ai/openai/v0/main.go
@@ -128,7 +128,13 @@ type workerResult struct {
 }
 
 func (e *execution) worker(ctx context.Context, client *httpclient.Client, batchIdx int, input *structpb.Struct, result chan<- *workerResult) {
-	defer func() { result <- &workerResult{done: true} }()
+	defer func() {
+		if r := recover(); r != nil {
+			fmt.Printf("panic: %+v", r)
+			result <- &workerResult{batchIdx: batchIdx, err: fmt.Errorf("panic: %+v", r)}
+		}
+		result <- &workerResult{done: true}
+	}()
 
 	switch e.Task {
 	case TextGenerationTask: