[server] Increase emb fetchTimeout to 15seconds
This commit is contained in:
parent
0dcc3c974c
commit
ef5740e728
|
@ -30,7 +30,8 @@ import (
|
||||||
|
|
||||||
const (
|
const (
|
||||||
// maxEmbeddingDataSize is the min size of an embedding object in bytes
|
// maxEmbeddingDataSize is the min size of an embedding object in bytes
|
||||||
minEmbeddingDataSize = 2048
|
minEmbeddingDataSize = 2048
|
||||||
|
embeddingFetchTimeout = 15 * gTime.Second
|
||||||
)
|
)
|
||||||
|
|
||||||
type Controller struct {
|
type Controller struct {
|
||||||
|
@ -345,7 +346,7 @@ func (c *Controller) getEmbeddingObjectsParallelV2(userID int64, dbEmbeddingRows
|
||||||
defer wg.Done()
|
defer wg.Done()
|
||||||
defer func() { <-globalFileFetchSemaphore }() // Release back to global semaphore
|
defer func() { <-globalFileFetchSemaphore }() // Release back to global semaphore
|
||||||
objectKey := c.getObjectKey(userID, dbEmbeddingRow.FileID, dbEmbeddingRow.Model)
|
objectKey := c.getObjectKey(userID, dbEmbeddingRow.FileID, dbEmbeddingRow.Model)
|
||||||
ctx, cancel := context.WithTimeout(context.Background(), gTime.Second*10) // 10 seconds timeout
|
ctx, cancel := context.WithTimeout(context.Background(), embeddingFetchTimeout)
|
||||||
defer cancel()
|
defer cancel()
|
||||||
obj, err := c.getEmbeddingObjectWithRetries(ctx, objectKey, downloader, 0)
|
obj, err := c.getEmbeddingObjectWithRetries(ctx, objectKey, downloader, 0)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|
Loading…
Reference in a new issue