diff options
| author | Grail Finder <wohilas@gmail.com> | 2026-03-06 18:58:23 +0300 |
|---|---|---|
| committer | Grail Finder <wohilas@gmail.com> | 2026-03-06 18:58:23 +0300 |
| commit | 17b68bc21fae99c17ec48e046e67a643b9d159bb (patch) | |
| tree | 00b2da2f55876e720aecccc10dbc59232da768db /rag/embedder.go | |
| parent | edfd43c52ae3f2fa16f6ab5d64cb48218a2c0a64 (diff) | |
Enha (rag): async writes
Diffstat (limited to 'rag/embedder.go')
| -rw-r--r-- | rag/embedder.go | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/rag/embedder.go b/rag/embedder.go index 39f4b5c..fd4cfa7 100644 --- a/rag/embedder.go +++ b/rag/embedder.go @@ -11,6 +11,7 @@ import ( "net/http" "os" "sync" + "time" "github.com/sugarme/tokenizer" "github.com/sugarme/tokenizer/pretrained" @@ -33,8 +34,10 @@ type APIEmbedder struct { func NewAPIEmbedder(l *slog.Logger, cfg *config.Config) *APIEmbedder { return &APIEmbedder{ logger: l, - client: &http.Client{}, - cfg: cfg, + client: &http.Client{ + Timeout: 30 * time.Second, + }, + cfg: cfg, } } |
