mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-09-30 01:52:29 +00:00
0dff85b9b5
* feat(seekdb): add Seekdb file and vector stores with pyseekdb>=1.2.0 * refactor(seekdb): add pyseekdb_conn and remote-only host/port config * refactor(embedding): remove env fallbacks from BaseEmbeddingModel; pass credentials in tests * refactor(seekdb): drop tenant from client kwargs; default database test and empty password * fix(deps): gate pyseekdb to Python >=3.11 for CI 3.10 compatibility * fix(seekdb): satisfy pre-commit pylint and formatting for seekdb stores
57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
"""Build ``pyseekdb.Client`` / ``AdminClient`` kwargs for embedded vs remote OceanBase / seekdb."""
|
|
|
|
from __future__ import annotations
|
|
|
|
DEFAULT_SEEKDB_PORT = 2881
|
|
DEFAULT_SEEKDB_USER = "root"
|
|
DEFAULT_SEEKDB_DATABASE = "test"
|
|
|
|
|
|
def parse_host_port(host: str | None, port: int | None) -> tuple[str | None, int | None]:
|
|
"""Resolve remote ``host`` / ``port`` (``port`` defaults to :data:`DEFAULT_SEEKDB_PORT`)."""
|
|
if host and host.strip():
|
|
return host.strip(), port if port is not None else DEFAULT_SEEKDB_PORT
|
|
return None, None
|
|
|
|
|
|
def build_pyseekdb_client_kwargs(
|
|
*,
|
|
path: str | None = None,
|
|
database: str,
|
|
host: str | None = None,
|
|
port: int | None = None,
|
|
user: str | None = None,
|
|
password: str = "",
|
|
) -> tuple[bool, dict]:
|
|
"""Return ``(is_remote, kwargs)`` for ``pyseekdb.Client``.
|
|
|
|
Remote when ``host`` is set. Embedded: optional ``path`` for the data directory; if
|
|
omitted, pyseekdb uses its default (typically ``seekdb.db`` under the CWD).
|
|
"""
|
|
h, p = parse_host_port(host, port)
|
|
if h:
|
|
return True, {
|
|
"host": h,
|
|
"port": p,
|
|
"database": database,
|
|
"user": user if user is not None else DEFAULT_SEEKDB_USER,
|
|
"password": password,
|
|
}
|
|
kw: dict = {"database": database}
|
|
if path:
|
|
kw["path"] = path
|
|
return False, kw
|
|
|
|
|
|
def admin_kwargs_from_client_kwargs(client_kw: dict) -> dict:
|
|
"""Strip ``database`` for ``AdminClient`` (admin uses system DB)."""
|
|
if "path" in client_kw:
|
|
return {"path": client_kw["path"]}
|
|
if "host" in client_kw:
|
|
return {
|
|
"host": client_kw["host"],
|
|
"port": client_kw["port"],
|
|
"user": client_kw["user"],
|
|
"password": client_kw["password"],
|
|
}
|
|
return {}
|