From a4475abfb2ba5ce51cc223f6e27c18213bc842b2 Mon Sep 17 00:00:00 2001
From: q <q@think>
Date: Sat, 18 Apr 2026 21:21:28 +0300
Subject: [PATCH] Optimize for 4 cores: BM25 preload, httpx pool, orjson, fix
 lambda
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- index: UVICORN_WORKERS 8→4, lifespan BM25 preload, explicit ThreadPoolExecutor(4), orjson
- search: lifespan BM25 preload, httpx limits (max_conn=100, keepalive=20, timeout=30s), fix asyncio.to_thread lambda, orjson
- both: ORJSONResponse as default_response_class

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
---
 index/main.py           | 43 +++++++++++++++++++++++++++++------------
 index/requirements.txt  |  1 +
 search/main.py          | 19 ++++++++++++++----
 search/requirements.txt |  1 +
 4 files changed, 48 insertions(+), 16 deletions(-)

diff --git a/index/main.py b/index/main.py
index d6083ea..0b4bd92 100644
--- a/index/main.py
+++ b/index/main.py
@@ -1,17 +1,19 @@
 import asyncio
 import logging
 import os
+from concurrent.futures import ThreadPoolExecutor
+from contextlib import asynccontextmanager
 from functools import lru_cache
 from typing import Any
 
 from fastapi import FastAPI, Request
 from fastapi.exceptions import RequestValidationError
-from fastapi.responses import JSONResponse
+from fastapi.responses import JSONResponse, ORJSONResponse
 from pydantic import BaseModel
 
 HOST = os.getenv("HOST", "0.0.0.0")
 PORT = int(os.getenv("PORT", "8000"))
-UVICORN_WORKERS = 8
+UVICORN_WORKERS = 4
 
 logging.basicConfig(level=os.getenv("LOG_LEVEL", "INFO"))
 logger = logging.getLogger("index-service")
@@ -78,6 +80,24 @@ OVERLAP_SIZE = 128
 SPARSE_MODEL_NAME = "Qdrant/bm25"
 FASTEMBED_CACHE_PATH = "/models/fastembed"
 
+_thread_pool = ThreadPoolExecutor(max_workers=4)
+
+
+@lru_cache(maxsize=1)
+def get_sparse_model():
+    from fastembed import SparseTextEmbedding
+
+    logger.info("Loading sparse model %s from cache %s", SPARSE_MODEL_NAME, FASTEMBED_CACHE_PATH)
+    return SparseTextEmbedding(model_name=SPARSE_MODEL_NAME)
+
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # Preload BM25 model on startup to avoid cold-start latency
+    await asyncio.to_thread(get_sparse_model)
+    logger.info("BM25 model preloaded")
+    yield
+
 
 def render_message(message: Message) -> str:
     parts_list: list[str] = []
@@ -186,7 +206,12 @@ def build_chunks(
     return result
 
 
-app = FastAPI(title="Index Service", version="0.1.0")
+app = FastAPI(
+    title="Index Service",
+    version="0.1.0",
+    lifespan=lifespan,
+    default_response_class=ORJSONResponse,
+)
 
 
 @app.get("/health")
@@ -205,14 +230,6 @@ async def index(payload: IndexAPIRequest) -> IndexAPIResponse:
     )
 
 
-@lru_cache(maxsize=1)
-def get_sparse_model():
-    from fastembed import SparseTextEmbedding
-
-    logger.info("Loading sparse model %s from cache %s", SPARSE_MODEL_NAME, FASTEMBED_CACHE_PATH)
-    return SparseTextEmbedding(model_name=SPARSE_MODEL_NAME)
-
-
 def embed_sparse_texts(texts: list[str]) -> list[dict]:
     model = get_sparse_model()
     vectors = []
@@ -228,7 +245,9 @@ def embed_sparse_texts(texts: list[str]) -> list[dict]:
 
 @app.post("/sparse_embedding")
 async def sparse_embedding(payload: SparseEmbeddingRequest) -> dict[str, Any]:
-    vectors = await asyncio.to_thread(embed_sparse_texts, payload.texts)
+    vectors = await asyncio.get_event_loop().run_in_executor(
+        _thread_pool, embed_sparse_texts, payload.texts
+    )
     return {"vectors": vectors}
 
 
diff --git a/index/requirements.txt b/index/requirements.txt
index a8683a5..a71d4c8 100644
--- a/index/requirements.txt
+++ b/index/requirements.txt
@@ -2,3 +2,4 @@ fastapi==0.135.1
 uvicorn[standard]==0.42.0
 pydantic==2.12.5
 fastembed==0.7.4
+orjson==3.10.18
diff --git a/search/main.py b/search/main.py
index c5f2624..7fd3847 100644
--- a/search/main.py
+++ b/search/main.py
@@ -9,7 +9,7 @@ import httpx
 from fastembed import SparseTextEmbedding
 from fastapi import FastAPI, HTTPException, Request
 from fastapi.exceptions import RequestValidationError
-from fastapi.responses import JSONResponse
+from fastapi.responses import JSONResponse, ORJSONResponse
 from pydantic import BaseModel, Field
 from qdrant_client import AsyncQdrantClient, models
 
@@ -148,7 +148,13 @@ def get_sparse_model() -> SparseTextEmbedding:
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    app.state.http = httpx.AsyncClient()
+    # Preload BM25 and set up HTTP client with connection pooling
+    await asyncio.to_thread(get_sparse_model)
+    logger.info("BM25 model preloaded")
+    app.state.http = httpx.AsyncClient(
+        timeout=30.0,
+        limits=httpx.Limits(max_connections=100, max_keepalive_connections=20),
+    )
     app.state.qdrant = AsyncQdrantClient(
         url=QDRANT_URL,
         api_key=API_KEY,
@@ -160,7 +166,12 @@ async def lifespan(app: FastAPI):
         await app.state.qdrant.close()
 
 
-app = FastAPI(title="Search Service", version="0.1.0", lifespan=lifespan)
+app = FastAPI(
+    title="Search Service",
+    version="0.1.0",
+    lifespan=lifespan,
+    default_response_class=ORJSONResponse,
+)
 
 DENSE_PREFETCH_K = 80
 SPARSE_PREFETCH_K = 200
@@ -354,7 +365,7 @@ async def search(payload: SearchAPIRequest) -> SearchAPIResponse:
     sparse_query = build_sparse_query(question)
 
     dense_task = embed_dense(client, dense_query)
-    sparse_task = asyncio.to_thread(lambda: embed_sparse_sync(sparse_query))
+    sparse_task = asyncio.to_thread(embed_sparse_sync, sparse_query)
     dense_vector, sparse_vector = await asyncio.gather(dense_task, sparse_task)
 
     dense_vectors = [dense_vector]
diff --git a/search/requirements.txt b/search/requirements.txt
index 50b3c1d..0ca00a6 100644
--- a/search/requirements.txt
+++ b/search/requirements.txt
@@ -4,3 +4,4 @@ pydantic==2.12.5
 httpx==0.28.1
 qdrant-client==1.15.1
 fastembed==0.7.4
+orjson==3.10.18