浏览代码

Add log when embedding is fetched after retry

Neeraj Gupta 1 年之前
父节点
当前提交
cc457eca98
共有 1 个文件被更改,包括 3 次插入0 次删除
  1. 3 0
      server/pkg/controller/embedding/controller.go

+ 3 - 0
server/pkg/controller/embedding/controller.go

@@ -420,6 +420,9 @@ func (c *Controller) getEmbeddingObject(ctx context.Context, objectKey string, d
 			obj, err := c.downloadObject(fetchCtx, objectKey, downloader, c.embeddingBucket)
 			obj, err := c.downloadObject(fetchCtx, objectKey, downloader, c.embeddingBucket)
 			cancel() // Ensure cancel is called to release resources
 			cancel() // Ensure cancel is called to release resources
 			if err == nil {
 			if err == nil {
+				if i > 0 {
+					ctxLogger.Infof("Fetched object after %d attempts", i)
+				}
 				return obj, nil
 				return obj, nil
 			}
 			}
 			// Check if the error is due to context timeout or cancellation
 			// Check if the error is due to context timeout or cancellation