ydb-qdrant 4.4.0 → 4.5.0

package/README.md

@@ -12,7 +12,7 @@
 
 [![npm version](https://img.shields.io/npm/v/ydb-qdrant.svg)](https://www.npmjs.com/package/ydb-qdrant)
 [![Docker Image](https://img.shields.io/badge/docker-ghcr.io%2Fastandrik%2Fydb--qdrant-blue?logo=docker)](https://github.com/users/astandrik/packages/container/package/ydb-qdrant)
-[![License: ISC](https://img.shields.io/badge/License-ISC-blue.svg)](https://opensource.org/licenses/ISC)
+[![License](https://img.shields.io/badge/License-Apache_2.0-blue.svg)](https://opensource.org/licenses/Apache-2.0)
 
 # YDB Qdrant-compatible Service
 

package/dist/server.d.ts

@@ -1 +1,3 @@
+import type { Request, Response } from "express";
+export declare function healthHandler(_req: Request, res: Response): Promise<void>;
 export declare function buildServer(): import("express-serve-static-core").Express;

package/dist/server.js

@@ -2,11 +2,20 @@ import express from "express";
 import { collectionsRouter } from "./routes/collections.js";
 import { pointsRouter } from "./routes/points.js";
 import { requestLogger } from "./middleware/requestLogger.js";
+import { isYdbAvailable } from "./ydb/client.js";
+export async function healthHandler(_req, res) {
+    const ok = await isYdbAvailable();
+    if (!ok) {
+        res.status(503).json({ status: "error", error: "YDB unavailable" });
+        return;
+    }
+    res.json({ status: "ok" });
+}
 export function buildServer() {
     const app = express();
     app.use(express.json({ limit: "20mb" }));
     app.use(requestLogger);
-    app.get("/health", (_req, res) => res.json({ status: "ok" }));
+    app.get("/health", healthHandler);
     app.use("/collections", collectionsRouter);
     app.use("/collections", pointsRouter);
     return app;

package/dist/ydb/client.d.ts

@@ -7,6 +7,8 @@ type DriverConfig = {
     connectionString?: string;
     authService?: IAuthService;
 };
+export declare function __setDriverForTests(fake: unknown): void;
 export declare function configureDriver(config: DriverConfig): void;
 export declare function readyOrThrow(): Promise<void>;
 export declare function withSession<T>(fn: (s: Session) => Promise<T>): Promise<T>;
+export declare function isYdbAvailable(timeoutMs?: number): Promise<boolean>;

package/dist/ydb/client.js

@@ -3,8 +3,14 @@ import { YDB_DATABASE, YDB_ENDPOINT } from "../config/env.js";
 const require = createRequire(import.meta.url);
 const { Driver, getCredentialsFromEnv, Types, TypedValues, TableDescription, Column, } = require("ydb-sdk");
 export { Types, TypedValues, TableDescription, Column };
+const DRIVER_READY_TIMEOUT_MS = 15000;
+const TABLE_SESSION_TIMEOUT_MS = 20000;
+const YDB_HEALTHCHECK_READY_TIMEOUT_MS = 5000;
 let overrideConfig;
 let driver;
+export function __setDriverForTests(fake) {
+    driver = fake;
+}
 export function configureDriver(config) {
     if (driver) {
         // Driver already created; keep existing connection settings.
@@ -30,12 +36,21 @@ function getOrCreateDriver() {
 }
 export async function readyOrThrow() {
     const d = getOrCreateDriver();
-    const ok = await d.ready(10000);
+    const ok = await d.ready(DRIVER_READY_TIMEOUT_MS);
     if (!ok) {
-        throw new Error("YDB driver is not ready in 10s. Check connectivity and credentials.");
+        throw new Error(`YDB driver is not ready in ${DRIVER_READY_TIMEOUT_MS / 1000}s. Check connectivity and credentials.`);
     }
 }
 export async function withSession(fn) {
     const d = getOrCreateDriver();
-    return await d.tableClient.withSession(fn, 15000);
+    return await d.tableClient.withSession(fn, TABLE_SESSION_TIMEOUT_MS);
+}
+export async function isYdbAvailable(timeoutMs = YDB_HEALTHCHECK_READY_TIMEOUT_MS) {
+    const d = getOrCreateDriver();
+    try {
+        return await d.ready(timeoutMs);
+    }
+    catch {
+        return false;
+    }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ydb-qdrant",
-  "version": "4.4.0",
+  "version": "4.5.0",
   "main": "dist/package/api.js",
   "types": "dist/package/api.d.ts",
   "exports": {
@@ -46,7 +46,7 @@
     "zod"
   ],
   "author": "",
-  "license": "ISC",
+  "license": "Apache-2.0",
   "description": "Qdrant-compatible Node.js/TypeScript API that stores/searches embeddings in YDB using single-phase top-k vector search with an automatic vector index and table-scan fallback.",
   "type": "module",
   "publishConfig": {