mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-08-05 07:10:29 +08:00
feat(serenedb): add SereneDB doc-store engine (Go + Python connectors) (#17375)
## What Adds [**SereneDB**](https://serenedb.com) as a selectable doc-store engine on **both** RAGFlow paths: - the **Go** `DocEngine` (`internal/engine/serenedb`), alongside Elasticsearch and Infinity; - the **Python** `DocStoreConnection` (`rag/utils/serenedb_conn.py`) + `DOC_ENGINE=serenedb` registration. SereneDB is a PostgreSQL-wire engine (DuckDB execution) whose single inverted index carries **both** a scored text column (`@@`, BM25) and an IVF vector column (`<#>`, inner product), so hybrid search is one SQL statement. The Go engine connects with `database/sql` + `lib/pq` (already a dependency, no new module); the Python connector uses psycopg2 (already a dependency). ## Storage model One table per tenant with `kb_id` as a filter column - the **Elasticsearch / OceanBase** model, not Infinity's per-dataset tables. This keeps BM25 statistics (IDF, avgdl) computed over the whole tenant corpus (global IDF). Both connectors use this identical layout, so they are storage- and retrieval-compatible: `hybrid` proxy routing and Python↔Go switching are safe. On the Python side the connector is wired as OceanBase's plain-SQL sibling (chunk_data JSON metadata, inline chunk vectors, verbatim ES field names); the ES tokenizer path is unchanged. Metadata stays one table per tenant (`ragflow_doc_meta_<tenant>`). The query shapes mirror the Python connector, including the five empirically-found landmines: the scored dictionary needs `frequency + norm` (else `BM25()` silently returns 0.0), the `@@` query is the tokenized query, the scored lexical branch matches one column, vectors use an L2-normalized shadow column with `ip`/`sq8`, and the similarity threshold goes directly in the ANN scan's `WHERE`. **Minimum engine version: SereneDB 26.07.4.** --------- Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
@@ -72,6 +72,14 @@ INFINITY_THRIFT_PORT=23817
|
||||
INFINITY_HTTP_PORT=23820
|
||||
INFINITY_PSQL_PORT=5432
|
||||
|
||||
# The hostname where the SereneDB service is exposed. SereneDB speaks the
|
||||
# PostgreSQL wire protocol, so DOC_ENGINE=serenedb connects over lib/pq/psycopg2.
|
||||
SERENEDB_HOST=serenedb
|
||||
# Port to expose SereneDB to the host
|
||||
SERENEDB_PORT=7890
|
||||
# The password for SereneDB (POSTGRES_PASSWORD in the container)
|
||||
SERENEDB_PASSWORD=infini_rag_flow
|
||||
|
||||
# The hostname where the OceanBase service is exposed
|
||||
OCEANBASE_HOST=oceanbase
|
||||
# The port used to expose the OceanBase service
|
||||
|
||||
@@ -100,6 +100,27 @@ services:
|
||||
retries: 120
|
||||
restart: unless-stopped
|
||||
|
||||
serenedb:
|
||||
profiles:
|
||||
- serenedb
|
||||
image: serenedb/serenedb:26.07.5
|
||||
env_file: .env
|
||||
environment:
|
||||
- POSTGRES_PASSWORD=${SERENEDB_PASSWORD}
|
||||
ports:
|
||||
- ${SERENEDB_PORT}:7890
|
||||
volumes:
|
||||
- serenedb_data:/var/lib/serenedb
|
||||
mem_limit: ${MEM_LIMIT}
|
||||
networks:
|
||||
- ragflow
|
||||
healthcheck:
|
||||
test: ["CMD", "pg_isready", "-h", "127.0.0.1", "-p", "7890", "-U", "postgres"]
|
||||
interval: 10s
|
||||
timeout: 10s
|
||||
retries: 120
|
||||
restart: unless-stopped
|
||||
|
||||
oceanbase:
|
||||
profiles:
|
||||
- oceanbase
|
||||
@@ -380,6 +401,8 @@ volumes:
|
||||
driver: local
|
||||
infinity_data:
|
||||
driver: local
|
||||
serenedb_data:
|
||||
driver: local
|
||||
ob_data:
|
||||
driver: local
|
||||
seekdb_data:
|
||||
|
||||
@@ -41,6 +41,12 @@ infinity:
|
||||
uri: '${INFINITY_HOST:-infinity}:23817'
|
||||
postgres_port: 5432
|
||||
db_name: 'default_db'
|
||||
serenedb:
|
||||
host: '${SERENEDB_HOST:-serenedb}'
|
||||
port: 7890
|
||||
user: '${SERENEDB_USER:-postgres}'
|
||||
password: '${SERENEDB_PASSWORD:-infini_rag_flow}'
|
||||
db_name: '${SERENEDB_DBNAME:-postgres}'
|
||||
oceanbase:
|
||||
scheme: 'oceanbase' # set 'mysql' to create connection using mysql config
|
||||
config:
|
||||
|
||||
Reference in New Issue
Block a user