Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
90f6685c31 |
@@ -1,6 +1,6 @@
|
|||||||
# 那拉提文旅智能问答
|
# 那拉提文旅智能问答
|
||||||
|
|
||||||
Vue 3 + TypeScript + Vite 前端;FastAPI + Pydantic + SQLAlchemy 后端;PostgreSQL + pgvector。当前后端提供知识管理、来源、文档、人工审核、关键词/向量/RRF 检索、文本与向量发布、历史记录及认证/角色权限,问答仍为占位接口。
|
Vue 3 + TypeScript + Vite 前端;FastAPI + Pydantic + SQLAlchemy 后端;PostgreSQL + pgvector。当前后端提供知识管理、来源、文档、人工审核、关键词检索、历史记录及认证/角色权限,问答仍为占位接口。
|
||||||
|
|
||||||
## 启动数据库
|
## 启动数据库
|
||||||
需要可用的 Docker Engine(Windows 可使用 Docker Desktop),在项目根目录运行:
|
需要可用的 Docker Engine(Windows 可使用 Docker Desktop),在项目根目录运行:
|
||||||
@@ -34,7 +34,7 @@ uvicorn app.main:app --reload --host 127.0.0.1
|
|||||||
|
|
||||||
配置始终从 `backend/.env` 读取,环境变量优先。必须先迁移再启动;应用启动只检查数据库版本,不再执行 `create_all`。旧版 `knowledge_items` 会被非破坏性接管;首次迁移前仍须备份。迁移会保留原条目及导入快照,不推测缺失的采集时间。旧版 `verified` 条目因缺少完整审核记录,会转为 `pending`,原状态保存在导入日志中。
|
配置始终从 `backend/.env` 读取,环境变量优先。必须先迁移再启动;应用启动只检查数据库版本,不再执行 `create_all`。旧版 `knowledge_items` 会被非破坏性接管;首次迁移前仍须备份。迁移会保留原条目及导入快照,不推测缺失的采集时间。旧版 `verified` 条目因缺少完整审核记录,会转为 `pending`,原状态保存在导入日志中。
|
||||||
|
|
||||||
认证迁移为 `0004_auth_sessions`。知识管理、来源、文档及检索历史现要求管理员登录;迁移不创建默认账号。运行 `python -m app.manage create-admin --username local_admin`,通过隐藏输入设置独立密码。默认 Cookie 要求 HTTPS;本机 HTTP 联调须在未入库的 `.env` 显式设置 `AUTH_COOKIE_SECURE=false`(仅 development/test 允许)。完整来源、CSRF、字段和权限说明见 [认证接口契约](docs/AUTH_API.md)。当前数据库最新迁移为 `0006_vectors`;文本构建/发布见 [INDEX_API.md](docs/INDEX_API.md),本地模型及向量发布见 [VECTOR_API.md](docs/VECTOR_API.md)。
|
认证迁移为 `0004_auth_sessions`。知识管理、来源、文档及检索历史现要求管理员登录;迁移不创建默认账号。运行 `python -m app.manage create-admin --username local_admin`,通过隐藏输入设置独立密码。默认 Cookie 要求 HTTPS;本机 HTTP 联调须在未入库的 `.env` 显式设置 `AUTH_COOKIE_SECURE=false`(仅 development/test 允许)。完整来源、CSRF、字段和权限说明见 [认证接口契约](docs/AUTH_API.md)。当前数据库最新迁移为 `0005_chunk_indexes`,索引构建/发布说明见 [切分与索引版本接口](docs/INDEX_API.md)。
|
||||||
|
|
||||||
## 启动前端(另开终端)
|
## 启动前端(另开终端)
|
||||||
```powershell
|
```powershell
|
||||||
@@ -102,7 +102,7 @@ Vite 通常使用 http://localhost:5173。
|
|||||||
- `GET /api/retrieval/runs` 使用标准分页参数;`GET /api/retrieval/runs/{run_id}` 读取当时结果。当前知识修改、失效或删除不会改写历史结果。检索记录保存在数据库,运行日志和实际查询不要提交到 Git。
|
- `GET /api/retrieval/runs` 使用标准分页参数;`GET /api/retrieval/runs/{run_id}` 读取当时结果。当前知识修改、失效或删除不会改写历史结果。检索记录保存在数据库,运行日志和实际查询不要提交到 Git。
|
||||||
- 本基线按请求计算语料词频,面向小规模资料集:筛选后的语料最多 5000 条、标题与正文合计最多 200 万字符。超过容量明确返回 503,不截断语料或伪造空结果;可缩小类别范围,大规模索引留到后续实现。
|
- 本基线按请求计算语料词频,面向小规模资料集:筛选后的语料最多 5000 条、标题与正文合计最多 200 万字符。超过容量明确返回 503,不截断语料或伪造空结果;可缩小类别范围,大规模索引留到后续实现。
|
||||||
|
|
||||||
前端 `src/api.ts` 已提供带类型的检索及历史调用;检索页面交互仍由前端阶段接入。chunk 关键词、精确向量与 RRF 的独立调用及比较口径见 [混合检索契约](docs/HYBRID_API.md)。LLM 问答与真实领域评测仍待后续阶段。
|
前端 `src/api.ts` 已提供带类型的检索及历史调用;检索页面交互仍由前端阶段接入。本阶段没有接入 Embedding、向量/混合检索或 LLM,不产生论文实验分数。
|
||||||
|
|
||||||
前端包括问答、知识库管理、来源与更新、实验评估四个页面。实验评估页面没有虚构数据;所有页面明确标记当前功能状态。
|
前端包括问答、知识库管理、来源与更新、实验评估四个页面。实验评估页面没有虚构数据;所有页面明确标记当前功能状态。
|
||||||
|
|
||||||
@@ -119,7 +119,7 @@ pytest -q
|
|||||||
|
|
||||||
测试需要真实 PostgreSQL + pgvector:设置 `TEST_DATABASE_URL`(`postgresql+psycopg://...`),或另行 `pip install pgserver` 使用可选的本机临时实例。`pgserver` 仅用于测试,不是应用运行依赖;外部测试数据库账号需有建 Schema 权限且已启用 vector 扩展。每个测试使用独立 Schema,不会清空业务表。测试使用的占位正文和审核人不代表真实资料。
|
测试需要真实 PostgreSQL + pgvector:设置 `TEST_DATABASE_URL`(`postgresql+psycopg://...`),或另行 `pip install pgserver` 使用可选的本机临时实例。`pgserver` 仅用于测试,不是应用运行依赖;外部测试数据库账号需有建 Schema 权限且已启用 vector 扩展。每个测试使用独立 Schema,不会清空业务表。测试使用的占位正文和审核人不代表真实资料。
|
||||||
|
|
||||||
迁移命令为 `alembic upgrade head`,可重复执行。仅撤回向量快照时,备份后配合旧后端执行 `alembic downgrade 0005_chunk_indexes`,保留文本、账号、知识及检索历史;新模式历史的旧代码兼容边界见 [HYBRID_API.md](docs/HYBRID_API.md)。继续撤回切分/索引版本时,备份后执行 `alembic downgrade 0004_auth_sessions`,保留账号、会话、知识和检索记录。继续撤回认证时,备份后执行 `alembic downgrade 0003_retrieval_runs`,删除用户/会话/预算表,保留知识和检索历史。继续执行 `alembic downgrade 0002_catalog` 还会删除检索历史。需要撤回阶段 2 时,先备份,再执行 `alembic downgrade 0001_baseline`;它保留基础知识字段,但会删除新增来源、文档、审核字段与所有历史记录,软删除的基础条目也会重新可见。降级须配合对应旧版后端,当前应用会拒绝旧 Schema。恢复完整数据应使用备份。不要在有业务数据的数据库执行 `downgrade base`。
|
迁移命令为 `alembic upgrade head`,可重复执行。仅撤回切分/索引版本时,备份后执行 `alembic downgrade 0004_auth_sessions`,保留账号、会话、知识和检索记录。继续撤回认证时,备份后执行 `alembic downgrade 0003_retrieval_runs`,删除用户/会话/预算表,保留知识和检索历史。继续执行 `alembic downgrade 0002_catalog` 还会删除检索历史。需要撤回阶段 2 时,先备份,再执行 `alembic downgrade 0001_baseline`;它保留基础知识字段,但会删除新增来源、文档、审核字段与所有历史记录,软删除的基础条目也会重新可见。降级须配合对应旧版后端,当前应用会拒绝旧 Schema。恢复完整数据应使用备份。不要在有业务数据的数据库执行 `downgrade base`。
|
||||||
|
|
||||||
## 团队协作文档
|
## 团队协作文档
|
||||||
|
|
||||||
|
|||||||
@@ -7,9 +7,3 @@ AUTH_ALLOWED_ORIGINS=http://localhost:5173,http://127.0.0.1:5173,http://localhos
|
|||||||
# 默认要求 HTTPS;仅本机 HTTP 联调可显式改为 false。
|
# 默认要求 HTTPS;仅本机 HTTP 联调可显式改为 false。
|
||||||
AUTH_COOKIE_SECURE=true
|
AUTH_COOKIE_SECURE=true
|
||||||
AUTH_SESSION_SECONDS=28800
|
AUTH_SESSION_SECONDS=28800
|
||||||
# 可选 CPU 模型;先在独立环境准备固定修订缓存,再显式启用。API 永不下载权重。
|
|
||||||
EMBEDDING_ENABLED=false
|
|
||||||
EMBEDDING_MODEL=small
|
|
||||||
EMBEDDING_CACHE_DIR=
|
|
||||||
EMBEDDING_BATCH_SIZE=8
|
|
||||||
EMBEDDING_THREADS=2
|
|
||||||
|
|||||||
@@ -7,23 +7,12 @@ from sqlalchemy.orm import Session
|
|||||||
from app.db.session import get_db
|
from app.db.session import get_db
|
||||||
from app.models import RetrievalRun
|
from app.models import RetrievalRun
|
||||||
from app.schemas.catalog import Page
|
from app.schemas.catalog import Page
|
||||||
from app.schemas.hybrid import (
|
|
||||||
ChunkKeywordRequest,
|
|
||||||
ChunkKeywordResponse,
|
|
||||||
HybridRequest,
|
|
||||||
HybridResponse,
|
|
||||||
)
|
|
||||||
from app.schemas.retrieval import RetrievalRequest, RetrievalResponse
|
from app.schemas.retrieval import RetrievalRequest, RetrievalResponse
|
||||||
from app.schemas.vectors import VectorRetrievalResponse
|
|
||||||
from app.services.auth import require_admin
|
from app.services.auth import require_admin
|
||||||
from app.services.embedding_runtime import get_embedding_provider
|
|
||||||
from app.services.hybrid import retrieve_chunks, retrieve_hybrid
|
|
||||||
from app.services.retrieval import retrieve
|
from app.services.retrieval import retrieve
|
||||||
|
|
||||||
router = APIRouter(prefix="/retrieval", tags=["retrieval"])
|
router = APIRouter(prefix="/retrieval", tags=["retrieval"])
|
||||||
|
|
||||||
RunResponse = RetrievalResponse | VectorRetrievalResponse | ChunkKeywordResponse | HybridResponse
|
|
||||||
|
|
||||||
|
|
||||||
@router.post("", response_model=RetrievalResponse)
|
@router.post("", response_model=RetrievalResponse)
|
||||||
def keyword_retrieval(payload: RetrievalRequest, db: Session = Depends(get_db)):
|
def keyword_retrieval(payload: RetrievalRequest, db: Session = Depends(get_db)):
|
||||||
@@ -31,25 +20,7 @@ def keyword_retrieval(payload: RetrievalRequest, db: Session = Depends(get_db)):
|
|||||||
return retrieve(db, payload)
|
return retrieve(db, payload)
|
||||||
|
|
||||||
|
|
||||||
@router.post("/chunks", response_model=ChunkKeywordResponse)
|
@router.get("/runs", response_model=Page[RetrievalResponse], dependencies=[Depends(require_admin)])
|
||||||
def chunk_keyword_retrieval(payload: ChunkKeywordRequest, db: Session = Depends(get_db)):
|
|
||||||
return retrieve_chunks(db, payload)
|
|
||||||
|
|
||||||
|
|
||||||
@router.post("/hybrid", response_model=HybridResponse)
|
|
||||||
def hybrid_retrieval(
|
|
||||||
payload: HybridRequest,
|
|
||||||
db: Session = Depends(get_db),
|
|
||||||
provider=Depends(get_embedding_provider),
|
|
||||||
):
|
|
||||||
return retrieve_hybrid(db, payload, provider)
|
|
||||||
|
|
||||||
|
|
||||||
@router.get(
|
|
||||||
"/runs",
|
|
||||||
response_model=Page[RunResponse],
|
|
||||||
dependencies=[Depends(require_admin)],
|
|
||||||
)
|
|
||||||
def list_runs(
|
def list_runs(
|
||||||
page: int = Query(1, ge=1),
|
page: int = Query(1, ge=1),
|
||||||
page_size: int = Query(20, ge=1, le=100),
|
page_size: int = Query(20, ge=1, le=100),
|
||||||
@@ -70,9 +41,7 @@ def list_runs(
|
|||||||
|
|
||||||
|
|
||||||
@router.get(
|
@router.get(
|
||||||
"/runs/{run_id}",
|
"/runs/{run_id}", response_model=RetrievalResponse, dependencies=[Depends(require_admin)]
|
||||||
response_model=RunResponse,
|
|
||||||
dependencies=[Depends(require_admin)],
|
|
||||||
)
|
)
|
||||||
def read_run(run_id: UUID, db: Session = Depends(get_db)):
|
def read_run(run_id: UUID, db: Session = Depends(get_db)):
|
||||||
row = db.get(RetrievalRun, str(run_id))
|
row = db.get(RetrievalRun, str(run_id))
|
||||||
|
|||||||
@@ -1,106 +0,0 @@
|
|||||||
from uuid import UUID
|
|
||||||
|
|
||||||
from fastapi import APIRouter, Depends, Query
|
|
||||||
from sqlalchemy import select
|
|
||||||
from sqlalchemy.orm import Session
|
|
||||||
|
|
||||||
from app.db.session import get_db
|
|
||||||
from app.models.vectors import EmbeddingSpace, VectorPublication, VectorVersion
|
|
||||||
from app.schemas.catalog import Page
|
|
||||||
from app.schemas.indexing import PublicationRead, PublishIndex
|
|
||||||
from app.schemas.vectors import (
|
|
||||||
ActiveVectors,
|
|
||||||
BuildVectors,
|
|
||||||
SpaceRead,
|
|
||||||
VectorRead,
|
|
||||||
VectorRequest,
|
|
||||||
VectorRetrievalResponse,
|
|
||||||
)
|
|
||||||
from app.services.auth import CurrentSession, require_admin
|
|
||||||
from app.services.catalog import page_result
|
|
||||||
from app.services.embedding_runtime import get_embedding_provider
|
|
||||||
from app.services.vectors import (
|
|
||||||
active_vectors,
|
|
||||||
build_vectors,
|
|
||||||
publish_vectors,
|
|
||||||
retrieve_vectors,
|
|
||||||
vector_or_404,
|
|
||||||
)
|
|
||||||
|
|
||||||
router = APIRouter(
|
|
||||||
prefix="/vector-indexes", tags=["vector-indexes"], dependencies=[Depends(require_admin)]
|
|
||||||
)
|
|
||||||
retrieval_router = APIRouter(tags=["retrieval"])
|
|
||||||
|
|
||||||
|
|
||||||
@router.post("", response_model=VectorRead, status_code=201)
|
|
||||||
def build(
|
|
||||||
payload: BuildVectors,
|
|
||||||
current: CurrentSession = Depends(require_admin),
|
|
||||||
db: Session = Depends(get_db),
|
|
||||||
provider=Depends(get_embedding_provider),
|
|
||||||
):
|
|
||||||
return build_vectors(db, payload.text_index_version_id, current.user.id, provider)
|
|
||||||
|
|
||||||
|
|
||||||
@router.get("", response_model=Page[VectorRead])
|
|
||||||
def versions(
|
|
||||||
page: int = Query(1, ge=1),
|
|
||||||
page_size: int = Query(20, ge=1, le=100),
|
|
||||||
db: Session = Depends(get_db),
|
|
||||||
):
|
|
||||||
return page_result(
|
|
||||||
db,
|
|
||||||
select(VectorVersion).order_by(VectorVersion.created_at.desc(), VectorVersion.id),
|
|
||||||
page,
|
|
||||||
page_size,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
@router.get("/active", response_model=ActiveVectors)
|
|
||||||
def active(db: Session = Depends(get_db)):
|
|
||||||
return active_vectors(db)
|
|
||||||
|
|
||||||
|
|
||||||
@router.get("/spaces", response_model=Page[SpaceRead])
|
|
||||||
def spaces(
|
|
||||||
page: int = Query(1, ge=1),
|
|
||||||
page_size: int = Query(20, ge=1, le=100),
|
|
||||||
db: Session = Depends(get_db),
|
|
||||||
):
|
|
||||||
return page_result(db, select(EmbeddingSpace).order_by(EmbeddingSpace.id), page, page_size)
|
|
||||||
|
|
||||||
|
|
||||||
@router.get("/publications", response_model=Page[PublicationRead])
|
|
||||||
def publications(
|
|
||||||
page: int = Query(1, ge=1),
|
|
||||||
page_size: int = Query(20, ge=1, le=100),
|
|
||||||
db: Session = Depends(get_db),
|
|
||||||
):
|
|
||||||
return page_result(
|
|
||||||
db, select(VectorPublication).order_by(VectorPublication.generation.desc()), page, page_size
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
@router.get("/{version_id}", response_model=VectorRead)
|
|
||||||
def read_version(version_id: UUID, db: Session = Depends(get_db)):
|
|
||||||
return vector_or_404(db, str(version_id))
|
|
||||||
|
|
||||||
|
|
||||||
@router.post("/{version_id}/publish", response_model=ActiveVectors)
|
|
||||||
def publish(
|
|
||||||
version_id: UUID,
|
|
||||||
payload: PublishIndex,
|
|
||||||
current: CurrentSession = Depends(require_admin),
|
|
||||||
db: Session = Depends(get_db),
|
|
||||||
):
|
|
||||||
return publish_vectors(db, version_id, payload.expected_generation, current.user.id)
|
|
||||||
|
|
||||||
|
|
||||||
@retrieval_router.post("/retrieval/vector", response_model=VectorRetrievalResponse)
|
|
||||||
def search(
|
|
||||||
payload: VectorRequest,
|
|
||||||
db: Session = Depends(get_db),
|
|
||||||
provider=Depends(get_embedding_provider),
|
|
||||||
):
|
|
||||||
return retrieve_vectors(db, payload, provider)
|
|
||||||
@@ -1,8 +1,7 @@
|
|||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Literal
|
|
||||||
from urllib.parse import urlsplit
|
from urllib.parse import urlsplit
|
||||||
|
|
||||||
from pydantic import Field, field_validator, model_validator
|
from pydantic import Field, model_validator
|
||||||
from pydantic_settings import BaseSettings, SettingsConfigDict
|
from pydantic_settings import BaseSettings, SettingsConfigDict
|
||||||
|
|
||||||
|
|
||||||
@@ -17,11 +16,6 @@ class Settings(BaseSettings):
|
|||||||
)
|
)
|
||||||
auth_cookie_secure: bool = True
|
auth_cookie_secure: bool = True
|
||||||
auth_session_seconds: int = Field(28800, ge=300, le=604800)
|
auth_session_seconds: int = Field(28800, ge=300, le=604800)
|
||||||
embedding_enabled: bool = False
|
|
||||||
embedding_model: Literal["small", "base"] = "small"
|
|
||||||
embedding_cache_dir: Path | None = None
|
|
||||||
embedding_batch_size: int = Field(8, ge=1, le=32)
|
|
||||||
embedding_threads: int = Field(2, ge=1, le=4)
|
|
||||||
model_config = SettingsConfigDict(
|
model_config = SettingsConfigDict(
|
||||||
env_file=Path(__file__).resolve().parents[2] / ".env", extra="ignore"
|
env_file=Path(__file__).resolve().parents[2] / ".env", extra="ignore"
|
||||||
)
|
)
|
||||||
@@ -34,11 +28,6 @@ class Settings(BaseSettings):
|
|||||||
def auth_origin_list(self) -> list[str]:
|
def auth_origin_list(self) -> list[str]:
|
||||||
return [x.strip() for x in self.auth_allowed_origins.split(",") if x.strip()]
|
return [x.strip() for x in self.auth_allowed_origins.split(",") if x.strip()]
|
||||||
|
|
||||||
@field_validator("embedding_cache_dir", mode="before")
|
|
||||||
@classmethod
|
|
||||||
def empty_embedding_cache_uses_default(cls, value):
|
|
||||||
return None if isinstance(value, str) and not value.strip() else value
|
|
||||||
|
|
||||||
@model_validator(mode="after")
|
@model_validator(mode="after")
|
||||||
def validate_auth_settings(self):
|
def validate_auth_settings(self):
|
||||||
if not self.auth_cookie_secure and self.app_env not in {"development", "test"}:
|
if not self.auth_cookie_secure and self.app_env not in {"development", "test"}:
|
||||||
|
|||||||
+3
-15
@@ -8,17 +8,7 @@ from sqlalchemy import text
|
|||||||
from sqlalchemy.exc import IntegrityError, OperationalError, ProgrammingError
|
from sqlalchemy.exc import IntegrityError, OperationalError, ProgrammingError
|
||||||
from sqlalchemy.orm.exc import StaleDataError
|
from sqlalchemy.orm.exc import StaleDataError
|
||||||
|
|
||||||
from app.api.routes import (
|
from app.api.routes import auth, chat, documents, health, indexes, knowledge, retrieval, sources
|
||||||
auth,
|
|
||||||
chat,
|
|
||||||
documents,
|
|
||||||
health,
|
|
||||||
indexes,
|
|
||||||
knowledge,
|
|
||||||
retrieval,
|
|
||||||
sources,
|
|
||||||
vectors,
|
|
||||||
)
|
|
||||||
from app.core.config import settings
|
from app.core.config import settings
|
||||||
from app.db.session import engine
|
from app.db.session import engine
|
||||||
from app.services.auth import dummy_hash
|
from app.services.auth import dummy_hash
|
||||||
@@ -31,7 +21,7 @@ def create_app(db_engine=engine) -> FastAPI:
|
|||||||
try:
|
try:
|
||||||
with db_engine.connect() as connection:
|
with db_engine.connect() as connection:
|
||||||
revision = connection.scalar(text("SELECT version_num FROM alembic_version"))
|
revision = connection.scalar(text("SELECT version_num FROM alembic_version"))
|
||||||
if revision != "0006_vectors":
|
if revision != "0005_chunk_indexes":
|
||||||
raise RuntimeError(
|
raise RuntimeError(
|
||||||
"数据库版本不匹配,请先在 backend 目录执行 alembic upgrade head"
|
"数据库版本不匹配,请先在 backend 目录执行 alembic upgrade head"
|
||||||
)
|
)
|
||||||
@@ -46,7 +36,7 @@ def create_app(db_engine=engine) -> FastAPI:
|
|||||||
finally:
|
finally:
|
||||||
db_engine.dispose()
|
db_engine.dispose()
|
||||||
|
|
||||||
app = FastAPI(title=settings.app_name, version="0.6.0", lifespan=lifespan)
|
app = FastAPI(title=settings.app_name, version="0.5.0", lifespan=lifespan)
|
||||||
app.add_middleware(
|
app.add_middleware(
|
||||||
CORSMiddleware,
|
CORSMiddleware,
|
||||||
allow_origins=settings.cors_origin_list,
|
allow_origins=settings.cors_origin_list,
|
||||||
@@ -58,8 +48,6 @@ def create_app(db_engine=engine) -> FastAPI:
|
|||||||
health.router,
|
health.router,
|
||||||
auth.router,
|
auth.router,
|
||||||
indexes.router,
|
indexes.router,
|
||||||
vectors.router,
|
|
||||||
vectors.retrieval_router,
|
|
||||||
chat.router,
|
chat.router,
|
||||||
knowledge.router,
|
knowledge.router,
|
||||||
sources.router,
|
sources.router,
|
||||||
|
|||||||
@@ -3,22 +3,13 @@ from app.models.catalog import AuditEvent, Document, Source
|
|||||||
from app.models.indexing import Chunk, IndexPublication, IndexState, IndexVersion
|
from app.models.indexing import Chunk, IndexPublication, IndexState, IndexVersion
|
||||||
from app.models.knowledge import KnowledgeItem
|
from app.models.knowledge import KnowledgeItem
|
||||||
from app.models.retrieval import RetrievalRun
|
from app.models.retrieval import RetrievalRun
|
||||||
from app.models.vectors import (
|
|
||||||
ChunkVector,
|
|
||||||
EmbeddingSpace,
|
|
||||||
VectorPublication,
|
|
||||||
VectorState,
|
|
||||||
VectorVersion,
|
|
||||||
)
|
|
||||||
|
|
||||||
__all__ = [
|
__all__ = [
|
||||||
"AuditEvent",
|
"AuditEvent",
|
||||||
"AuthRateLimit",
|
"AuthRateLimit",
|
||||||
"AuthSession",
|
"AuthSession",
|
||||||
"Chunk",
|
"Chunk",
|
||||||
"ChunkVector",
|
|
||||||
"Document",
|
"Document",
|
||||||
"EmbeddingSpace",
|
|
||||||
"IndexPublication",
|
"IndexPublication",
|
||||||
"IndexState",
|
"IndexState",
|
||||||
"IndexVersion",
|
"IndexVersion",
|
||||||
@@ -26,7 +17,4 @@ __all__ = [
|
|||||||
"RetrievalRun",
|
"RetrievalRun",
|
||||||
"Source",
|
"Source",
|
||||||
"User",
|
"User",
|
||||||
"VectorPublication",
|
|
||||||
"VectorState",
|
|
||||||
"VectorVersion",
|
|
||||||
]
|
]
|
||||||
|
|||||||
@@ -50,7 +50,6 @@ class Chunk(Base):
|
|||||||
__tablename__ = "chunks"
|
__tablename__ = "chunks"
|
||||||
__table_args__ = (
|
__table_args__ = (
|
||||||
UniqueConstraint("index_version_id", "knowledge_id", "ordinal", name="uq_chunk_position"),
|
UniqueConstraint("index_version_id", "knowledge_id", "ordinal", name="uq_chunk_position"),
|
||||||
UniqueConstraint("id", "index_version_id", name="uq_chunk_text_scope"),
|
|
||||||
CheckConstraint("start_offset >= 0 AND end_offset > start_offset", name="ck_chunk_offsets"),
|
CheckConstraint("start_offset >= 0 AND end_offset > start_offset", name="ck_chunk_offsets"),
|
||||||
)
|
)
|
||||||
id: Mapped[str] = mapped_column(String(36), primary_key=True)
|
id: Mapped[str] = mapped_column(String(36), primary_key=True)
|
||||||
|
|||||||
@@ -1,112 +0,0 @@
|
|||||||
"""Versioned vector spaces, complete snapshots and independent publication history."""
|
|
||||||
|
|
||||||
from datetime import datetime
|
|
||||||
|
|
||||||
from pgvector.sqlalchemy import Vector
|
|
||||||
from sqlalchemy import (
|
|
||||||
JSON,
|
|
||||||
CheckConstraint,
|
|
||||||
DateTime,
|
|
||||||
ForeignKey,
|
|
||||||
ForeignKeyConstraint,
|
|
||||||
Integer,
|
|
||||||
String,
|
|
||||||
UniqueConstraint,
|
|
||||||
func,
|
|
||||||
)
|
|
||||||
from sqlalchemy.dialects.postgresql.base import ischema_names
|
|
||||||
from sqlalchemy.orm import Mapped, mapped_column
|
|
||||||
|
|
||||||
from app.db.session import Base
|
|
||||||
|
|
||||||
# PostgreSQL reflects a qualified name when public is intentionally absent from search_path.
|
|
||||||
# Retain real type comparison during Alembic inspection in isolated schemas.
|
|
||||||
ischema_names.setdefault("public.vector", Vector)
|
|
||||||
|
|
||||||
|
|
||||||
class EmbeddingSpace(Base):
|
|
||||||
__tablename__ = "embedding_spaces"
|
|
||||||
__table_args__ = (
|
|
||||||
UniqueConstraint("id", "dimension", name="uq_embedding_space_dimension"),
|
|
||||||
CheckConstraint("dimension BETWEEN 1 AND 2000", name="ck_embedding_space_dimension"),
|
|
||||||
)
|
|
||||||
id: Mapped[str] = mapped_column(String(64), primary_key=True)
|
|
||||||
dimension: Mapped[int] = mapped_column(Integer, nullable=False)
|
|
||||||
spec: Mapped[dict] = mapped_column(JSON, nullable=False)
|
|
||||||
|
|
||||||
|
|
||||||
class VectorVersion(Base):
|
|
||||||
__tablename__ = "vector_versions"
|
|
||||||
__table_args__ = (
|
|
||||||
ForeignKeyConstraint(
|
|
||||||
["space_id", "dimension"],
|
|
||||||
["embedding_spaces.id", "embedding_spaces.dimension"],
|
|
||||||
name="fk_vector_version_space",
|
|
||||||
),
|
|
||||||
UniqueConstraint(
|
|
||||||
"id", "text_index_version_id", "space_id", "dimension", name="uq_vector_version_scope"
|
|
||||||
),
|
|
||||||
CheckConstraint(
|
|
||||||
"status IN ('ready', 'published', 'superseded')", name="ck_vector_version_status"
|
|
||||||
),
|
|
||||||
CheckConstraint("chunk_count BETWEEN 0 AND 2000", name="ck_vector_version_count"),
|
|
||||||
)
|
|
||||||
id: Mapped[str] = mapped_column(String(36), primary_key=True)
|
|
||||||
text_index_version_id: Mapped[str] = mapped_column(ForeignKey("index_versions.id"))
|
|
||||||
space_id: Mapped[str] = mapped_column(String(64), nullable=False)
|
|
||||||
dimension: Mapped[int] = mapped_column(Integer, nullable=False)
|
|
||||||
chunk_count: Mapped[int] = mapped_column(Integer, nullable=False)
|
|
||||||
status: Mapped[str] = mapped_column(String(20), nullable=False, default="ready")
|
|
||||||
created_by: Mapped[int] = mapped_column(ForeignKey("users.id"))
|
|
||||||
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
|
|
||||||
published_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
|
|
||||||
|
|
||||||
|
|
||||||
class ChunkVector(Base):
|
|
||||||
__tablename__ = "chunk_vectors"
|
|
||||||
__table_args__ = (
|
|
||||||
ForeignKeyConstraint(
|
|
||||||
["version_id", "text_index_version_id", "space_id", "dimension"],
|
|
||||||
[
|
|
||||||
"vector_versions.id",
|
|
||||||
"vector_versions.text_index_version_id",
|
|
||||||
"vector_versions.space_id",
|
|
||||||
"vector_versions.dimension",
|
|
||||||
],
|
|
||||||
name="fk_chunk_vector_scope",
|
|
||||||
),
|
|
||||||
ForeignKeyConstraint(
|
|
||||||
["chunk_id", "text_index_version_id"],
|
|
||||||
["chunks.id", "chunks.index_version_id"],
|
|
||||||
name="fk_chunk_vector_text",
|
|
||||||
),
|
|
||||||
CheckConstraint("public.vector_dims(embedding) = dimension", name="ck_vector_dimension"),
|
|
||||||
CheckConstraint(
|
|
||||||
"public.vector_norm(embedding) BETWEEN 0.9999 AND 1.0001", name="ck_vector_unit_norm"
|
|
||||||
),
|
|
||||||
)
|
|
||||||
version_id: Mapped[str] = mapped_column(String(36), primary_key=True)
|
|
||||||
chunk_id: Mapped[str] = mapped_column(String(36), primary_key=True)
|
|
||||||
text_index_version_id: Mapped[str] = mapped_column(String(36), nullable=False)
|
|
||||||
space_id: Mapped[str] = mapped_column(String(64), nullable=False)
|
|
||||||
dimension: Mapped[int] = mapped_column(Integer, nullable=False)
|
|
||||||
embedding = mapped_column(Vector(), nullable=False)
|
|
||||||
|
|
||||||
|
|
||||||
class VectorState(Base):
|
|
||||||
__tablename__ = "vector_state"
|
|
||||||
__table_args__ = (CheckConstraint("id = 1", name="ck_vector_state_singleton"),)
|
|
||||||
id: Mapped[int] = mapped_column(primary_key=True)
|
|
||||||
active_version_id: Mapped[str | None] = mapped_column(ForeignKey("vector_versions.id"))
|
|
||||||
generation: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
|
||||||
|
|
||||||
|
|
||||||
class VectorPublication(Base):
|
|
||||||
__tablename__ = "vector_publications"
|
|
||||||
generation: Mapped[int] = mapped_column(primary_key=True)
|
|
||||||
previous_version_id: Mapped[str | None] = mapped_column(ForeignKey("vector_versions.id"))
|
|
||||||
version_id: Mapped[str] = mapped_column(ForeignKey("vector_versions.id"), nullable=False)
|
|
||||||
actor_id: Mapped[int] = mapped_column(ForeignKey("users.id"), nullable=False)
|
|
||||||
published_at: Mapped[datetime] = mapped_column(
|
|
||||||
DateTime(timezone=True), server_default=func.now()
|
|
||||||
)
|
|
||||||
@@ -1,95 +0,0 @@
|
|||||||
"""Chunk-level lexical and hybrid contracts; the item-level baseline stays distinct."""
|
|
||||||
|
|
||||||
from datetime import datetime
|
|
||||||
from typing import Literal
|
|
||||||
from uuid import UUID
|
|
||||||
|
|
||||||
from pydantic import BaseModel, Field, model_validator
|
|
||||||
|
|
||||||
from app.schemas.catalog import Category, Input
|
|
||||||
from app.schemas.indexing import IndexConfig
|
|
||||||
from app.schemas.retrieval import RetrievalConfig
|
|
||||||
from app.schemas.vectors import SpaceRead, VectorHit
|
|
||||||
|
|
||||||
|
|
||||||
class ChunkKeywordRequest(Input):
|
|
||||||
query: str = Field(min_length=1, max_length=500)
|
|
||||||
mode: Literal["keyword_chunks"] = "keyword_chunks"
|
|
||||||
top_k: int = Field(5, ge=1, le=20)
|
|
||||||
category: Category | None = None
|
|
||||||
|
|
||||||
|
|
||||||
class HybridRequest(ChunkKeywordRequest):
|
|
||||||
mode: Literal["hybrid"] = "hybrid"
|
|
||||||
candidate_k: int = Field(50, ge=1, le=200)
|
|
||||||
|
|
||||||
@model_validator(mode="after")
|
|
||||||
def sufficient_candidates(self):
|
|
||||||
if self.candidate_k < self.top_k:
|
|
||||||
raise ValueError("candidate_k 不得小于 top_k")
|
|
||||||
return self
|
|
||||||
|
|
||||||
|
|
||||||
class ChunkRetrievalConfig(BaseModel):
|
|
||||||
ranking_unit: Literal["chunk"] = "chunk"
|
|
||||||
lexical: RetrievalConfig
|
|
||||||
chunking: IndexConfig
|
|
||||||
tie_break: Literal["knowledge_id,chunk_ordinal,chunk_id"] = (
|
|
||||||
"knowledge_id,chunk_ordinal,chunk_id"
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
class HybridConfig(ChunkRetrievalConfig):
|
|
||||||
scorer: Literal["rrf-v1"] = "rrf-v1"
|
|
||||||
rrf_k: Literal[60] = 60
|
|
||||||
weights: tuple[Literal[1], Literal[1]] = (1, 1)
|
|
||||||
candidate_k: int
|
|
||||||
vector_scorer: Literal["pgvector-cosine-exact-v1"] = "pgvector-cosine-exact-v1"
|
|
||||||
space: SpaceRead
|
|
||||||
|
|
||||||
|
|
||||||
class RankedCandidate(BaseModel):
|
|
||||||
chunk_id: UUID
|
|
||||||
rank: int
|
|
||||||
score: float
|
|
||||||
|
|
||||||
|
|
||||||
class HybridHit(VectorHit):
|
|
||||||
lexical_rank: int | None
|
|
||||||
lexical_score: float | None
|
|
||||||
vector_rank: int | None
|
|
||||||
vector_score: float | None
|
|
||||||
lexical_contribution: float
|
|
||||||
vector_contribution: float
|
|
||||||
|
|
||||||
|
|
||||||
class ChunkKeywordResponse(BaseModel):
|
|
||||||
run_id: UUID
|
|
||||||
query: str
|
|
||||||
mode: Literal["keyword_chunks"] = "keyword_chunks"
|
|
||||||
status: Literal["ok", "no_evidence"]
|
|
||||||
top_k: int
|
|
||||||
category: str | None
|
|
||||||
query_terms: list[str]
|
|
||||||
text_index_version_id: UUID
|
|
||||||
text_generation: int
|
|
||||||
catalog_revision: int
|
|
||||||
corpus_fingerprint: str
|
|
||||||
corpus_size: int
|
|
||||||
total_hits: int
|
|
||||||
results: list[VectorHit]
|
|
||||||
elapsed_ms: float
|
|
||||||
created_at: datetime
|
|
||||||
config: ChunkRetrievalConfig
|
|
||||||
|
|
||||||
|
|
||||||
class HybridResponse(ChunkKeywordResponse):
|
|
||||||
mode: Literal["hybrid"] = "hybrid"
|
|
||||||
vector_version_id: UUID
|
|
||||||
generation: int
|
|
||||||
lexical_total_hits: int
|
|
||||||
vector_total_hits: int
|
|
||||||
lexical_candidates: list[RankedCandidate]
|
|
||||||
vector_candidates: list[RankedCandidate]
|
|
||||||
results: list[HybridHit]
|
|
||||||
config: HybridConfig
|
|
||||||
@@ -1,88 +0,0 @@
|
|||||||
from datetime import datetime
|
|
||||||
from typing import Literal
|
|
||||||
from uuid import UUID
|
|
||||||
|
|
||||||
from pydantic import BaseModel, ConfigDict, Field
|
|
||||||
|
|
||||||
from app.schemas.catalog import Category, Input
|
|
||||||
from app.schemas.indexing import TimestampRead
|
|
||||||
from app.schemas.retrieval import RetrievalHit
|
|
||||||
|
|
||||||
|
|
||||||
class EmbeddingSpecRead(BaseModel):
|
|
||||||
model: str
|
|
||||||
revision: str
|
|
||||||
dimension: int
|
|
||||||
token_limit: int
|
|
||||||
query_instruction: str
|
|
||||||
implementation: str
|
|
||||||
checkpoint_format: str
|
|
||||||
|
|
||||||
|
|
||||||
class SpaceRead(BaseModel):
|
|
||||||
model_config = ConfigDict(from_attributes=True)
|
|
||||||
id: str
|
|
||||||
dimension: int
|
|
||||||
spec: EmbeddingSpecRead
|
|
||||||
|
|
||||||
|
|
||||||
class BuildVectors(Input):
|
|
||||||
text_index_version_id: UUID
|
|
||||||
|
|
||||||
|
|
||||||
class VectorRead(TimestampRead):
|
|
||||||
id: UUID
|
|
||||||
text_index_version_id: UUID
|
|
||||||
space_id: str
|
|
||||||
dimension: int
|
|
||||||
chunk_count: int
|
|
||||||
status: Literal["ready", "published", "superseded"]
|
|
||||||
created_by: int
|
|
||||||
created_at: datetime
|
|
||||||
published_at: datetime | None
|
|
||||||
|
|
||||||
|
|
||||||
class ActiveVectors(BaseModel):
|
|
||||||
generation: int
|
|
||||||
text_generation: int
|
|
||||||
catalog_revision: int
|
|
||||||
is_current: bool
|
|
||||||
active_version: VectorRead | None
|
|
||||||
|
|
||||||
|
|
||||||
class VectorRequest(Input):
|
|
||||||
query: str = Field(min_length=1, max_length=500)
|
|
||||||
mode: Literal["vector"] = "vector"
|
|
||||||
top_k: int = Field(5, ge=1, le=20)
|
|
||||||
category: Category | None = None
|
|
||||||
|
|
||||||
|
|
||||||
class VectorHit(RetrievalHit):
|
|
||||||
chunk_id: UUID
|
|
||||||
chunk_ordinal: int
|
|
||||||
|
|
||||||
|
|
||||||
class VectorRetrievalConfig(BaseModel):
|
|
||||||
scorer: Literal["pgvector-cosine-exact-v1"] = "pgvector-cosine-exact-v1"
|
|
||||||
ranking_unit: Literal["chunk"] = "chunk"
|
|
||||||
space: SpaceRead
|
|
||||||
|
|
||||||
|
|
||||||
class VectorRetrievalResponse(BaseModel):
|
|
||||||
run_id: UUID
|
|
||||||
query: str
|
|
||||||
mode: Literal["vector"] = "vector"
|
|
||||||
status: Literal["ok", "no_evidence"]
|
|
||||||
top_k: int
|
|
||||||
category: str | None
|
|
||||||
vector_version_id: UUID
|
|
||||||
text_index_version_id: UUID
|
|
||||||
generation: int
|
|
||||||
text_generation: int
|
|
||||||
corpus_fingerprint: str
|
|
||||||
corpus_size: int
|
|
||||||
total_hits: int
|
|
||||||
results: list[VectorHit]
|
|
||||||
elapsed_ms: float
|
|
||||||
created_at: datetime
|
|
||||||
config: VectorRetrievalConfig
|
|
||||||
@@ -1,49 +0,0 @@
|
|||||||
"""Offline, lazy CPU runtime; one encoder call per process, with a fast busy response."""
|
|
||||||
|
|
||||||
from functools import lru_cache
|
|
||||||
from threading import Lock
|
|
||||||
|
|
||||||
from fastapi import HTTPException
|
|
||||||
|
|
||||||
from app.core.config import settings
|
|
||||||
from app.services.embedding import LOCAL_MODELS, EmbeddingSpec, LocalEmbeddingProvider
|
|
||||||
|
|
||||||
|
|
||||||
class EmbeddingRuntime:
|
|
||||||
def __init__(self, spec: EmbeddingSpec, cache_dir, batch_size: int, threads: int):
|
|
||||||
self.spec = spec
|
|
||||||
self.cache_dir, self.batch_size, self.threads = cache_dir, batch_size, threads
|
|
||||||
self._provider = None
|
|
||||||
self._lock = Lock()
|
|
||||||
|
|
||||||
def encode(self, texts: list[str], *, query: bool = False) -> list[list[float]]:
|
|
||||||
if not self._lock.acquire(blocking=False):
|
|
||||||
raise HTTPException(503, "本地编码器忙,请稍后重试")
|
|
||||||
try:
|
|
||||||
if self._provider is None:
|
|
||||||
self._provider = LocalEmbeddingProvider(
|
|
||||||
self.spec,
|
|
||||||
cache_dir=self.cache_dir,
|
|
||||||
batch_size=self.batch_size,
|
|
||||||
threads=self.threads,
|
|
||||||
offline=True,
|
|
||||||
)
|
|
||||||
return self._provider.encode(texts, query=query)
|
|
||||||
finally:
|
|
||||||
self._lock.release()
|
|
||||||
|
|
||||||
|
|
||||||
@lru_cache(maxsize=1)
|
|
||||||
def runtime(spec, cache_dir, batch_size, threads):
|
|
||||||
return EmbeddingRuntime(spec, cache_dir, batch_size, threads)
|
|
||||||
|
|
||||||
|
|
||||||
def get_embedding_provider():
|
|
||||||
if not settings.embedding_enabled:
|
|
||||||
raise HTTPException(503, "本地 Embedding 未启用,请先完成模型缓存和运行环境配置")
|
|
||||||
return runtime(
|
|
||||||
LOCAL_MODELS[settings.embedding_model],
|
|
||||||
settings.embedding_cache_dir,
|
|
||||||
settings.embedding_batch_size,
|
|
||||||
settings.embedding_threads,
|
|
||||||
)
|
|
||||||
@@ -1,205 +0,0 @@
|
|||||||
"""Bounded chunk BM25 and equal-weight reciprocal rank fusion on one publication."""
|
|
||||||
|
|
||||||
from collections import Counter
|
|
||||||
from time import perf_counter
|
|
||||||
from uuid import uuid4
|
|
||||||
|
|
||||||
from fastapi import HTTPException
|
|
||||||
from sqlalchemy import select
|
|
||||||
from sqlalchemy.orm import Session
|
|
||||||
|
|
||||||
from app.models.catalog import utc_now
|
|
||||||
from app.models.indexing import Chunk
|
|
||||||
from app.models.retrieval import RetrievalRun
|
|
||||||
from app.schemas.hybrid import (
|
|
||||||
ChunkKeywordRequest,
|
|
||||||
ChunkKeywordResponse,
|
|
||||||
ChunkRetrievalConfig,
|
|
||||||
HybridConfig,
|
|
||||||
HybridHit,
|
|
||||||
HybridRequest,
|
|
||||||
HybridResponse,
|
|
||||||
RankedCandidate,
|
|
||||||
)
|
|
||||||
from app.schemas.vectors import SpaceRead, VectorHit
|
|
||||||
from app.services.indexing import locked_state
|
|
||||||
from app.services.retrieval import CONFIG, bm25, tokenize
|
|
||||||
from app.services.vectors import (
|
|
||||||
checked_chunks,
|
|
||||||
current_text,
|
|
||||||
encode,
|
|
||||||
rank_vectors,
|
|
||||||
retrieval_scope,
|
|
||||||
)
|
|
||||||
|
|
||||||
RRF_K = 60
|
|
||||||
CHUNK_LEXICAL_CONFIG = CONFIG.model_copy(update={"scorer": "bm25-chunk-body-v1", "title_weight": 0})
|
|
||||||
|
|
||||||
|
|
||||||
def chunk_key(chunk):
|
|
||||||
return chunk.knowledge_id, chunk.ordinal, chunk.id
|
|
||||||
|
|
||||||
|
|
||||||
def chunks_in_scope(db, text_version, category):
|
|
||||||
statement = select(Chunk).where(Chunk.index_version_id == text_version.id)
|
|
||||||
if category is not None:
|
|
||||||
statement = statement.where(Chunk.provenance["category"].as_string() == category)
|
|
||||||
return list(db.scalars(statement.order_by(Chunk.knowledge_id, Chunk.ordinal, Chunk.id)))
|
|
||||||
|
|
||||||
|
|
||||||
def lexical_ranking(chunks, terms):
|
|
||||||
counters = {chunk.id: Counter(tokenize(chunk.text)) for chunk in chunks}
|
|
||||||
scores = bm25(list(counters.values()), terms)
|
|
||||||
ranked = sorted(
|
|
||||||
((chunk, score) for chunk, score in zip(chunks, scores) if score > 0),
|
|
||||||
key=lambda pair: (-pair[1], chunk_key(pair[0])),
|
|
||||||
)
|
|
||||||
return ranked, counters
|
|
||||||
|
|
||||||
|
|
||||||
def candidate_list(rows):
|
|
||||||
return [
|
|
||||||
RankedCandidate(chunk_id=chunk.id, rank=rank, score=score)
|
|
||||||
for rank, (chunk, score) in enumerate(rows, 1)
|
|
||||||
]
|
|
||||||
|
|
||||||
|
|
||||||
def fuse(lexical, vector):
|
|
||||||
"""Ranks start at one; an absent candidate contributes zero, never a padded rank."""
|
|
||||||
scores = {}
|
|
||||||
for leg in (lexical, vector):
|
|
||||||
for candidate in leg:
|
|
||||||
key = str(candidate.chunk_id)
|
|
||||||
scores[key] = scores.get(key, 0.0) + 1.0 / (RRF_K + candidate.rank)
|
|
||||||
return scores
|
|
||||||
|
|
||||||
|
|
||||||
def hit_fields(chunk, terms, counters):
|
|
||||||
return dict(
|
|
||||||
**chunk.provenance,
|
|
||||||
chunk_id=chunk.id,
|
|
||||||
chunk_ordinal=chunk.ordinal,
|
|
||||||
snippet=chunk.text,
|
|
||||||
snippet_start=chunk.start_offset,
|
|
||||||
snippet_end=chunk.end_offset,
|
|
||||||
matched_terms=[term for term in terms if counters[chunk.id][term]],
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def save_run(db, response):
|
|
||||||
db.add(
|
|
||||||
RetrievalRun(
|
|
||||||
id=str(response.run_id),
|
|
||||||
query=response.query,
|
|
||||||
mode=response.mode,
|
|
||||||
corpus_fingerprint=response.corpus_fingerprint,
|
|
||||||
response=response.model_dump(mode="json"),
|
|
||||||
created_at=response.created_at,
|
|
||||||
)
|
|
||||||
)
|
|
||||||
db.commit()
|
|
||||||
return response
|
|
||||||
|
|
||||||
|
|
||||||
def response_fields(payload, state, text_version, chunks, terms, total, started):
|
|
||||||
return dict(
|
|
||||||
run_id=uuid4(),
|
|
||||||
query=payload.query,
|
|
||||||
status="ok" if total else "no_evidence",
|
|
||||||
top_k=payload.top_k,
|
|
||||||
category=payload.category,
|
|
||||||
query_terms=terms,
|
|
||||||
text_index_version_id=text_version.id,
|
|
||||||
text_generation=state.generation,
|
|
||||||
catalog_revision=state.catalog_revision,
|
|
||||||
corpus_fingerprint=text_version.corpus_fingerprint,
|
|
||||||
corpus_size=len(chunks),
|
|
||||||
total_hits=total,
|
|
||||||
elapsed_ms=round((perf_counter() - started) * 1000, 3),
|
|
||||||
created_at=utc_now(),
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def retrieve_chunks(db: Session, payload: ChunkKeywordRequest):
|
|
||||||
started = perf_counter()
|
|
||||||
terms = list(dict.fromkeys(tokenize(payload.query)))
|
|
||||||
if not terms:
|
|
||||||
raise HTTPException(422, "问题中没有可检索词,请补充具体关键词")
|
|
||||||
state = locked_state(db)
|
|
||||||
text_version, corpus = current_text(db, state)
|
|
||||||
checked_chunks(db, text_version, corpus)
|
|
||||||
chunks = chunks_in_scope(db, text_version, payload.category)
|
|
||||||
ranked, counters = lexical_ranking(chunks, terms)
|
|
||||||
hits = [
|
|
||||||
VectorHit(**hit_fields(chunk, terms, counters), rank=rank, score=score)
|
|
||||||
for rank, (chunk, score) in enumerate(ranked[: payload.top_k], 1)
|
|
||||||
]
|
|
||||||
return save_run(
|
|
||||||
db,
|
|
||||||
ChunkKeywordResponse(
|
|
||||||
**response_fields(payload, state, text_version, chunks, terms, len(ranked), started),
|
|
||||||
results=hits,
|
|
||||||
config=ChunkRetrievalConfig(lexical=CHUNK_LEXICAL_CONFIG, chunking=text_version.config),
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def retrieve_hybrid(db: Session, payload: HybridRequest, provider):
|
|
||||||
started = perf_counter()
|
|
||||||
terms = list(dict.fromkeys(tokenize(payload.query)))
|
|
||||||
*_, token = retrieval_scope(db, provider)
|
|
||||||
db.commit()
|
|
||||||
query_vector = encode(provider, [payload.query], query=True)[0]
|
|
||||||
state, vector_state, text_version, version, space, current = retrieval_scope(db, provider)
|
|
||||||
if token != current:
|
|
||||||
raise HTTPException(409, "查询编码期间发布版本或审核依据已变化,请重试")
|
|
||||||
# Hold the shared catalog/publication locks until both rankings and the run are saved.
|
|
||||||
chunks = chunks_in_scope(db, text_version, payload.category)
|
|
||||||
ranked, counters = lexical_ranking(chunks, terms)
|
|
||||||
lexical = candidate_list(ranked[: payload.candidate_k])
|
|
||||||
vector_total, vector_rows = rank_vectors(
|
|
||||||
db, version, space, payload.category, payload.candidate_k, query_vector
|
|
||||||
)
|
|
||||||
vector = candidate_list(
|
|
||||||
[(chunk, max(-1.0, min(1.0, 1.0 - float(distance)))) for chunk, distance in vector_rows]
|
|
||||||
)
|
|
||||||
scores = fuse(lexical, vector)
|
|
||||||
by_id = {chunk.id: chunk for chunk in chunks}
|
|
||||||
ordered = sorted(scores, key=lambda key: (-scores[key], chunk_key(by_id[key])))
|
|
||||||
lexical_by_id = {str(c.chunk_id): c for c in lexical}
|
|
||||||
vector_by_id = {str(c.chunk_id): c for c in vector}
|
|
||||||
hits = []
|
|
||||||
for rank, key in enumerate(ordered[: payload.top_k], 1):
|
|
||||||
left, right = lexical_by_id.get(key), vector_by_id.get(key)
|
|
||||||
hits.append(
|
|
||||||
HybridHit(
|
|
||||||
**hit_fields(by_id[key], terms, counters),
|
|
||||||
rank=rank,
|
|
||||||
score=scores[key],
|
|
||||||
lexical_rank=left.rank if left else None,
|
|
||||||
lexical_score=left.score if left else None,
|
|
||||||
vector_rank=right.rank if right else None,
|
|
||||||
vector_score=right.score if right else None,
|
|
||||||
lexical_contribution=1.0 / (RRF_K + left.rank) if left else 0.0,
|
|
||||||
vector_contribution=1.0 / (RRF_K + right.rank) if right else 0.0,
|
|
||||||
)
|
|
||||||
)
|
|
||||||
return save_run(
|
|
||||||
db,
|
|
||||||
HybridResponse(
|
|
||||||
**response_fields(payload, state, text_version, chunks, terms, len(scores), started),
|
|
||||||
vector_version_id=version.id,
|
|
||||||
generation=vector_state.generation,
|
|
||||||
lexical_total_hits=len(ranked),
|
|
||||||
vector_total_hits=vector_total,
|
|
||||||
lexical_candidates=lexical,
|
|
||||||
vector_candidates=vector,
|
|
||||||
results=hits,
|
|
||||||
config=HybridConfig(
|
|
||||||
lexical=CHUNK_LEXICAL_CONFIG,
|
|
||||||
chunking=text_version.config,
|
|
||||||
candidate_k=payload.candidate_k,
|
|
||||||
space=SpaceRead.model_validate(space),
|
|
||||||
),
|
|
||||||
),
|
|
||||||
)
|
|
||||||
@@ -1,356 +0,0 @@
|
|||||||
"""Generate outside publication locks; recheck reviewed evidence before persistence/query."""
|
|
||||||
|
|
||||||
import hashlib
|
|
||||||
from dataclasses import asdict
|
|
||||||
from time import perf_counter
|
|
||||||
from uuid import UUID, uuid4
|
|
||||||
|
|
||||||
from fastapi import HTTPException
|
|
||||||
from sqlalchemy import Float, func, select
|
|
||||||
from sqlalchemy.orm import Session
|
|
||||||
|
|
||||||
from app.models.catalog import utc_now
|
|
||||||
from app.models.indexing import Chunk, IndexVersion
|
|
||||||
from app.models.retrieval import RetrievalRun
|
|
||||||
from app.models.vectors import (
|
|
||||||
ChunkVector,
|
|
||||||
EmbeddingSpace,
|
|
||||||
VectorPublication,
|
|
||||||
VectorState,
|
|
||||||
VectorVersion,
|
|
||||||
)
|
|
||||||
from app.schemas.vectors import (
|
|
||||||
ActiveVectors,
|
|
||||||
SpaceRead,
|
|
||||||
VectorHit,
|
|
||||||
VectorRead,
|
|
||||||
VectorRequest,
|
|
||||||
VectorRetrievalConfig,
|
|
||||||
VectorRetrievalResponse,
|
|
||||||
)
|
|
||||||
from app.services.embedding import EmbeddingError, EmbeddingProvider, validate_vectors
|
|
||||||
from app.services.indexing import locked_state, split_text
|
|
||||||
from app.services.retrieval import corpus_fingerprint, load_corpus
|
|
||||||
|
|
||||||
MAX_VECTOR_CHUNKS = 2000
|
|
||||||
|
|
||||||
|
|
||||||
def vector_state(db: Session) -> VectorState:
|
|
||||||
# Always acquire index_state first: the same ordering as catalog/text publication.
|
|
||||||
state = db.scalar(
|
|
||||||
select(VectorState)
|
|
||||||
.where(VectorState.id == 1)
|
|
||||||
.with_for_update()
|
|
||||||
.execution_options(populate_existing=True)
|
|
||||||
)
|
|
||||||
if state is None:
|
|
||||||
raise HTTPException(503, "向量状态未初始化,请检查迁移")
|
|
||||||
return state
|
|
||||||
|
|
||||||
|
|
||||||
def vector_or_404(db: Session, version_id: str) -> VectorVersion:
|
|
||||||
version = db.get(VectorVersion, version_id)
|
|
||||||
if version is None:
|
|
||||||
raise HTTPException(404, "向量版本不存在")
|
|
||||||
return version
|
|
||||||
|
|
||||||
|
|
||||||
def current_text(db: Session, state):
|
|
||||||
version = db.get(IndexVersion, state.active_version_id) if state.active_version_id else None
|
|
||||||
if version is None or version.catalog_revision != state.catalog_revision:
|
|
||||||
raise HTTPException(409, "尚无当前有效的文本发布版本,请构建并发布")
|
|
||||||
corpus = load_corpus(db, None)
|
|
||||||
if version.corpus_fingerprint != corpus_fingerprint(corpus):
|
|
||||||
raise HTTPException(409, "审核依据已变化,请重新构建并发布文本版本")
|
|
||||||
return version, corpus
|
|
||||||
|
|
||||||
|
|
||||||
def encode(provider: EmbeddingProvider, texts: list[str], *, query=False):
|
|
||||||
try:
|
|
||||||
spec = provider.spec
|
|
||||||
result = provider.encode(texts, query=query)
|
|
||||||
if provider.spec != spec:
|
|
||||||
raise EmbeddingError("编码期间模型身份改变")
|
|
||||||
validate_vectors(result, len(texts), spec)
|
|
||||||
return result
|
|
||||||
except HTTPException:
|
|
||||||
raise
|
|
||||||
except EmbeddingError as exc:
|
|
||||||
raise HTTPException(422, "编码校验失败,请核对文本 token 上限、向量维度和归一化") from exc
|
|
||||||
except Exception as exc:
|
|
||||||
# Optional dependency/cache failures must not expose server paths or library tracebacks.
|
|
||||||
raise HTTPException(503, "本地模型不可用,请检查可选运行依赖和固定版本缓存") from exc
|
|
||||||
|
|
||||||
|
|
||||||
def active_vectors(db: Session) -> ActiveVectors:
|
|
||||||
state = locked_state(db)
|
|
||||||
vector = vector_state(db)
|
|
||||||
version = db.get(VectorVersion, vector.active_version_id) if vector.active_version_id else None
|
|
||||||
current = False
|
|
||||||
if version and version.text_index_version_id == state.active_version_id:
|
|
||||||
try:
|
|
||||||
text_version, corpus = current_text(db, state)
|
|
||||||
checked_chunks(db, text_version, corpus)
|
|
||||||
complete(db, version)
|
|
||||||
current = version.chunk_count == text_version.chunk_count
|
|
||||||
except HTTPException as exc:
|
|
||||||
if exc.status_code != 409:
|
|
||||||
raise
|
|
||||||
return ActiveVectors(
|
|
||||||
generation=vector.generation,
|
|
||||||
text_generation=state.generation,
|
|
||||||
catalog_revision=state.catalog_revision,
|
|
||||||
is_current=current,
|
|
||||||
active_version=VectorRead.model_validate(version) if version else None,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def checked_chunks(db, text_version, corpus):
|
|
||||||
if text_version.chunk_count > MAX_VECTOR_CHUNKS:
|
|
||||||
raise HTTPException(503, "超过同步向量构建容量(2000 chunks)")
|
|
||||||
chunks = list(
|
|
||||||
db.scalars(
|
|
||||||
select(Chunk)
|
|
||||||
.where(Chunk.index_version_id == text_version.id)
|
|
||||||
.order_by(Chunk.knowledge_id, Chunk.ordinal)
|
|
||||||
.limit(MAX_VECTOR_CHUNKS + 1)
|
|
||||||
)
|
|
||||||
)
|
|
||||||
if len(chunks) != text_version.chunk_count:
|
|
||||||
raise HTTPException(409, "文本切分快照不完整,请重新构建")
|
|
||||||
expected = [
|
|
||||||
(
|
|
||||||
item["knowledge_id"],
|
|
||||||
ordinal,
|
|
||||||
start,
|
|
||||||
end,
|
|
||||||
body,
|
|
||||||
hashlib.sha256(body.encode()).hexdigest(),
|
|
||||||
{k: v for k, v in item.items() if k != "content"},
|
|
||||||
)
|
|
||||||
for item in corpus
|
|
||||||
for ordinal, (start, end, body) in enumerate(
|
|
||||||
split_text(
|
|
||||||
item["content"], text_version.config["chunk_size"], text_version.config["overlap"]
|
|
||||||
)
|
|
||||||
)
|
|
||||||
]
|
|
||||||
actual = [
|
|
||||||
(
|
|
||||||
c.knowledge_id,
|
|
||||||
c.ordinal,
|
|
||||||
c.start_offset,
|
|
||||||
c.end_offset,
|
|
||||||
c.text,
|
|
||||||
c.text_sha256,
|
|
||||||
c.provenance,
|
|
||||||
)
|
|
||||||
for c in chunks
|
|
||||||
]
|
|
||||||
if actual != expected:
|
|
||||||
raise HTTPException(409, "文本切分与审核语料不一致,请重新构建")
|
|
||||||
return [(c.id, c.text) for c in chunks]
|
|
||||||
|
|
||||||
|
|
||||||
def build_vectors(db: Session, text_version_id: UUID, actor_id: int, provider: EmbeddingProvider):
|
|
||||||
state = locked_state(db)
|
|
||||||
text_version, corpus = current_text(db, state)
|
|
||||||
if str(text_version_id) != text_version.id:
|
|
||||||
raise HTTPException(409, "只能对当前已发布的文本版本构建向量")
|
|
||||||
inputs = checked_chunks(db, text_version, corpus)
|
|
||||||
token = (state.generation, state.catalog_revision, text_version.id)
|
|
||||||
spec = provider.spec
|
|
||||||
db.commit() # No publication/catalog row locks or open read transaction during model work.
|
|
||||||
vectors = encode(provider, [body for _, body in inputs]) if inputs else []
|
|
||||||
if provider.spec != spec:
|
|
||||||
raise HTTPException(409, "模型身份已变化,请重试")
|
|
||||||
state = locked_state(db)
|
|
||||||
new_text, new_corpus = current_text(db, state)
|
|
||||||
if token != (state.generation, state.catalog_revision, state.active_version_id):
|
|
||||||
raise HTTPException(409, "编码期间文本或审核版本已变化,请重新构建")
|
|
||||||
if checked_chunks(db, new_text, new_corpus) != inputs:
|
|
||||||
raise HTTPException(409, "编码期间文本切分快照已变化,请重新构建")
|
|
||||||
space = db.get(EmbeddingSpace, spec.space_id)
|
|
||||||
if space is None:
|
|
||||||
db.add(EmbeddingSpace(id=spec.space_id, dimension=spec.dimension, spec=asdict(spec)))
|
|
||||||
db.flush()
|
|
||||||
elif space.spec != asdict(spec) or space.dimension != spec.dimension:
|
|
||||||
raise HTTPException(409, "已保存向量空间配置不一致")
|
|
||||||
version = VectorVersion(
|
|
||||||
id=str(uuid4()),
|
|
||||||
text_index_version_id=str(text_version_id),
|
|
||||||
space_id=spec.space_id,
|
|
||||||
dimension=spec.dimension,
|
|
||||||
chunk_count=len(inputs),
|
|
||||||
created_by=actor_id,
|
|
||||||
status="ready",
|
|
||||||
)
|
|
||||||
db.add(version)
|
|
||||||
db.flush()
|
|
||||||
db.add_all(
|
|
||||||
[
|
|
||||||
ChunkVector(
|
|
||||||
version_id=version.id,
|
|
||||||
chunk_id=chunk_id,
|
|
||||||
text_index_version_id=version.text_index_version_id,
|
|
||||||
space_id=spec.space_id,
|
|
||||||
dimension=spec.dimension,
|
|
||||||
embedding=embedding,
|
|
||||||
)
|
|
||||||
for (chunk_id, _), embedding in zip(inputs, vectors, strict=True)
|
|
||||||
]
|
|
||||||
)
|
|
||||||
result = VectorRead.model_validate(version)
|
|
||||||
db.commit()
|
|
||||||
return result
|
|
||||||
|
|
||||||
|
|
||||||
def complete(db, version):
|
|
||||||
count = db.scalar(
|
|
||||||
select(func.count()).select_from(ChunkVector).where(ChunkVector.version_id == version.id)
|
|
||||||
)
|
|
||||||
if count != version.chunk_count:
|
|
||||||
raise HTTPException(409, "向量快照不完整,请重新构建")
|
|
||||||
|
|
||||||
|
|
||||||
def publish_vectors(db: Session, version_id: UUID, expected_generation: int, actor_id: int):
|
|
||||||
state = locked_state(db)
|
|
||||||
vector = vector_state(db)
|
|
||||||
if vector.generation != expected_generation:
|
|
||||||
raise HTTPException(409, "向量发布代数已变化,请刷新后重试")
|
|
||||||
version = vector_or_404(db, str(version_id))
|
|
||||||
text_version, corpus = current_text(db, state)
|
|
||||||
if version.text_index_version_id != text_version.id:
|
|
||||||
raise HTTPException(409, "该向量版本不属于当前文本发布版本")
|
|
||||||
checked_chunks(db, text_version, corpus)
|
|
||||||
complete(db, version)
|
|
||||||
if version.chunk_count != text_version.chunk_count:
|
|
||||||
raise HTTPException(409, "向量与文本切分数量不一致")
|
|
||||||
if vector.active_version_id == version.id:
|
|
||||||
return active_vectors(db)
|
|
||||||
previous = vector.active_version_id
|
|
||||||
if previous:
|
|
||||||
db.get(VectorVersion, previous).status = "superseded"
|
|
||||||
vector.active_version_id = version.id
|
|
||||||
vector.generation += 1
|
|
||||||
version.status, version.published_at = "published", utc_now()
|
|
||||||
db.add(
|
|
||||||
VectorPublication(
|
|
||||||
generation=vector.generation,
|
|
||||||
previous_version_id=previous,
|
|
||||||
version_id=version.id,
|
|
||||||
actor_id=actor_id,
|
|
||||||
published_at=version.published_at,
|
|
||||||
)
|
|
||||||
)
|
|
||||||
result = ActiveVectors(
|
|
||||||
generation=vector.generation,
|
|
||||||
text_generation=state.generation,
|
|
||||||
catalog_revision=state.catalog_revision,
|
|
||||||
is_current=True,
|
|
||||||
active_version=VectorRead.model_validate(version),
|
|
||||||
)
|
|
||||||
db.commit()
|
|
||||||
return result
|
|
||||||
|
|
||||||
|
|
||||||
def retrieval_scope(db, provider):
|
|
||||||
state = locked_state(db)
|
|
||||||
vector = vector_state(db)
|
|
||||||
text_version, corpus = current_text(db, state)
|
|
||||||
version = db.get(VectorVersion, vector.active_version_id) if vector.active_version_id else None
|
|
||||||
if version is None or version.text_index_version_id != text_version.id:
|
|
||||||
raise HTTPException(409, "尚无匹配当前文本版本的向量发布,请先构建并发布")
|
|
||||||
checked_chunks(db, text_version, corpus)
|
|
||||||
space = db.get(EmbeddingSpace, version.space_id)
|
|
||||||
if (
|
|
||||||
space.id != provider.spec.space_id
|
|
||||||
or space.spec != asdict(provider.spec)
|
|
||||||
or space.dimension != provider.spec.dimension
|
|
||||||
or version.dimension != provider.spec.dimension
|
|
||||||
):
|
|
||||||
raise HTTPException(
|
|
||||||
409, "当前编码器与已发布向量空间不一致,请使用同一配置重建或切换运行配置"
|
|
||||||
)
|
|
||||||
complete(db, version)
|
|
||||||
if version.chunk_count != text_version.chunk_count:
|
|
||||||
raise HTTPException(409, "向量与文本切分数量不一致")
|
|
||||||
token = (state.generation, state.catalog_revision, vector.generation, version.id)
|
|
||||||
return state, vector, text_version, version, space, token
|
|
||||||
|
|
||||||
|
|
||||||
def retrieve_vectors(db: Session, payload: VectorRequest, provider: EmbeddingProvider):
|
|
||||||
started = perf_counter()
|
|
||||||
*_, token = retrieval_scope(db, provider)
|
|
||||||
db.commit()
|
|
||||||
query_vector = encode(provider, [payload.query], query=True)[0]
|
|
||||||
state, vector, text_version, version, space, current = retrieval_scope(db, provider)
|
|
||||||
if token != current:
|
|
||||||
raise HTTPException(409, "查询编码期间发布版本或审核依据已变化,请重试")
|
|
||||||
total, rows = rank_vectors(db, version, space, payload.category, payload.top_k, query_vector)
|
|
||||||
hits = [
|
|
||||||
VectorHit(
|
|
||||||
**chunk.provenance,
|
|
||||||
rank=rank,
|
|
||||||
chunk_id=chunk.id,
|
|
||||||
chunk_ordinal=chunk.ordinal,
|
|
||||||
snippet=chunk.text,
|
|
||||||
snippet_start=chunk.start_offset,
|
|
||||||
snippet_end=chunk.end_offset,
|
|
||||||
score=round(max(-1.0, min(1.0, 1.0 - float(dist))), 8),
|
|
||||||
matched_terms=[],
|
|
||||||
)
|
|
||||||
for rank, (chunk, dist) in enumerate(rows, 1)
|
|
||||||
]
|
|
||||||
result = VectorRetrievalResponse(
|
|
||||||
run_id=uuid4(),
|
|
||||||
query=payload.query,
|
|
||||||
status="ok" if hits else "no_evidence",
|
|
||||||
top_k=payload.top_k,
|
|
||||||
category=payload.category,
|
|
||||||
vector_version_id=version.id,
|
|
||||||
text_index_version_id=text_version.id,
|
|
||||||
generation=vector.generation,
|
|
||||||
text_generation=state.generation,
|
|
||||||
corpus_fingerprint=text_version.corpus_fingerprint,
|
|
||||||
corpus_size=total,
|
|
||||||
total_hits=total,
|
|
||||||
results=hits,
|
|
||||||
elapsed_ms=round((perf_counter() - started) * 1000, 3),
|
|
||||||
created_at=utc_now(),
|
|
||||||
config=VectorRetrievalConfig(space=SpaceRead.model_validate(space)),
|
|
||||||
)
|
|
||||||
db.add(
|
|
||||||
RetrievalRun(
|
|
||||||
id=str(result.run_id),
|
|
||||||
query=result.query,
|
|
||||||
mode="vector",
|
|
||||||
corpus_fingerprint=result.corpus_fingerprint,
|
|
||||||
response=result.model_dump(mode="json"),
|
|
||||||
created_at=result.created_at,
|
|
||||||
)
|
|
||||||
)
|
|
||||||
db.commit()
|
|
||||||
return result
|
|
||||||
|
|
||||||
|
|
||||||
def rank_vectors(db, version, space, category, limit, query_vector):
|
|
||||||
# Extension-qualified operator works even with an isolated business schema search_path.
|
|
||||||
distance = ChunkVector.embedding.op("OPERATOR(public.<=>)", return_type=Float)(query_vector)
|
|
||||||
statement = (
|
|
||||||
select(Chunk, distance.label("distance"))
|
|
||||||
.join(ChunkVector, ChunkVector.chunk_id == Chunk.id)
|
|
||||||
.where(
|
|
||||||
ChunkVector.version_id == version.id,
|
|
||||||
ChunkVector.space_id == space.id,
|
|
||||||
ChunkVector.text_index_version_id == version.text_index_version_id,
|
|
||||||
ChunkVector.dimension == space.dimension,
|
|
||||||
)
|
|
||||||
)
|
|
||||||
if category is not None:
|
|
||||||
statement = statement.where(Chunk.provenance["category"].as_string() == category)
|
|
||||||
total = db.scalar(select(func.count()).select_from(statement.subquery()))
|
|
||||||
rows = db.execute(
|
|
||||||
statement.order_by(distance, Chunk.knowledge_id, Chunk.ordinal, Chunk.id).limit(limit)
|
|
||||||
)
|
|
||||||
return total, list(rows)
|
|
||||||
@@ -1,122 +0,0 @@
|
|||||||
"""Persist dimension-checked vector snapshots with composite scope constraints."""
|
|
||||||
|
|
||||||
import sqlalchemy as sa
|
|
||||||
from alembic import op
|
|
||||||
|
|
||||||
revision = "0006_vectors"
|
|
||||||
down_revision = "0005_chunk_indexes"
|
|
||||||
branch_labels = None
|
|
||||||
depends_on = None
|
|
||||||
|
|
||||||
|
|
||||||
class ExtensionVector(sa.types.UserDefinedType):
|
|
||||||
"""Qualify extension type without adding public business tables to search_path."""
|
|
||||||
|
|
||||||
def get_col_spec(self, **kw):
|
|
||||||
return "public.vector"
|
|
||||||
|
|
||||||
|
|
||||||
def upgrade():
|
|
||||||
schema = op.get_bind().scalar(
|
|
||||||
sa.text("SELECT extnamespace::regnamespace::text FROM pg_extension WHERE extname='vector'")
|
|
||||||
)
|
|
||||||
if schema != "public":
|
|
||||||
raise RuntimeError("vector 扩展须安装在 public;请由数据库管理员核对,迁移不会移动共享扩展")
|
|
||||||
op.create_unique_constraint("uq_chunk_text_scope", "chunks", ["id", "index_version_id"])
|
|
||||||
op.create_table(
|
|
||||||
"embedding_spaces",
|
|
||||||
sa.Column("id", sa.String(64), primary_key=True),
|
|
||||||
sa.Column("dimension", sa.Integer(), nullable=False),
|
|
||||||
sa.Column("spec", sa.JSON(), nullable=False),
|
|
||||||
sa.UniqueConstraint("id", "dimension", name="uq_embedding_space_dimension"),
|
|
||||||
sa.CheckConstraint("dimension BETWEEN 1 AND 2000", name="ck_embedding_space_dimension"),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"vector_versions",
|
|
||||||
sa.Column("id", sa.String(36), primary_key=True),
|
|
||||||
sa.Column(
|
|
||||||
"text_index_version_id",
|
|
||||||
sa.String(36),
|
|
||||||
sa.ForeignKey("index_versions.id"),
|
|
||||||
nullable=False,
|
|
||||||
),
|
|
||||||
sa.Column("space_id", sa.String(64), nullable=False),
|
|
||||||
sa.Column("dimension", sa.Integer(), nullable=False),
|
|
||||||
sa.Column("chunk_count", sa.Integer(), nullable=False),
|
|
||||||
sa.Column("status", sa.String(20), nullable=False),
|
|
||||||
sa.Column("created_by", sa.Integer(), sa.ForeignKey("users.id"), nullable=False),
|
|
||||||
sa.Column(
|
|
||||||
"created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False
|
|
||||||
),
|
|
||||||
sa.Column("published_at", sa.DateTime(timezone=True)),
|
|
||||||
sa.ForeignKeyConstraint(
|
|
||||||
["space_id", "dimension"],
|
|
||||||
["embedding_spaces.id", "embedding_spaces.dimension"],
|
|
||||||
name="fk_vector_version_space",
|
|
||||||
),
|
|
||||||
sa.UniqueConstraint(
|
|
||||||
"id", "text_index_version_id", "space_id", "dimension", name="uq_vector_version_scope"
|
|
||||||
),
|
|
||||||
sa.CheckConstraint(
|
|
||||||
"status IN ('ready', 'published', 'superseded')", name="ck_vector_version_status"
|
|
||||||
),
|
|
||||||
sa.CheckConstraint("chunk_count BETWEEN 0 AND 2000", name="ck_vector_version_count"),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"chunk_vectors",
|
|
||||||
sa.Column("version_id", sa.String(36), primary_key=True),
|
|
||||||
sa.Column("chunk_id", sa.String(36), primary_key=True),
|
|
||||||
sa.Column("text_index_version_id", sa.String(36), nullable=False),
|
|
||||||
sa.Column("space_id", sa.String(64), nullable=False),
|
|
||||||
sa.Column("dimension", sa.Integer(), nullable=False),
|
|
||||||
sa.Column("embedding", ExtensionVector(), nullable=False),
|
|
||||||
sa.ForeignKeyConstraint(
|
|
||||||
["version_id", "text_index_version_id", "space_id", "dimension"],
|
|
||||||
[
|
|
||||||
"vector_versions.id",
|
|
||||||
"vector_versions.text_index_version_id",
|
|
||||||
"vector_versions.space_id",
|
|
||||||
"vector_versions.dimension",
|
|
||||||
],
|
|
||||||
name="fk_chunk_vector_scope",
|
|
||||||
),
|
|
||||||
sa.ForeignKeyConstraint(
|
|
||||||
["chunk_id", "text_index_version_id"],
|
|
||||||
["chunks.id", "chunks.index_version_id"],
|
|
||||||
name="fk_chunk_vector_text",
|
|
||||||
),
|
|
||||||
sa.CheckConstraint("public.vector_dims(embedding) = dimension", name="ck_vector_dimension"),
|
|
||||||
sa.CheckConstraint(
|
|
||||||
"public.vector_norm(embedding) BETWEEN 0.9999 AND 1.0001", name="ck_vector_unit_norm"
|
|
||||||
),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"vector_state",
|
|
||||||
sa.Column("id", sa.Integer(), primary_key=True),
|
|
||||||
sa.Column("active_version_id", sa.String(36), sa.ForeignKey("vector_versions.id")),
|
|
||||||
sa.Column("generation", sa.Integer(), nullable=False),
|
|
||||||
sa.CheckConstraint("id = 1", name="ck_vector_state_singleton"),
|
|
||||||
)
|
|
||||||
op.execute(sa.text("INSERT INTO vector_state (id, generation) VALUES (1, 0)"))
|
|
||||||
op.create_table(
|
|
||||||
"vector_publications",
|
|
||||||
sa.Column("generation", sa.Integer(), primary_key=True),
|
|
||||||
sa.Column("previous_version_id", sa.String(36), sa.ForeignKey("vector_versions.id")),
|
|
||||||
sa.Column("version_id", sa.String(36), sa.ForeignKey("vector_versions.id"), nullable=False),
|
|
||||||
sa.Column("actor_id", sa.Integer(), sa.ForeignKey("users.id"), nullable=False),
|
|
||||||
sa.Column(
|
|
||||||
"published_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def downgrade():
|
|
||||||
for table in [
|
|
||||||
"vector_publications",
|
|
||||||
"vector_state",
|
|
||||||
"chunk_vectors",
|
|
||||||
"vector_versions",
|
|
||||||
"embedding_spaces",
|
|
||||||
]:
|
|
||||||
op.drop_table(table)
|
|
||||||
op.drop_constraint("uq_chunk_text_scope", "chunks", type_="unique")
|
|
||||||
@@ -1,306 +0,0 @@
|
|||||||
"""Synthetic coordinates/evidence test fusion semantics, not domain/model quality."""
|
|
||||||
|
|
||||||
import math
|
|
||||||
from dataclasses import replace
|
|
||||||
from uuid import UUID, uuid4
|
|
||||||
|
|
||||||
import pytest
|
|
||||||
from sqlalchemy import func, select, text
|
|
||||||
from sqlalchemy.orm import Session
|
|
||||||
from test_indexing import build
|
|
||||||
from test_retrieval import add_item
|
|
||||||
from test_vectors import ToyProvider, text_version, vector_build, vector_publish
|
|
||||||
|
|
||||||
from app.models.retrieval import RetrievalRun
|
|
||||||
from app.schemas.hybrid import RankedCandidate
|
|
||||||
from app.services import vectors
|
|
||||||
from app.services.embedding import EmbeddingError
|
|
||||||
from app.services.embedding_runtime import get_embedding_provider
|
|
||||||
from app.services.hybrid import fuse
|
|
||||||
from app.services.indexing import publish_index
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
|
||||||
def provider(client):
|
|
||||||
value = ToyProvider()
|
|
||||||
client.app.dependency_overrides[get_embedding_provider] = lambda: value
|
|
||||||
return value
|
|
||||||
|
|
||||||
|
|
||||||
def search(client, path="hybrid", **kwargs):
|
|
||||||
return client.post(f"/api/retrieval/{path}", json={"query": "query", **kwargs})
|
|
||||||
|
|
||||||
|
|
||||||
def publish_vectors(client):
|
|
||||||
text_ = text_version(client)
|
|
||||||
version = vector_build(client, text_)
|
|
||||||
assert vector_publish(client, version).status_code == 200
|
|
||||||
return text_, version
|
|
||||||
|
|
||||||
|
|
||||||
def test_rrf_hand_calculation_absent_leg_and_raw_score_independence():
|
|
||||||
a, b, c = uuid4(), uuid4(), uuid4()
|
|
||||||
lexical = [
|
|
||||||
RankedCandidate(chunk_id=a, rank=1, score=1000),
|
|
||||||
RankedCandidate(chunk_id=b, rank=2, score=1),
|
|
||||||
]
|
|
||||||
vector = [
|
|
||||||
RankedCandidate(chunk_id=b, rank=1, score=-0.9),
|
|
||||||
RankedCandidate(chunk_id=c, rank=2, score=-1),
|
|
||||||
]
|
|
||||||
scores = fuse(lexical, vector)
|
|
||||||
assert scores == {str(a): 1 / 61, str(b): 1 / 62 + 1 / 61, str(c): 1 / 62}
|
|
||||||
for candidate in lexical + vector:
|
|
||||||
candidate.score *= 10000
|
|
||||||
assert fuse(lexical, vector) == scores
|
|
||||||
assert fuse([], []) == {}
|
|
||||||
|
|
||||||
|
|
||||||
def test_same_chunk_scope_standalone_rankings_formula_and_history(client, provider):
|
|
||||||
items = [
|
|
||||||
add_item(client, content="alpha query extra words words"),
|
|
||||||
add_item(client, content="query"),
|
|
||||||
add_item(client, content="beta"),
|
|
||||||
]
|
|
||||||
add_item(client, status="pending", content="query alpha")
|
|
||||||
text_, version = publish_vectors(client)
|
|
||||||
response = search(client, top_k=3, candidate_k=3)
|
|
||||||
assert response.status_code == 200, response.text
|
|
||||||
result = response.json()
|
|
||||||
lexical = search(client, "chunks", top_k=3).json()
|
|
||||||
vector = search(client, "vector", top_k=3).json()
|
|
||||||
assert result["corpus_size"] == lexical["corpus_size"] == vector["corpus_size"] == 3
|
|
||||||
assert (
|
|
||||||
result["text_index_version_id"]
|
|
||||||
== lexical["text_index_version_id"]
|
|
||||||
== vector["text_index_version_id"]
|
|
||||||
== text_["id"]
|
|
||||||
)
|
|
||||||
assert result["vector_version_id"] == version["id"]
|
|
||||||
assert result["config"]["space"]["id"] == provider.spec.space_id
|
|
||||||
assert result["config"]["lexical"]["title_weight"] == 0
|
|
||||||
assert result["config"]["lexical"]["k1"] == 1.2
|
|
||||||
assert result["config"]["lexical"]["b"] == 0.75
|
|
||||||
assert result["config"]["chunking"] == text_["config"]
|
|
||||||
assert result["lexical_total_hits"] == 2 and result["vector_total_hits"] == 3
|
|
||||||
assert [c["chunk_id"] for c in result["lexical_candidates"]] == [
|
|
||||||
c["chunk_id"] for c in lexical["results"]
|
|
||||||
]
|
|
||||||
assert [c["chunk_id"] for c in result["vector_candidates"]] == [
|
|
||||||
c["chunk_id"] for c in vector["results"]
|
|
||||||
]
|
|
||||||
# Independently calculate positive-log-IDF BM25 on bodies of lengths 5, 1, 1.
|
|
||||||
expected = math.log(1 + 1.5 / 2.5) * 2.2 / (1 + 1.2 * (0.25 + 0.75 / (7 / 3)))
|
|
||||||
assert lexical["results"][0]["knowledge_id"] == items[1]["id"]
|
|
||||||
assert lexical["results"][0]["score"] == pytest.approx(expected)
|
|
||||||
for hit in result["results"]:
|
|
||||||
left = next(
|
|
||||||
(c for c in result["lexical_candidates"] if c["chunk_id"] == hit["chunk_id"]), None
|
|
||||||
)
|
|
||||||
right = next(
|
|
||||||
(c for c in result["vector_candidates"] if c["chunk_id"] == hit["chunk_id"]), None
|
|
||||||
)
|
|
||||||
assert hit["score"] == pytest.approx(
|
|
||||||
(1 / (60 + left["rank"]) if left else 0) + (1 / (60 + right["rank"]) if right else 0)
|
|
||||||
)
|
|
||||||
assert hit["score"] == hit["lexical_contribution"] + hit["vector_contribution"]
|
|
||||||
assert hit["source_url"] == "https://example.com/"
|
|
||||||
assert search(client, candidate_k=3, top_k=3).json()["results"] == result["results"]
|
|
||||||
for value in (result, lexical, vector):
|
|
||||||
assert client.get(f"/api/retrieval/runs/{value['run_id']}").json() == value
|
|
||||||
old = client.post("/api/retrieval", json={"query": "query"}).json()
|
|
||||||
assert old["mode"] == "keyword" and "chunk_id" not in old["results"][0]
|
|
||||||
history = client.get("/api/retrieval/runs?page_size=100").json()
|
|
||||||
assert {r["mode"] for r in history["items"]} == {
|
|
||||||
"keyword",
|
|
||||||
"vector",
|
|
||||||
"keyword_chunks",
|
|
||||||
"hybrid",
|
|
||||||
}
|
|
||||||
client.cookies.clear()
|
|
||||||
assert search(client).status_code == search(client, "chunks").status_code == 200
|
|
||||||
assert client.get("/api/retrieval/runs").status_code == 401
|
|
||||||
|
|
||||||
|
|
||||||
def test_candidate_cutoffs_disjoint_legs_and_deterministic_tie(client, provider):
|
|
||||||
first = add_item(client, content="alpha")
|
|
||||||
second = add_item(client, content="query")
|
|
||||||
publish_vectors(client)
|
|
||||||
small = search(client, candidate_k=1, top_k=1).json()
|
|
||||||
assert small["total_hits"] == 2 and len(small["results"]) == 1
|
|
||||||
assert small["results"][0]["knowledge_id"] == first["id"]
|
|
||||||
assert small["results"][0]["lexical_rank"] is None
|
|
||||||
assert small["results"][0]["lexical_score"] is None
|
|
||||||
assert small["results"][0]["lexical_contribution"] == 0
|
|
||||||
assert small["results"][0]["score"] == 1 / 61
|
|
||||||
full = search(client, candidate_k=2, top_k=2).json()
|
|
||||||
assert full["results"][0]["knowledge_id"] == second["id"]
|
|
||||||
assert full["results"][0]["lexical_rank"] == 1 and full["results"][0]["vector_rank"] == 2
|
|
||||||
|
|
||||||
|
|
||||||
def test_chunk_bodies_offsets_overlap_and_category_apply_before_both_rankings(client, provider):
|
|
||||||
item = add_item(client, title="uniquetitle", content="alpha query 文本。" * 30, category="交通")
|
|
||||||
add_item(client, content="query", category="其他")
|
|
||||||
publish_vectors(client)
|
|
||||||
result = search(client, category="交通", top_k=20, candidate_k=20).json()
|
|
||||||
assert result["corpus_size"] > 1 and result["total_hits"] == result["corpus_size"]
|
|
||||||
assert len({h["chunk_id"] for h in result["results"]}) == len(result["results"])
|
|
||||||
for hit in result["results"]:
|
|
||||||
assert hit["knowledge_id"] == item["id"]
|
|
||||||
assert hit["snippet"] == item["content"][hit["snippet_start"] : hit["snippet_end"]]
|
|
||||||
assert search(client, "chunks", query="uniquetitle").json()["status"] == "no_evidence"
|
|
||||||
empty = search(client, category="不存在分类").json()
|
|
||||||
assert empty["status"] == "no_evidence" and empty["results"] == []
|
|
||||||
assert empty["lexical_candidates"] == empty["vector_candidates"] == []
|
|
||||||
|
|
||||||
|
|
||||||
def test_empty_lexical_leg_and_empty_corpus_are_explicit(client, provider):
|
|
||||||
add_item(client, content="alpha")
|
|
||||||
publish_vectors(client)
|
|
||||||
for query in ["unmatchedword", "!!"]:
|
|
||||||
result = search(client, query=query).json()
|
|
||||||
assert result["lexical_candidates"] == [] and result["status"] == "ok"
|
|
||||||
assert result["results"][0]["score"] == 1 / 61
|
|
||||||
assert search(client, "chunks", query="!!").status_code == 422
|
|
||||||
|
|
||||||
|
|
||||||
def test_empty_published_corpus(client, provider):
|
|
||||||
publish_vectors(client)
|
|
||||||
for path in ["chunks", "hybrid"]:
|
|
||||||
result = search(client, path).json()
|
|
||||||
assert result["status"] == "no_evidence" and result["total_hits"] == 0
|
|
||||||
|
|
||||||
|
|
||||||
def test_keyword_chunks_need_no_runtime_or_vector_publication(client):
|
|
||||||
add_item(client, content="query")
|
|
||||||
assert search(client, "chunks").status_code == 409
|
|
||||||
text_version(client)
|
|
||||||
assert search(client, "chunks").status_code == 200
|
|
||||||
assert search(client).status_code == 503
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
"payload",
|
|
||||||
[
|
|
||||||
{"candidate_k": 1, "top_k": 2},
|
|
||||||
{"candidate_k": 201},
|
|
||||||
{"candidate_k": 0},
|
|
||||||
{"top_k": 21},
|
|
||||||
{"top_k": 0},
|
|
||||||
{"mode": "vector"},
|
|
||||||
{"rrf_k": 10},
|
|
||||||
{"query": " "},
|
|
||||||
{"query": "x" * 501},
|
|
||||||
],
|
|
||||||
)
|
|
||||||
def test_request_budget_and_fixed_config_validation(client, provider, payload):
|
|
||||||
assert search(client, **payload).status_code == 422
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize("change", ["catalog", "text", "vector"])
|
|
||||||
def test_query_encoding_releases_locks_and_rechecks_publication(
|
|
||||||
client, provider, migrated_database, change
|
|
||||||
):
|
|
||||||
item = add_item(client, content="alpha query")
|
|
||||||
text_, first = publish_vectors(client)
|
|
||||||
replacement_text = build(client)
|
|
||||||
replacement_vector = vector_build(client, text_)
|
|
||||||
|
|
||||||
def mutate(query):
|
|
||||||
assert query
|
|
||||||
with Session(migrated_database) as other:
|
|
||||||
other.execute(text("SET LOCAL lock_timeout='1s'"))
|
|
||||||
if change == "catalog":
|
|
||||||
state = vectors.locked_state(other)
|
|
||||||
state.catalog_revision += 1
|
|
||||||
other.execute(
|
|
||||||
text(
|
|
||||||
"UPDATE knowledge_items SET content='changed synthetic evidence' WHERE id=:id"
|
|
||||||
),
|
|
||||||
{"id": item["id"]},
|
|
||||||
)
|
|
||||||
other.commit()
|
|
||||||
elif change == "text":
|
|
||||||
publish_index(other, UUID(replacement_text["id"]), 1, first["created_by"])
|
|
||||||
else:
|
|
||||||
vectors.publish_vectors(
|
|
||||||
other, UUID(replacement_vector["id"]), 1, first["created_by"]
|
|
||||||
)
|
|
||||||
|
|
||||||
provider.callback = mutate
|
|
||||||
assert search(client).status_code == 409
|
|
||||||
with Session(migrated_database) as db:
|
|
||||||
assert db.scalar(select(func.count()).select_from(RetrievalRun)) == 0
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize("change", ["body", "chunks", "vectors", "space"])
|
|
||||||
def test_corrupt_or_incompatible_snapshot_fails_closed(client, provider, migrated_database, change):
|
|
||||||
add_item(client, content="alpha query")
|
|
||||||
publish_vectors(client)
|
|
||||||
if change == "space":
|
|
||||||
provider.spec = replace(provider.spec, revision="b" * 40)
|
|
||||||
else:
|
|
||||||
with migrated_database.begin() as connection:
|
|
||||||
connection.execute(
|
|
||||||
text(
|
|
||||||
{
|
|
||||||
"body": "UPDATE knowledge_items SET content='out of band change'",
|
|
||||||
"chunks": "UPDATE chunks SET text='tampered'",
|
|
||||||
"vectors": "DELETE FROM chunk_vectors",
|
|
||||||
}[change]
|
|
||||||
)
|
|
||||||
)
|
|
||||||
assert search(client).status_code == 409
|
|
||||||
with Session(migrated_database) as db:
|
|
||||||
assert db.scalar(select(func.count()).select_from(RetrievalRun)) == 0
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize("failure,status", [("token", 422), ("runtime", 503)])
|
|
||||||
def test_model_failure_has_no_fallback_or_partial_history(
|
|
||||||
client, provider, migrated_database, failure, status
|
|
||||||
):
|
|
||||||
add_item(client, content="alpha query")
|
|
||||||
publish_vectors(client)
|
|
||||||
|
|
||||||
def fail(*args, **kwargs):
|
|
||||||
if failure == "token":
|
|
||||||
raise EmbeddingError("token budget")
|
|
||||||
raise OSError("private-runtime-path")
|
|
||||||
|
|
||||||
provider.encode = fail
|
|
||||||
response = search(client)
|
|
||||||
assert response.status_code == status and "private-runtime-path" not in response.text
|
|
||||||
with Session(migrated_database) as db:
|
|
||||||
assert db.scalar(select(func.count()).select_from(RetrievalRun)) == 0
|
|
||||||
|
|
||||||
|
|
||||||
def test_failed_history_insert_rolls_back_both_leg_result(client, provider, migrated_database):
|
|
||||||
add_item(client, content="alpha query")
|
|
||||||
publish_vectors(client)
|
|
||||||
with migrated_database.begin() as connection:
|
|
||||||
connection.execute(
|
|
||||||
text("ALTER TABLE retrieval_runs ADD CONSTRAINT test_no_runs CHECK (false) NOT VALID")
|
|
||||||
)
|
|
||||||
assert search(client).status_code == 409
|
|
||||||
with Session(migrated_database) as db:
|
|
||||||
assert db.scalar(select(func.count()).select_from(RetrievalRun)) == 0
|
|
||||||
|
|
||||||
|
|
||||||
def test_stale_history_survives_evidence_change_and_capacity_is_explicit(
|
|
||||||
client, provider, migrated_database, monkeypatch
|
|
||||||
):
|
|
||||||
item = add_item(client, content="alpha query")
|
|
||||||
publish_vectors(client)
|
|
||||||
result = search(client).json()
|
|
||||||
monkeypatch.setattr(vectors, "MAX_VECTOR_CHUNKS", 0)
|
|
||||||
assert search(client).status_code == search(client, "chunks").status_code == 503
|
|
||||||
monkeypatch.undo()
|
|
||||||
assert (
|
|
||||||
client.patch(
|
|
||||||
f"/api/knowledge/{item['id']}",
|
|
||||||
json={"expected_version": item["version"], "content": "updated synthetic body"},
|
|
||||||
).status_code
|
|
||||||
== 200
|
|
||||||
)
|
|
||||||
assert search(client).status_code == search(client, "chunks").status_code == 409
|
|
||||||
assert client.get(f"/api/retrieval/runs/{result['run_id']}").json() == result
|
|
||||||
@@ -16,7 +16,7 @@ def test_startup_requires_migration(database):
|
|||||||
|
|
||||||
|
|
||||||
def test_startup_rejects_previous_schema(database):
|
def test_startup_rejects_previous_schema(database):
|
||||||
migrate(database, "0005_chunk_indexes")
|
migrate(database, "0004_auth_sessions")
|
||||||
with pytest.raises(RuntimeError, match="alembic upgrade head"):
|
with pytest.raises(RuntimeError, match="alembic upgrade head"):
|
||||||
with TestClient(create_app(database)):
|
with TestClient(create_app(database)):
|
||||||
pass
|
pass
|
||||||
@@ -40,7 +40,10 @@ def test_upgrade_preserves_original_rows_and_repeated_upgrade(database):
|
|||||||
assert row["collected_at"] is None
|
assert row["collected_at"] is None
|
||||||
assert row["verification_status"] == "pending"
|
assert row["verification_status"] == "pending"
|
||||||
assert connection.scalar(text("SELECT count(*) FROM audit_events")) == 2
|
assert connection.scalar(text("SELECT count(*) FROM audit_events")) == 2
|
||||||
assert connection.scalar(text("SELECT version_num FROM alembic_version")) == "0006_vectors"
|
assert (
|
||||||
|
connection.scalar(text("SELECT version_num FROM alembic_version"))
|
||||||
|
== "0005_chunk_indexes"
|
||||||
|
)
|
||||||
assert compare_metadata(MigrationContext.configure(connection), Base.metadata) == []
|
assert compare_metadata(MigrationContext.configure(connection), Base.metadata) == []
|
||||||
migrate(database, "0001_baseline", downgrade=True)
|
migrate(database, "0001_baseline", downgrade=True)
|
||||||
with database.connect() as connection:
|
with database.connect() as connection:
|
||||||
@@ -95,29 +98,3 @@ def test_chunk_index_downgrade_preserves_users_sessions_knowledge_and_retrieval(
|
|||||||
assert client.get("/api/auth/me").status_code == 200
|
assert client.get("/api/auth/me").status_code == 200
|
||||||
assert client.get("/api/indexes").json()["total"] == 0
|
assert client.get("/api/indexes").json()["total"] == 0
|
||||||
assert client.get("/api/indexes/active").json()["generation"] == 0
|
assert client.get("/api/indexes/active").json()["generation"] == 0
|
||||||
|
|
||||||
|
|
||||||
def test_vector_downgrade_retains_text_publications_and_prior_retrieval_runs(
|
|
||||||
client, migrated_database
|
|
||||||
):
|
|
||||||
from test_retrieval import add_item
|
|
||||||
from test_vectors import ToyProvider, text_version, vector_build, vector_publish, vector_search
|
|
||||||
|
|
||||||
from app.services.embedding_runtime import get_embedding_provider
|
|
||||||
|
|
||||||
client.app.dependency_overrides[get_embedding_provider] = ToyProvider
|
|
||||||
add_item(client)
|
|
||||||
text_ = text_version(client)
|
|
||||||
version = vector_build(client, text_)
|
|
||||||
vector_publish(client, version)
|
|
||||||
history = vector_search(client).json()
|
|
||||||
migrate(migrated_database, "0005_chunk_indexes", downgrade=True)
|
|
||||||
with migrated_database.connect() as c:
|
|
||||||
assert c.scalar(text("SELECT active_version_id FROM index_state")) == text_["id"]
|
|
||||||
assert c.scalar(text("SELECT count(*) FROM chunks")) == 1
|
|
||||||
assert c.scalar(text("SELECT count(*) FROM users")) == 1
|
|
||||||
assert c.scalar(text("SELECT count(*) FROM index_publications")) == 1
|
|
||||||
assert c.scalar(text("SELECT id FROM retrieval_runs")) == history["run_id"]
|
|
||||||
migrate(migrated_database)
|
|
||||||
assert client.get("/api/vector-indexes/active").json()["generation"] == 0
|
|
||||||
assert client.get(f"/api/retrieval/runs/{history['run_id']}").json() == history
|
|
||||||
|
|||||||
@@ -17,7 +17,7 @@ from app.services.auth import PASSWORD_HASHER
|
|||||||
|
|
||||||
|
|
||||||
@contextmanager
|
@contextmanager
|
||||||
def server(engine, *, authenticate=True, environment_overrides=None):
|
def server(engine, *, authenticate=True):
|
||||||
with socket.socket() as socket_:
|
with socket.socket() as socket_:
|
||||||
socket_.bind(("127.0.0.1", 0))
|
socket_.bind(("127.0.0.1", 0))
|
||||||
port = socket_.getsockname()[1]
|
port = socket_.getsockname()[1]
|
||||||
@@ -26,7 +26,6 @@ def server(engine, *, authenticate=True, environment_overrides=None):
|
|||||||
"DATABASE_URL": engine.url.render_as_string(hide_password=False),
|
"DATABASE_URL": engine.url.render_as_string(hide_password=False),
|
||||||
"APP_ENV": "test",
|
"APP_ENV": "test",
|
||||||
"AUTH_COOKIE_SECURE": "false",
|
"AUTH_COOKIE_SECURE": "false",
|
||||||
**(environment_overrides or {}),
|
|
||||||
}
|
}
|
||||||
if authenticate:
|
if authenticate:
|
||||||
with Session(engine) as db:
|
with Session(engine) as db:
|
||||||
|
|||||||
@@ -1,115 +0,0 @@
|
|||||||
"""Explicit opt-in: actual pinned model -> HTTP -> PostgreSQL -> HTTP process restart."""
|
|
||||||
|
|
||||||
import os
|
|
||||||
|
|
||||||
import pytest
|
|
||||||
from sqlalchemy import text
|
|
||||||
from test_runtime import server
|
|
||||||
|
|
||||||
from app.services.embedding import SMALL_ZH
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.skipif(
|
|
||||||
os.environ.get("RUN_LOCAL_EMBEDDING_TEST") != "1",
|
|
||||||
reason="requires optional CPU runtime and offline pinned model cache",
|
|
||||||
)
|
|
||||||
def test_real_local_model_vectors_retrieval_and_http_restart(migrated_database):
|
|
||||||
options = {
|
|
||||||
"EMBEDDING_ENABLED": "true",
|
|
||||||
"EMBEDDING_MODEL": "small",
|
|
||||||
"EMBEDDING_CACHE_DIR": os.environ["EMBEDDING_CACHE_DIR"],
|
|
||||||
"HF_HUB_OFFLINE": "1",
|
|
||||||
"HF_HUB_DISABLE_TELEMETRY": "1",
|
|
||||||
"HF_HUB_DISABLE_IMPLICIT_TOKEN": "1",
|
|
||||||
"TOKENIZERS_PARALLELISM": "false",
|
|
||||||
}
|
|
||||||
query = {"query": "合成测试中交通安排应查阅哪份资料?", "top_k": 3}
|
|
||||||
with server(migrated_database, environment_overrides=options) as client:
|
|
||||||
ids = []
|
|
||||||
for topic, body in [
|
|
||||||
("交通", "【合成测试】占位交通资料介绍测试公交线路与班次的查询入口。"),
|
|
||||||
("住宿", "【合成测试】占位住宿资料介绍测试房型与入住须知。"),
|
|
||||||
("审核", "【合成测试】占位审核资料说明测试出处核验流程。"),
|
|
||||||
]:
|
|
||||||
item = client.post(
|
|
||||||
"/api/knowledge",
|
|
||||||
json={
|
|
||||||
"title": f"{topic}合成测试",
|
|
||||||
"content": body,
|
|
||||||
"source_url": "https://example.com/",
|
|
||||||
"collected_at": "2026-10-16T00:00:00Z",
|
|
||||||
},
|
|
||||||
).json()
|
|
||||||
response = client.post(
|
|
||||||
f"/api/knowledge/{item['id']}/review",
|
|
||||||
json={
|
|
||||||
"expected_version": item["version"],
|
|
||||||
"verification_status": "verified",
|
|
||||||
"reviewer": "test-reviewer",
|
|
||||||
"verification_notes": "仅为模型/HTTP/数据库持久化测试,不是事实资料",
|
|
||||||
},
|
|
||||||
)
|
|
||||||
assert response.status_code == 200
|
|
||||||
ids.append(item["id"])
|
|
||||||
text_ = client.post("/api/indexes", json={"chunk_size": 200, "overlap": 20}).json()
|
|
||||||
assert (
|
|
||||||
client.post(
|
|
||||||
f"/api/indexes/{text_['id']}/publish", json={"expected_generation": 0}
|
|
||||||
).status_code
|
|
||||||
== 200
|
|
||||||
)
|
|
||||||
built = client.post(
|
|
||||||
"/api/vector-indexes", json={"text_index_version_id": text_["id"]}, timeout=60
|
|
||||||
)
|
|
||||||
assert built.status_code == 201, built.text
|
|
||||||
version = built.json()
|
|
||||||
assert version["space_id"] == SMALL_ZH.space_id and version["dimension"] == 512
|
|
||||||
published = client.post(
|
|
||||||
f"/api/vector-indexes/{version['id']}/publish", json={"expected_generation": 0}
|
|
||||||
)
|
|
||||||
assert published.status_code == 200
|
|
||||||
active = published.json()
|
|
||||||
response = client.post("/api/retrieval/vector", json=query, timeout=60)
|
|
||||||
assert response.status_code == 200, response.text
|
|
||||||
first = response.json()
|
|
||||||
assert {h["knowledge_id"] for h in first["results"]} == set(ids)
|
|
||||||
assert all(-1 <= h["score"] <= 1 for h in first["results"])
|
|
||||||
assert first["config"]["space"]["spec"]["revision"] == SMALL_ZH.revision
|
|
||||||
response = client.post(
|
|
||||||
"/api/retrieval/hybrid", json={**query, "candidate_k": 3}, timeout=60
|
|
||||||
)
|
|
||||||
assert response.status_code == 200, response.text
|
|
||||||
hybrid = response.json()
|
|
||||||
assert hybrid["vector_version_id"] == version["id"]
|
|
||||||
assert hybrid["text_index_version_id"] == text_["id"]
|
|
||||||
assert hybrid["config"]["space"]["spec"]["revision"] == SMALL_ZH.revision
|
|
||||||
assert {h["knowledge_id"] for h in hybrid["results"]} == set(ids)
|
|
||||||
assert hybrid["vector_candidates"]
|
|
||||||
keyword = client.post("/api/retrieval/chunks", json=query).json()
|
|
||||||
assert keyword["mode"] == "keyword_chunks" and keyword["results"]
|
|
||||||
with migrated_database.connect() as c:
|
|
||||||
assert (
|
|
||||||
c.scalar(
|
|
||||||
text("SELECT count(*) FROM chunk_vectors WHERE public.vector_dims(embedding)=512")
|
|
||||||
)
|
|
||||||
== 3
|
|
||||||
)
|
|
||||||
assert c.scalar(text("SELECT count(*) FROM vector_publications")) == 1
|
|
||||||
with server(migrated_database, environment_overrides=options) as client:
|
|
||||||
assert client.get("/api/vector-indexes/active").json() == active
|
|
||||||
assert client.get(f"/api/retrieval/runs/{first['run_id']}").json() == first
|
|
||||||
response = client.post("/api/retrieval/vector", json=query, timeout=60)
|
|
||||||
assert response.status_code == 200, response.text
|
|
||||||
assert response.json()["results"] == first["results"]
|
|
||||||
for result, endpoint in [(hybrid, "hybrid"), (keyword, "chunks")]:
|
|
||||||
assert client.get(f"/api/retrieval/runs/{result['run_id']}").json() == result
|
|
||||||
response = client.post(
|
|
||||||
f"/api/retrieval/{endpoint}",
|
|
||||||
json={**query, **({"candidate_k": 3} if endpoint == "hybrid" else {})},
|
|
||||||
timeout=60,
|
|
||||||
)
|
|
||||||
assert response.status_code == 200, response.text
|
|
||||||
assert response.json()["results"] == result["results"]
|
|
||||||
if endpoint == "hybrid":
|
|
||||||
assert response.json()["lexical_candidates"] == result["lexical_candidates"]
|
|
||||||
assert response.json()["vector_candidates"] == result["vector_candidates"]
|
|
||||||
@@ -1,552 +0,0 @@
|
|||||||
"""Toy vectors test database mathematics and isolation, never model/domain quality."""
|
|
||||||
|
|
||||||
from dataclasses import replace
|
|
||||||
from uuid import UUID, uuid4
|
|
||||||
|
|
||||||
import pytest
|
|
||||||
from fastapi import HTTPException
|
|
||||||
from sqlalchemy import func, select, text, update
|
|
||||||
from sqlalchemy.exc import DBAPIError, OperationalError
|
|
||||||
from sqlalchemy.orm import Session
|
|
||||||
from test_indexing import build, publish
|
|
||||||
from test_retrieval import add_item
|
|
||||||
|
|
||||||
from app.models.indexing import Chunk
|
|
||||||
from app.models.retrieval import RetrievalRun
|
|
||||||
from app.models.vectors import ChunkVector, EmbeddingSpace, VectorPublication, VectorVersion
|
|
||||||
from app.services import vectors
|
|
||||||
from app.services.embedding import EmbeddingError, EmbeddingSpec
|
|
||||||
from app.services.embedding_runtime import EmbeddingRuntime, get_embedding_provider
|
|
||||||
from app.services.indexing import locked_state, publish_index
|
|
||||||
|
|
||||||
|
|
||||||
class ToyProvider:
|
|
||||||
spec = EmbeddingSpec("test-only-coordinate-fixture", "a" * 40, 3)
|
|
||||||
callback = None
|
|
||||||
|
|
||||||
def encode(self, texts, *, query=False):
|
|
||||||
if self.callback:
|
|
||||||
self.callback(query)
|
|
||||||
return [[1.0, 0.0, 0.0] if query or "alpha" in t else [0.0, 1.0, 0.0] for t in texts]
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
|
||||||
def provider(client):
|
|
||||||
value = ToyProvider()
|
|
||||||
client.app.dependency_overrides[get_embedding_provider] = lambda: value
|
|
||||||
return value
|
|
||||||
|
|
||||||
|
|
||||||
def text_version(client, **kwargs):
|
|
||||||
version = build(client, chunk_size=100, overlap=0, **kwargs)
|
|
||||||
generation = client.get("/api/indexes/active").json()["generation"]
|
|
||||||
assert publish(client, version, generation).status_code == 200
|
|
||||||
return version
|
|
||||||
|
|
||||||
|
|
||||||
def vector_build(client, version):
|
|
||||||
response = client.post("/api/vector-indexes", json={"text_index_version_id": version["id"]})
|
|
||||||
assert response.status_code == 201, response.text
|
|
||||||
return response.json()
|
|
||||||
|
|
||||||
|
|
||||||
def vector_publish(client, version, generation=0):
|
|
||||||
return client.post(
|
|
||||||
f"/api/vector-indexes/{version['id']}/publish", json={"expected_generation": generation}
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def vector_search(client, **kwargs):
|
|
||||||
return client.post("/api/retrieval/vector", json={"query": "alpha query", **kwargs})
|
|
||||||
|
|
||||||
|
|
||||||
def test_exact_cosine_original_chunks_category_and_mixed_history(client, provider):
|
|
||||||
first = add_item(client, content="alpha 占位正文", category="交通")
|
|
||||||
second = add_item(client, content="beta 占位正文", category="其他")
|
|
||||||
third = add_item(client, content="gamma 占位正文", category="其他")
|
|
||||||
add_item(client, status="pending", content="alpha 待审核不应进入")
|
|
||||||
text_ = text_version(client)
|
|
||||||
version = vector_build(client, text_)
|
|
||||||
assert (
|
|
||||||
version["status"] == "ready" and version["dimension"] == 3 and version["chunk_count"] == 3
|
|
||||||
)
|
|
||||||
assert vector_search(client).status_code == 409
|
|
||||||
assert vector_publish(client, version).json()["is_current"]
|
|
||||||
result = vector_search(client).json()
|
|
||||||
assert result["mode"] == "vector" and result["total_hits"] == 3
|
|
||||||
assert [h["knowledge_id"] for h in result["results"]] == [
|
|
||||||
first["id"],
|
|
||||||
second["id"],
|
|
||||||
third["id"],
|
|
||||||
]
|
|
||||||
assert [h["score"] for h in result["results"]] == [1.0, 0.0, 0.0]
|
|
||||||
assert result["config"]["space"]["id"] == provider.spec.space_id
|
|
||||||
assert result["text_index_version_id"] == text_["id"]
|
|
||||||
assert result["vector_version_id"] == version["id"]
|
|
||||||
assert result["corpus_fingerprint"] == text_["corpus_fingerprint"]
|
|
||||||
assert result["elapsed_ms"] > 0
|
|
||||||
assert result["results"][0]["snippet"] == first["content"]
|
|
||||||
assert result["results"][0]["snippet_start"] == 0
|
|
||||||
assert result["results"][0]["snippet_end"] == len(first["content"])
|
|
||||||
assert result["results"][0]["source_url"] == "https://example.com/"
|
|
||||||
filtered = vector_search(client, category="其他", top_k=1).json()
|
|
||||||
assert filtered["corpus_size"] == 2 and filtered["results"][0]["knowledge_id"] == second["id"]
|
|
||||||
empty = vector_search(client, category="不存在分类").json()
|
|
||||||
assert empty["status"] == "no_evidence" and empty["results"] == []
|
|
||||||
keyword = client.post("/api/retrieval", json={"query": "alpha"}).json()
|
|
||||||
history = client.get("/api/retrieval/runs?page_size=100").json()
|
|
||||||
assert {r["mode"] for r in history["items"]} == {"keyword", "vector"}
|
|
||||||
assert client.get(f"/api/retrieval/runs/{result['run_id']}").json() == result
|
|
||||||
assert client.get(f"/api/retrieval/runs/{keyword['run_id']}").json() == keyword
|
|
||||||
client.cookies.clear()
|
|
||||||
assert vector_search(client).status_code == 200
|
|
||||||
assert client.get("/api/retrieval/runs").status_code == 401
|
|
||||||
|
|
||||||
|
|
||||||
def test_multiple_chunks_keep_original_offsets_and_order(client, provider):
|
|
||||||
item = add_item(client, content="alpha 占位正文。" * 45)
|
|
||||||
version = vector_build(client, text_version(client))
|
|
||||||
vector_publish(client, version)
|
|
||||||
hits = vector_search(client, top_k=20).json()["results"]
|
|
||||||
assert len(hits) > 1
|
|
||||||
assert [h["chunk_ordinal"] for h in hits] == list(range(len(hits)))
|
|
||||||
assert len({h["chunk_id"] for h in hits}) == len(hits)
|
|
||||||
for h in hits:
|
|
||||||
assert h["snippet"] == item["content"][h["snippet_start"] : h["snippet_end"]]
|
|
||||||
|
|
||||||
|
|
||||||
def test_same_dimension_different_revision_is_a_separate_space(client, provider, migrated_database):
|
|
||||||
add_item(client)
|
|
||||||
text_ = text_version(client)
|
|
||||||
first = vector_build(client, text_)
|
|
||||||
vector_publish(client, first)
|
|
||||||
old_space = provider.spec.space_id
|
|
||||||
provider.spec = replace(provider.spec, revision="b" * 40)
|
|
||||||
assert vector_search(client).status_code == 409
|
|
||||||
second = vector_build(client, text_)
|
|
||||||
assert second["space_id"] != old_space and second["dimension"] == first["dimension"]
|
|
||||||
assert vector_publish(client, second, 1).status_code == 200
|
|
||||||
assert vector_search(client).json()["config"]["space"]["id"] == provider.spec.space_id
|
|
||||||
assert client.get("/api/vector-indexes/spaces").json()["total"] == 2
|
|
||||||
with Session(migrated_database) as db:
|
|
||||||
assert db.scalar(select(func.count()).select_from(EmbeddingSpace)) == 2
|
|
||||||
assert db.scalar(select(func.count()).select_from(ChunkVector)) == 2
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize("bad", ["dimension", "zero", "nan", "count", "overflow", "unavailable"])
|
|
||||||
def test_invalid_or_failed_encode_never_persists_partial_vectors(
|
|
||||||
client, provider, migrated_database, bad
|
|
||||||
):
|
|
||||||
add_item(client, content="alpha 占位")
|
|
||||||
text_ = text_version(client)
|
|
||||||
original = vector_build(client, text_)
|
|
||||||
vector_publish(client, original)
|
|
||||||
before = client.get("/api/vector-indexes/active").json()
|
|
||||||
|
|
||||||
def fail(texts, **kwargs):
|
|
||||||
if bad == "overflow":
|
|
||||||
raise EmbeddingError("token limit")
|
|
||||||
if bad == "unavailable":
|
|
||||||
raise OSError("private-path-must-not-appear")
|
|
||||||
return {
|
|
||||||
"dimension": [[1, 0]],
|
|
||||||
"zero": [[0, 0, 0]],
|
|
||||||
"nan": [[float("nan"), 0, 0]],
|
|
||||||
"count": [],
|
|
||||||
}[bad]
|
|
||||||
|
|
||||||
provider.encode = fail
|
|
||||||
response = client.post("/api/vector-indexes", json={"text_index_version_id": text_["id"]})
|
|
||||||
assert response.status_code == (503 if bad == "unavailable" else 422)
|
|
||||||
assert "private-path" not in response.text
|
|
||||||
assert client.get("/api/vector-indexes/active").json() == before
|
|
||||||
with Session(migrated_database) as db:
|
|
||||||
assert db.scalar(select(func.count()).select_from(VectorVersion)) == 1
|
|
||||||
assert db.scalar(select(func.count()).select_from(ChunkVector)) == 1
|
|
||||||
|
|
||||||
|
|
||||||
def test_build_and_query_encode_release_locks_but_recheck_catalog(
|
|
||||||
client, provider, migrated_database
|
|
||||||
):
|
|
||||||
item = add_item(client, content="alpha 占位")
|
|
||||||
text_ = text_version(client)
|
|
||||||
original = vector_build(client, text_)
|
|
||||||
vector_publish(client, original)
|
|
||||||
|
|
||||||
def mutate(query):
|
|
||||||
with Session(migrated_database) as other:
|
|
||||||
other.execute(text("SET LOCAL lock_timeout = '100ms'"))
|
|
||||||
state = locked_state(other) # Must succeed during both document and query inference.
|
|
||||||
state.catalog_revision += 1
|
|
||||||
other.execute(
|
|
||||||
text("UPDATE knowledge_items SET content=:value WHERE id=:id"),
|
|
||||||
{"value": "修订占位", "id": item["id"]},
|
|
||||||
)
|
|
||||||
other.commit()
|
|
||||||
|
|
||||||
provider.callback = mutate
|
|
||||||
response = client.post("/api/vector-indexes", json={"text_index_version_id": text_["id"]})
|
|
||||||
assert response.status_code == 409
|
|
||||||
assert not client.get("/api/vector-indexes/active").json()["is_current"]
|
|
||||||
assert vector_search(client).status_code == 409
|
|
||||||
# Restore an active text/vector snapshot, then race during query encoding too.
|
|
||||||
provider.callback = None
|
|
||||||
text_ = text_version(client)
|
|
||||||
replacement = vector_build(client, text_)
|
|
||||||
vector_publish(client, replacement, 1)
|
|
||||||
provider.callback = mutate
|
|
||||||
assert vector_search(client).status_code == 409
|
|
||||||
with Session(migrated_database) as db:
|
|
||||||
assert db.scalar(select(func.count()).select_from(VectorVersion)) == 2
|
|
||||||
assert db.scalar(select(func.count()).select_from(RetrievalRun)) == 0
|
|
||||||
|
|
||||||
|
|
||||||
def test_query_rejects_publication_changed_during_encode(client, provider, migrated_database):
|
|
||||||
add_item(client)
|
|
||||||
text_ = text_version(client)
|
|
||||||
first, second = vector_build(client, text_), vector_build(client, text_)
|
|
||||||
vector_publish(client, first)
|
|
||||||
|
|
||||||
def switch(query):
|
|
||||||
if query:
|
|
||||||
with Session(migrated_database) as other:
|
|
||||||
vectors.publish_vectors(other, UUID(second["id"]), 1, second["created_by"])
|
|
||||||
|
|
||||||
provider.callback = switch
|
|
||||||
assert vector_search(client).status_code == 409
|
|
||||||
assert client.get("/api/vector-indexes/active").json()["active_version"]["id"] == second["id"]
|
|
||||||
|
|
||||||
|
|
||||||
def test_text_switch_during_build_discards_new_vectors(client, provider, migrated_database):
|
|
||||||
add_item(client)
|
|
||||||
first = text_version(client)
|
|
||||||
replacement = build(client)
|
|
||||||
|
|
||||||
def switch(query):
|
|
||||||
with Session(migrated_database) as other:
|
|
||||||
publish_index(other, UUID(replacement["id"]), 1, replacement["created_by"])
|
|
||||||
|
|
||||||
provider.callback = switch
|
|
||||||
assert (
|
|
||||||
client.post("/api/vector-indexes", json={"text_index_version_id": first["id"]}).status_code
|
|
||||||
== 409
|
|
||||||
)
|
|
||||||
assert client.get("/api/vector-indexes").json()["total"] == 0
|
|
||||||
|
|
||||||
|
|
||||||
def test_out_of_band_evidence_change_fails_closed(client, provider, migrated_database):
|
|
||||||
item = add_item(client)
|
|
||||||
text_ = text_version(client)
|
|
||||||
version = vector_build(client, text_)
|
|
||||||
vector_publish(client, version)
|
|
||||||
with migrated_database.begin() as c:
|
|
||||||
c.execute(
|
|
||||||
text("UPDATE knowledge_items SET content='SQL 更新占位' WHERE id=:id"),
|
|
||||||
{"id": item["id"]},
|
|
||||||
)
|
|
||||||
assert not client.get("/api/vector-indexes/active").json()["is_current"]
|
|
||||||
assert vector_search(client).status_code == 409
|
|
||||||
assert vector_publish(client, version, 1).status_code == 409
|
|
||||||
|
|
||||||
|
|
||||||
def test_optimistic_publication_audit_and_rollback(client, provider, migrated_database):
|
|
||||||
add_item(client)
|
|
||||||
text_ = text_version(client)
|
|
||||||
first, second = vector_build(client, text_), vector_build(client, text_)
|
|
||||||
assert vector_publish(client, first).json()["generation"] == 1
|
|
||||||
assert vector_publish(client, second).status_code == 409
|
|
||||||
assert vector_publish(client, first, 1).json()["generation"] == 1
|
|
||||||
with migrated_database.begin() as c:
|
|
||||||
c.execute(
|
|
||||||
text(
|
|
||||||
"ALTER TABLE vector_publications ADD CONSTRAINT test_reject_publication CHECK (false) NOT VALID"
|
|
||||||
)
|
|
||||||
)
|
|
||||||
assert vector_publish(client, second, 1).status_code == 409
|
|
||||||
assert client.get("/api/vector-indexes/active").json()["active_version"]["id"] == first["id"]
|
|
||||||
assert client.get(f"/api/vector-indexes/{first['id']}").json()["status"] == "published"
|
|
||||||
assert client.get(f"/api/vector-indexes/{second['id']}").json()["status"] == "ready"
|
|
||||||
assert client.get("/api/vector-indexes/publications").json()["total"] == 1
|
|
||||||
with migrated_database.begin() as c:
|
|
||||||
c.execute(text("ALTER TABLE vector_publications DROP CONSTRAINT test_reject_publication"))
|
|
||||||
assert vector_publish(client, second, 1).json()["generation"] == 2
|
|
||||||
assert client.get(f"/api/vector-indexes/{first['id']}").json()["status"] == "superseded"
|
|
||||||
|
|
||||||
|
|
||||||
def test_vector_insert_failure_rolls_back_entire_build(client, provider, migrated_database):
|
|
||||||
add_item(client)
|
|
||||||
text_ = text_version(client)
|
|
||||||
with migrated_database.begin() as c:
|
|
||||||
c.execute(
|
|
||||||
text(
|
|
||||||
"ALTER TABLE chunk_vectors ADD CONSTRAINT test_reject_vector CHECK (false) NOT VALID"
|
|
||||||
)
|
|
||||||
)
|
|
||||||
assert (
|
|
||||||
client.post("/api/vector-indexes", json={"text_index_version_id": text_["id"]}).status_code
|
|
||||||
== 409
|
|
||||||
)
|
|
||||||
with Session(migrated_database) as db:
|
|
||||||
assert db.scalar(select(func.count()).select_from(VectorVersion)) == 0
|
|
||||||
assert db.scalar(select(func.count()).select_from(ChunkVector)) == 0
|
|
||||||
assert db.scalar(select(func.count()).select_from(EmbeddingSpace)) == 0
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
"column,value",
|
|
||||||
[
|
|
||||||
("dimension", 2),
|
|
||||||
("space_id", "0" * 64),
|
|
||||||
("text_index_version_id", str(uuid4())),
|
|
||||||
("embedding", [1, 0]),
|
|
||||||
("embedding", [2, 0, 0]),
|
|
||||||
("embedding", [float("nan"), 0, 0]),
|
|
||||||
],
|
|
||||||
)
|
|
||||||
def test_postgresql_rejects_cross_scope_dimensions_and_nonunit_values(
|
|
||||||
client, provider, migrated_database, column, value
|
|
||||||
):
|
|
||||||
add_item(client)
|
|
||||||
version = vector_build(client, text_version(client))
|
|
||||||
with Session(migrated_database) as db:
|
|
||||||
with pytest.raises(DBAPIError):
|
|
||||||
db.execute(
|
|
||||||
update(ChunkVector)
|
|
||||||
.where(ChunkVector.version_id == version["id"])
|
|
||||||
.values(**{column: value})
|
|
||||||
)
|
|
||||||
db.commit()
|
|
||||||
db.rollback()
|
|
||||||
row = db.scalar(select(ChunkVector))
|
|
||||||
assert row.space_id == provider.spec.space_id and row.dimension == 3
|
|
||||||
|
|
||||||
|
|
||||||
def test_same_width_cross_space_and_cross_text_chunks_are_rejected_by_database(
|
|
||||||
client, provider, migrated_database
|
|
||||||
):
|
|
||||||
add_item(client)
|
|
||||||
original_text = text_version(client)
|
|
||||||
first = vector_build(client, original_text)
|
|
||||||
provider.spec = replace(provider.spec, revision="c" * 40)
|
|
||||||
second = vector_build(client, original_text)
|
|
||||||
newer_text = text_version(client)
|
|
||||||
with Session(migrated_database) as db:
|
|
||||||
other_chunk = db.scalar(select(Chunk.id).where(Chunk.index_version_id == newer_text["id"]))
|
|
||||||
for fields in [{"space_id": second["space_id"]}, {"chunk_id": other_chunk}]:
|
|
||||||
with pytest.raises(DBAPIError):
|
|
||||||
db.execute(
|
|
||||||
update(ChunkVector)
|
|
||||||
.where(ChunkVector.version_id == first["id"])
|
|
||||||
.values(**fields)
|
|
||||||
)
|
|
||||||
db.commit()
|
|
||||||
db.rollback()
|
|
||||||
|
|
||||||
|
|
||||||
def test_incomplete_vector_set_cannot_be_published_or_queried(client, provider, migrated_database):
|
|
||||||
add_item(client)
|
|
||||||
version = vector_build(client, text_version(client))
|
|
||||||
vector_publish(client, version)
|
|
||||||
with migrated_database.begin() as c:
|
|
||||||
c.execute(text("DELETE FROM chunk_vectors"))
|
|
||||||
assert vector_publish(client, version, 1).status_code == 409
|
|
||||||
assert vector_search(client).status_code == 409
|
|
||||||
|
|
||||||
|
|
||||||
def test_only_current_text_and_bounded_complete_snapshots_can_build(
|
|
||||||
client, provider, migrated_database, monkeypatch
|
|
||||||
):
|
|
||||||
assert vector_search(client).status_code == 409
|
|
||||||
add_item(client)
|
|
||||||
unpublished = build(client)
|
|
||||||
assert (
|
|
||||||
client.post(
|
|
||||||
"/api/vector-indexes", json={"text_index_version_id": unpublished["id"]}
|
|
||||||
).status_code
|
|
||||||
== 409
|
|
||||||
)
|
|
||||||
text_ = text_version(client)
|
|
||||||
assert (
|
|
||||||
client.post(
|
|
||||||
"/api/vector-indexes", json={"text_index_version_id": unpublished["id"]}
|
|
||||||
).status_code
|
|
||||||
== 409
|
|
||||||
)
|
|
||||||
monkeypatch.setattr(vectors, "MAX_VECTOR_CHUNKS", 0)
|
|
||||||
assert (
|
|
||||||
client.post("/api/vector-indexes", json={"text_index_version_id": text_["id"]}).status_code
|
|
||||||
== 503
|
|
||||||
)
|
|
||||||
monkeypatch.undo()
|
|
||||||
with migrated_database.begin() as c:
|
|
||||||
c.execute(text("DELETE FROM chunks"))
|
|
||||||
assert (
|
|
||||||
client.post("/api/vector-indexes", json={"text_index_version_id": text_["id"]}).status_code
|
|
||||||
== 409
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def test_empty_reviewed_corpus_can_be_published_without_fabricated_results(client, provider):
|
|
||||||
version = vector_build(client, text_version(client))
|
|
||||||
assert version["chunk_count"] == 0
|
|
||||||
assert vector_publish(client, version).json()["is_current"]
|
|
||||||
result = vector_search(client).json()
|
|
||||||
assert (
|
|
||||||
result["status"] == "no_evidence" and result["results"] == [] and result["corpus_size"] == 0
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def test_metadata_and_build_require_admin_and_csrf(client, provider):
|
|
||||||
add_item(client)
|
|
||||||
version = vector_build(client, text_version(client))
|
|
||||||
assert (
|
|
||||||
client.post(
|
|
||||||
"/api/vector-indexes",
|
|
||||||
json={"text_index_version_id": version["text_index_version_id"]},
|
|
||||||
headers={"X-CSRF-Token": ""},
|
|
||||||
).status_code
|
|
||||||
== 403
|
|
||||||
)
|
|
||||||
client.cookies.clear()
|
|
||||||
for method, path in [
|
|
||||||
("GET", "/api/vector-indexes"),
|
|
||||||
("GET", "/api/vector-indexes/active"),
|
|
||||||
("GET", "/api/vector-indexes/spaces"),
|
|
||||||
("GET", "/api/vector-indexes/publications"),
|
|
||||||
("GET", f"/api/vector-indexes/{version['id']}"),
|
|
||||||
("POST", "/api/vector-indexes"),
|
|
||||||
("POST", f"/api/vector-indexes/{version['id']}/publish"),
|
|
||||||
]:
|
|
||||||
assert client.request(method, path, json={}).status_code == 401
|
|
||||||
assert (
|
|
||||||
client.post(
|
|
||||||
"/api/auth/register", json={"username": "vector_user", "password": "test-password-123"}
|
|
||||||
).status_code
|
|
||||||
== 201
|
|
||||||
)
|
|
||||||
auth = client.post(
|
|
||||||
"/api/auth/login", json={"username": "vector_user", "password": "test-password-123"}
|
|
||||||
).json()
|
|
||||||
client.headers["X-CSRF-Token"] = auth["csrf_token"]
|
|
||||||
assert client.get("/api/vector-indexes").status_code == 403
|
|
||||||
assert client.post("/api/vector-indexes", json={}).status_code == 403
|
|
||||||
|
|
||||||
|
|
||||||
def test_vector_publication_serializes_with_catalog_and_other_publishers(
|
|
||||||
client, provider, migrated_database
|
|
||||||
):
|
|
||||||
add_item(client)
|
|
||||||
text_ = text_version(client)
|
|
||||||
first, second = vector_build(client, text_), vector_build(client, text_)
|
|
||||||
with Session(migrated_database) as owner, Session(migrated_database) as other:
|
|
||||||
locked_state(owner)
|
|
||||||
other.execute(text("SET LOCAL lock_timeout = '100ms'"))
|
|
||||||
with pytest.raises(OperationalError) as blocked:
|
|
||||||
vectors.publish_vectors(other, UUID(first["id"]), 0, first["created_by"])
|
|
||||||
assert blocked.value.orig.sqlstate == "55P03"
|
|
||||||
other.rollback()
|
|
||||||
vectors.publish_vectors(owner, UUID(first["id"]), 0, first["created_by"])
|
|
||||||
with pytest.raises(HTTPException) as stale:
|
|
||||||
vectors.publish_vectors(other, UUID(second["id"]), 0, second["created_by"])
|
|
||||||
assert stale.value.status_code == 409
|
|
||||||
other.rollback()
|
|
||||||
with Session(migrated_database) as db:
|
|
||||||
assert db.scalar(select(func.count()).select_from(VectorPublication)) == 1
|
|
||||||
|
|
||||||
|
|
||||||
def test_lazy_runtime_is_offline_and_does_not_wait_when_busy(monkeypatch):
|
|
||||||
from app.services import embedding_runtime
|
|
||||||
|
|
||||||
calls = []
|
|
||||||
|
|
||||||
def load(spec, **kwargs):
|
|
||||||
calls.append(kwargs)
|
|
||||||
return ToyProvider()
|
|
||||||
|
|
||||||
monkeypatch.setattr(embedding_runtime, "LocalEmbeddingProvider", load)
|
|
||||||
runtime = EmbeddingRuntime(ToyProvider.spec, None, 8, 2)
|
|
||||||
assert calls == []
|
|
||||||
assert runtime.encode(["alpha"]) == [[1, 0, 0]]
|
|
||||||
assert calls[0]["offline"] is True
|
|
||||||
runtime.encode(["alpha"], query=True)
|
|
||||||
assert len(calls) == 1
|
|
||||||
with runtime._lock:
|
|
||||||
with pytest.raises(HTTPException) as busy:
|
|
||||||
runtime.encode(["alpha"])
|
|
||||||
assert busy.value.status_code == 503
|
|
||||||
|
|
||||||
|
|
||||||
def test_disabled_runtime_is_explicit_and_keyword_baseline_remains_available(client):
|
|
||||||
assert vector_search(client).status_code == 503
|
|
||||||
assert client.post("/api/retrieval", json={"query": "那拉提"}).status_code == 200
|
|
||||||
|
|
||||||
|
|
||||||
def test_different_width_spaces_coexist_and_queries_never_compare_across_widths(client, provider):
|
|
||||||
add_item(client)
|
|
||||||
text_ = text_version(client)
|
|
||||||
first = vector_build(client, text_)
|
|
||||||
vector_publish(client, first)
|
|
||||||
provider.spec = replace(provider.spec, dimension=2)
|
|
||||||
provider.encode = lambda texts, **kwargs: [[1, 0] for _ in texts]
|
|
||||||
second = vector_build(client, text_)
|
|
||||||
assert vector_publish(client, second, 1).status_code == 200
|
|
||||||
result = vector_search(client)
|
|
||||||
assert result.status_code == 200, result.text
|
|
||||||
assert result.json()["total_hits"] == 1 and result.json()["results"][0]["score"] == 1
|
|
||||||
assert client.get("/api/vector-indexes/spaces").json()["total"] == 2
|
|
||||||
|
|
||||||
|
|
||||||
def test_new_text_publication_requires_corresponding_vectors_and_keeps_history(client, provider):
|
|
||||||
add_item(client)
|
|
||||||
old_text = text_version(client)
|
|
||||||
old = vector_build(client, old_text)
|
|
||||||
vector_publish(client, old)
|
|
||||||
history = vector_search(client).json()
|
|
||||||
new_text = text_version(client)
|
|
||||||
active = client.get("/api/vector-indexes/active").json()
|
|
||||||
assert not active["is_current"] and active["active_version"]["id"] == old["id"]
|
|
||||||
assert vector_search(client).status_code == 409
|
|
||||||
assert vector_publish(client, old, 1).status_code == 409
|
|
||||||
replacement = vector_build(client, new_text)
|
|
||||||
assert vector_publish(client, replacement, 1).json()["is_current"]
|
|
||||||
assert client.get(f"/api/retrieval/runs/{history['run_id']}").json() == history
|
|
||||||
assert client.get("/api/vector-indexes/publications").json()["total"] == 2
|
|
||||||
|
|
||||||
|
|
||||||
def test_snapshot_text_corruption_cannot_be_encoded_published_or_served(
|
|
||||||
client, provider, migrated_database
|
|
||||||
):
|
|
||||||
add_item(client)
|
|
||||||
text_ = text_version(client)
|
|
||||||
version = vector_build(client, text_)
|
|
||||||
vector_publish(client, version)
|
|
||||||
with migrated_database.begin() as c:
|
|
||||||
c.execute(text("UPDATE chunks SET text='篡改占位'"))
|
|
||||||
assert not client.get("/api/vector-indexes/active").json()["is_current"]
|
|
||||||
assert (
|
|
||||||
client.post("/api/vector-indexes", json={"text_index_version_id": text_["id"]}).status_code
|
|
||||||
== 409
|
|
||||||
)
|
|
||||||
assert vector_publish(client, version, 1).status_code == 409
|
|
||||||
assert vector_search(client).status_code == 409
|
|
||||||
|
|
||||||
|
|
||||||
def test_request_shape_pagination_and_missing_vector_version(client, provider):
|
|
||||||
assert (
|
|
||||||
client.post(
|
|
||||||
"/api/vector-indexes", json={"text_index_version_id": str(uuid4()), "embedding": [1, 0]}
|
|
||||||
).status_code
|
|
||||||
== 422
|
|
||||||
)
|
|
||||||
assert vector_search(client, top_k=21).status_code == 422
|
|
||||||
assert vector_search(client, mode="keyword").status_code == 422
|
|
||||||
assert vector_search(client, query=" ").status_code == 422
|
|
||||||
assert client.get("/api/vector-indexes?page=0").status_code == 422
|
|
||||||
assert client.get(f"/api/vector-indexes/{uuid4()}").status_code == 404
|
|
||||||
assert client.get("/api/vector-indexes/spaces").json()["total"] == 0
|
|
||||||
|
|
||||||
|
|
||||||
def test_blank_optional_cache_path_is_not_the_working_directory():
|
|
||||||
from app.core.config import Settings
|
|
||||||
|
|
||||||
assert Settings(embedding_cache_dir="", _env_file=None).embedding_cache_dir is None
|
|
||||||
@@ -14,7 +14,7 @@ FastAPI + Pydantic
|
|||||||
|-- 来源获取 -> 人工核验 -> 文本切分 -> 生成向量 -> 发布版本
|
|-- 来源获取 -> 人工核验 -> 文本切分 -> 生成向量 -> 发布版本
|
||||||
```
|
```
|
||||||
|
|
||||||
当前后端实现健康/数据库就绪检查、来源与文档管理、知识增删改查、分页筛选、人工审核、历史快照、认证会话/角色权限、文本/向量快照及独立关键词/向量/RRF 检索;问答仍为占位接口。数据库由 Alembic 显式迁移并启用 pgvector,RAG 待后续实现。前端来源登记/查询已接入,实验页仍为静态原型,文档、审核、完整分页与检索页面需继续联调。
|
当前后端实现健康/数据库就绪检查、来源与文档管理、知识增删改查、分页筛选、人工审核、历史快照、认证会话/角色权限及独立关键词检索;问答仍为占位接口。数据库由 Alembic 显式迁移并启用 pgvector,尚无向量字段、向量索引和 RAG。前端来源登记/查询已接入,实验页仍为静态原型,文档、审核、完整分页与检索页面需继续联调。
|
||||||
|
|
||||||
## 阶段 2 数据边界
|
## 阶段 2 数据边界
|
||||||
|
|
||||||
@@ -37,7 +37,7 @@ FastAPI + Pydantic
|
|||||||
|
|
||||||
新增 `retrieval_runs` 保存查询、模式、语料指纹和完整结果快照,支持分页读取及按 UUID 复核。零命中如实记录;当前依据修改后失效不会改写历史,但后续检索必须重新检查审核与父记录状态。响应耗时是本次真实读取、分词、排序及片段生成耗时,不含写日志或模型调用。
|
新增 `retrieval_runs` 保存查询、模式、语料指纹和完整结果快照,支持分页读取及按 UUID 复核。零命中如实记录;当前依据修改后失效不会改写历史,但后续检索必须重新检查审核与父记录状态。响应耗时是本次真实读取、分词、排序及片段生成耗时,不含写日志或模型调用。
|
||||||
|
|
||||||
关键词基线仍按知识条目检索,显示片段最长 240 字符;独立的持久化 chunks 和文本索引版本见下方。该接口不读取 chunks;chunk 与向量/RRF 的独立接口见后文。全语料计算明确限制条目数及总字符数,超限报错,适用于首批小规模资料与后续对照实验准备,不能当作生产检索性能结论。
|
关键词基线仍按知识条目检索,显示片段最长 240 字符;独立的持久化 chunks 和文本索引版本见下方。当前尚无倒排索引、Embedding、向量检索、RRF 或 RAG。全语料计算明确限制条目数及总字符数,超限报错,适用于首批小规模资料与后续对照实验准备,不能当作生产检索性能结论。
|
||||||
|
|
||||||
|
|
||||||
## 切分与索引版本(T05)
|
## 切分与索引版本(T05)
|
||||||
@@ -47,10 +47,3 @@ FastAPI + Pydantic
|
|||||||
目录写入、构建与发布共用状态行锁,目录写入先于父记录锁获取该锁并递增修订号。发布核对代数、修订号和当前语料指纹;失败回滚,保留旧版本。目录变化后旧指针仍可追溯,但 is_current=false,不可据此绕过正式审核。历史快照不会被改写。接口均限管理员并沿用 Cookie/CSRF,完整边界见 [INDEX_API.md](INDEX_API.md)。
|
目录写入、构建与发布共用状态行锁,目录写入先于父记录锁获取该锁并递增修订号。发布核对代数、修订号和当前语料指纹;失败回滚,保留旧版本。目录变化后旧指针仍可追溯,但 is_current=false,不可据此绕过正式审核。历史快照不会被改写。接口均限管理员并沿用 Cookie/CSRF,完整边界见 [INDEX_API.md](INDEX_API.md)。
|
||||||
|
|
||||||
这只是后续索引的文本版本基础,关键词基线行为不变;没有生成向量或宣称已实现 ANN/混合检索。同步构建和全局写锁面向当前有限语料,后续异步/增量发布须重新设计并验证并发协议。
|
这只是后续索引的文本版本基础,关键词基线行为不变;没有生成向量或宣称已实现 ANN/混合检索。同步构建和全局写锁面向当前有限语料,后续异步/增量发布须重新设计并验证并发协议。
|
||||||
|
|
||||||
|
|
||||||
## 向量与 RRF 检索(T07)
|
|
||||||
|
|
||||||
文本发布之后,固定 Embedding 空间的完整向量快照通过独立代数发布,精确余弦检索只使用当前审核与发布版本。编码在发布锁外进行,完成后重新核对语料、空间及发布状态,失败保留旧版本。运行与空间身份见 [VECTOR_API.md](VECTOR_API.md)。
|
|
||||||
|
|
||||||
新增 chunk 正文 BM25 与等权 RRF;两路限定同一文本/向量快照和分类,固定 k=60,先各取 candidate_k 再融合。保存完整候选排名、原始分数、贡献、切分/分词/BM25/模型配置及出处。旧条目关键词模式保留,评测不能混同两种证据单元。两路排序至检索记录写入共用当前性锁,面向最多 2000 chunks 的同步精确计算。字段、失败及回滚边界见 [HYBRID_API.md](HYBRID_API.md)。RAG、ANN 与真实领域质量验收仍待后续阶段。
|
|
||||||
|
|||||||
+1
-1
@@ -54,7 +54,7 @@ python -m app.manage create-admin --username local_admin
|
|||||||
- `knowledge`、`sources`、`documents` 的全部读写及历史、管理员用户列表、检索历史列表/详情均要求 `admin`。因此旧匿名知识管理调用现在返回 401,普通用户返回 403。健康检查、公开的已核验关键词检索和当前占位问答保持公开;正式问答、实验接口接入时须重新明确各自授权。
|
- `knowledge`、`sources`、`documents` 的全部读写及历史、管理员用户列表、检索历史列表/详情均要求 `admin`。因此旧匿名知识管理调用现在返回 401,普通用户返回 403。健康检查、公开的已核验关键词检索和当前占位问答保持公开;正式问答、实验接口接入时须重新明确各自授权。
|
||||||
- 登录及注册共用每客户端地址 30 次/5 分钟预算;登录另有规范用户名 10 次/5 分钟预算,成功/失败都计数。计数通过 PostgreSQL 原子更新,跨进程/重启有效;仅保存带类型前缀的摘要,旧预算在下一次认证请求时清理。此处为基础防暴力尝试,不代替部署网关限流与监控。
|
- 登录及注册共用每客户端地址 30 次/5 分钟预算;登录另有规范用户名 10 次/5 分钟预算,成功/失败都计数。计数通过 PostgreSQL 原子更新,跨进程/重启有效;仅保存带类型前缀的摘要,旧预算在下一次认证请求时清理。此处为基础防暴力尝试,不代替部署网关限流与监控。
|
||||||
|
|
||||||
前端 `src/services/auth.ts` 和 `src/services/session.ts` 提供真实 Cookie 请求封装:CSRF 只存内存,刷新后通过 `/me` 恢复;知识写入也使用同一封装。`dev` 已提供登录、注册、个人中心、账户设置、管理员用户列表与路由守卫;导航菜单按当前角色隐藏管理员入口。账户资料更新与改密均调用真实接口,前端的新密码规则与后端一致为至少 15 个字符。浏览器端 Cookie、CORS 和 PostgreSQL 联调仍须在目标部署环境验收,不能由静态构建替代。
|
前端 `src/services/auth.ts` 和 `src/services/session.ts` 提供真实 Cookie 请求封装:CSRF 只存内存,刷新后通过 `/me` 恢复;知识写入也使用同一封装。登录入口、路由守卫和权限页面交互仍由前端接入;远程 main 注册原型的最低密码提示须同步为 15 字符。当前 `dev` 页面没有完整登录流程,不能把 API 完成视作浏览器验收完成。
|
||||||
|
|
||||||
## 验证与回滚
|
## 验证与回滚
|
||||||
|
|
||||||
|
|||||||
@@ -42,10 +42,3 @@
|
|||||||
状态:T05 分支实现,待审阅。先对当前有效审核语料构建带原文偏移/哈希/出处的 chunks,再以独立的发布请求切换指针。固定 Unicode 字符窗口配置留存在版本中,关键词接口继续按条目运行,防止未经评估就改变实验口径。
|
状态:T05 分支实现,待审阅。先对当前有效审核语料构建带原文偏移/哈希/出处的 chunks,再以独立的发布请求切换指针。固定 Unicode 字符窗口配置留存在版本中,关键词接口继续按条目运行,防止未经评估就改变实验口径。
|
||||||
|
|
||||||
为当前小规模语料选择统一数据库行锁:应用目录写入、构建、发布串行,发布同时校验代数、目录修订和语料指纹;失败回滚全部更新,旧版本保留为历史。任何成功目录写入保守地使旧索引过期,历史不能视作当前有效依据。实现、容量与后续并发扩展边界见 [INDEX_API.md](INDEX_API.md)。
|
为当前小规模语料选择统一数据库行锁:应用目录写入、构建、发布串行,发布同时校验代数、目录修订和语料指纹;失败回滚全部更新,旧版本保留为历史。任何成功目录写入保守地使旧索引过期,历史不能视作当前有效依据。实现、容量与后续并发扩展边界见 [INDEX_API.md](INDEX_API.md)。
|
||||||
|
|
||||||
|
|
||||||
## ADR-011 同粒度 chunk 排序与等权 RRF
|
|
||||||
|
|
||||||
状态:后端分支实现,待审阅。基于已合入的向量快照,新增正文 BM25 的 chunk 基线,与向量模式使用同一文本发布及分类范围;保留条目关键词接口作为独立口径。两路候选各限 candidate_k,使用从 1 起的排名、固定 k=60、等权倒数排名融合,缺席一路贡献为零。保存候选全排名、原始分数、两路贡献和配置,确保实验准备可复核;不把 BM25 与余弦原始分数直接相加。
|
|
||||||
|
|
||||||
公式与常数参考 [RRF 原论文](https://cormack.uwaterloo.ca/cormacksigir09-rrf.pdf),本项目尚未依据核验领域问题集调优。同步精确计算限制 2000 chunks,固定配置、版本一致性、历史兼容与回滚见 [HYBRID_API.md](HYBRID_API.md)。
|
|
||||||
|
|||||||
@@ -1,67 +0,0 @@
|
|||||||
# Chunk 关键词与 RRF 混合检索
|
|
||||||
|
|
||||||
本阶段基于已合入 `dev` 的文本/向量发布,增加可独立调用的 chunk BM25 和 RRF。既有条目关键词接口继续保留。三个 chunk 模式使用相同的当前文本版本、原文 chunk 与分类范围;比较时应固定发布版本、分类和 `top_k`,同时记录混合检索的 `candidate_k`。
|
|
||||||
|
|
||||||
| 接口 | 请求 mode | 排序/证据单元 | 前置条件 |
|
|
||||||
|---|---|---|---|
|
|
||||||
| `POST /api/retrieval` | `keyword` | 知识条目,标题权重 2、正文权重 1 | 当前有效审核语料 |
|
|
||||||
| `POST /api/retrieval/chunks` | `keyword_chunks` | chunk 正文 BM25,标题不参与打分 | 当前有效文本发布;无需模型或向量发布 |
|
|
||||||
| `POST /api/retrieval/vector` | `vector` | chunk 精确余弦 | 当前有效文本/向量发布,运行模型空间匹配 |
|
|
||||||
| `POST /api/retrieval/hybrid` | `hybrid` | 同一批 chunk 的两路 RRF | 与向量模式相同 |
|
|
||||||
|
|
||||||
`mode` 可省略,由接口固定;传入其他模式或未知字段返回 422。公开检索无需登录;历史分页及详情继续限管理员。问答接口仍为占位,当前响应不生成 LLM 答案。
|
|
||||||
|
|
||||||
## 请求与排名
|
|
||||||
|
|
||||||
```json
|
|
||||||
{"query":"合成资料中的交通查询入口","mode":"hybrid","top_k":5,"candidate_k":50,"category":"交通"}
|
|
||||||
```
|
|
||||||
|
|
||||||
- `query` 去除首尾空白后 1–500 字符,`top_k` 默认 5、范围 1–20,`category` 可省略。chunk 关键词请求不接收 `candidate_k`。
|
|
||||||
- 混合检索每路最多取 `candidate_k` 条候选,默认 50、范围 1–200,并须不小于 `top_k`。这是固定排名后的候选截断,不会改变文本或输入 token。最终从候选并集返回最多 `top_k` 条。
|
|
||||||
- chunk BM25 只对正文分词。沿用固定 Jieba 0.42.1 搜索分词、NFKC/casefold、HMM 关闭、项目词与 `zh-small-v1` 停用词表;查询词去重,`k1=1.2`、`b=0.75`,正值平滑 IDF。语料长度和词频在分类过滤后的全部 chunks 上计算,只将正分候选纳入关键词一路。标题仅作为出处字段。
|
|
||||||
- 向量一路复用精确余弦排序,同样先按分类筛选,不设置相似度阈值。负分和零分仍可能进入候选,分数不是事实置信度,`status=ok` 只表示存在排序结果。
|
|
||||||
- 两路先按各自原始分数排序;同分按 `knowledge_id, chunk_ordinal, chunk_id` 升序,排名从 1 开始。RRF 使用等权、固定 `k=60`:`score(c) = sum(1 / (60 + rank_leg(c)))`。不在某路候选列表中的 chunk,该路贡献为 0。原始 BM25 和余弦分数不直接相加;融合分数同分沿用相同 ID 次序。排序与日志使用未预先四舍五入的分数。
|
|
||||||
- 融合以 chunk ID 去重。同一知识的不同片段(含重叠片段)保留各自排名和原文偏移,不做条目折叠或多样性重排。条目基线与 chunk 模式的分数、命中数、语料数及评测单元不同,不能直接混算指标。
|
|
||||||
|
|
||||||
RRF 方法与常数参考 [Cormack、Clarke、Büttcher,SIGIR 2009 原论文](https://cormack.uwaterloo.ca/cormacksigir09-rrf.pdf)。这里把 `k=60` 作为首版可复现配置,并未通过本项目领域问题集调优。
|
|
||||||
|
|
||||||
## 响应与复核
|
|
||||||
|
|
||||||
全部模式保留原文、字符偏移(左闭右开)、知识/文档/来源 ID 及版本、URL、排名与耗时。chunk 关键词和混合响应另含:
|
|
||||||
|
|
||||||
| 字段 | 含义 |
|
|
||||||
|---|---|
|
|
||||||
| `mode` | `keyword_chunks` 或 `hybrid`,可用于历史类型判别 |
|
|
||||||
| `text_index_version_id`, `text_generation`, `catalog_revision` | 本次文本发布和目录修订 |
|
|
||||||
| `corpus_fingerprint` | 整份已审核语料的指纹,分类筛选不改变它 |
|
|
||||||
| `corpus_size` | 分类范围内参与两路打分的 chunk 数 |
|
|
||||||
| `total_hits` | 关键词模式为全部正分 chunk 数;混合模式为两路截断后的候选并集大小,可能大于返回条数 |
|
|
||||||
| `query_terms`, `config.lexical`, `config.chunking`, `config.tie_break` | 实际分词、BM25 参数、切分配置、稳定排序规则 |
|
|
||||||
| `elapsed_ms` | 语料检查、模型编码(混合模式)、排序及响应组装耗时;不含记录写入及网络时间 |
|
|
||||||
|
|
||||||
混合模式还保存:
|
|
||||||
|
|
||||||
- `vector_version_id`、`generation`:向量发布版本与代数;`config.space` 包含完整模型 revision、维度、token 预算、指令、实现/池化/归一化与权重格式身份;与向量契约一致。
|
|
||||||
- `config.rrf_k/weights/candidate_k/vector_scorer`:融合与候选预算,不允许请求覆盖固定算法参数。
|
|
||||||
- `lexical_candidates/vector_candidates`:两路完整的截断候选列表,每条为 `chunk_id/rank/score`,即使未进最终前 `top_k` 也保留。
|
|
||||||
- `lexical_total_hits/vector_total_hits`:截断前正分关键词候选数、分类范围内向量候选总数。
|
|
||||||
- 命中的 `lexical_rank/lexical_score/vector_rank/vector_score` 和两路 `*_contribution`:缺席的排名/原始分数为 `null`、贡献为 0。`score` 等于两路贡献之和。
|
|
||||||
|
|
||||||
结果完整写入现有 `retrieval_runs`,历史接口联合类型支持四种 mode,不需要新增迁移,当前 Schema 仍为 `0006_vectors`。修改依据后历史保持原样,只代表当时快照;重查须通过当前性验证。原始查询和运行记录属于数据库数据,不应提交到仓库。
|
|
||||||
|
|
||||||
## 一致性、失败与容量
|
|
||||||
|
|
||||||
混合检索先检查发布版本与模型空间,然后提交只读事务释放发布锁,再真实编码查询;编码后重新获取目录/向量状态锁,复核目录修订、文本/向量代数、审核指纹、实际 chunks、完整向量集及空间身份。变化时 409,且不保存部分结果。两路排名及写入检索历史在同一个受锁保护的事务内完成。
|
|
||||||
|
|
||||||
关键词一路没有命中或没有分词时,混合模式明确保留空候选列表,可仅有向量一路;纯关键词模式没有可检索词则 422。分类无资料或空的有效发布返回 `no_evidence`。模型 token 超限/输出校验失败返回 422,运行环境不可用或编码忙返回 503;模型失败不降级成一份声称成功的混合结果。
|
|
||||||
|
|
||||||
两种新接口均沿用向量阶段最多 2000 chunks、5000 条知识/200 万字符的整体核验预算,超限明确 503。只在分类范围内打分,但当前性及容量检查针对整份发布。文本快照允许的 20000 chunks 不等同于本接口容量。同步精确排序及 BM25 在锁内运行,会短暂串行化目录写入;当前没有 ANN、后台队列或生产吞吐量结论。
|
|
||||||
|
|
||||||
运行依赖、固定缓存、默认关闭的本地模型、构建/发布步骤见 [VECTOR_API.md](VECTOR_API.md)。前端新增 `services/hybrid.ts` 两个调用及历史类型;模式切换页面、Docker/浏览器端和真实领域验收仍待交接。
|
|
||||||
|
|
||||||
## 验证与回滚
|
|
||||||
|
|
||||||
轻量测试使用明确的合成正文及坐标 fixture,覆盖手算 BM25/RRF、原始分数与融合的独立性、候选边界、两路缺席、稳定排序、chunk 偏移、分类、权限、历史、失效/破坏快照、编码期间版本竞争、无部分日志及容量拒绝。真实模型专项沿用 `test_vector_runtime.py` 的显式 opt-in,增加固定 BGE small 的 HTTP 混合检索和进程重启后的两路排名/历史复核。运行命令见 [VECTOR_API.md](VECTOR_API.md)。这些用例验证实际模型/数据库/API 贯通,合成问题不代表领域质量或论文实验结论。
|
|
||||||
|
|
||||||
撤回本功能代码即可撤回两个新增接口;数据库无需降级。旧后端历史响应模型不认识 `keyword_chunks/hybrid`,回滚时应限制旧历史端点或先以备份保留新模式记录,避免把新记录交给旧模型解析;不要为回滚而无备份删除检索历史。
|
|
||||||
@@ -74,6 +74,4 @@ export HF_HUB_DISABLE_XET=1
|
|||||||
|
|
||||||
将 Provider 接入已发布的文本版本,完成 pgvector 存储与严格空间隔离、向量检索及 RRF,再使用真实审核语料评测。模型编码耗时应在发布锁外运行,发布前重新核对目录/语料版本;过期或失败时保留旧版本。当前没有新增向量数据库字段、检索 API 或论文结论。
|
将 Provider 接入已发布的文本版本,完成 pgvector 存储与严格空间隔离、向量检索及 RRF,再使用真实审核语料评测。模型编码耗时应在发布锁外运行,发布前重新核对目录/语料版本;过期或失败时保留旧版本。当前没有新增向量数据库字段、检索 API 或论文结论。
|
||||||
|
|
||||||
以上描述 PR #9 独立评估阶段的范围;后续向量存储与精确检索集成另见 [VECTOR_API.md](VECTOR_API.md),不改变本页已记录的合成评估结果。
|
|
||||||
|
|
||||||
依据:[BGE small 官方模型说明](https://huggingface.co/BAAI/bge-small-zh-v1.5)、[BGE base 官方文件](https://huggingface.co/BAAI/bge-base-zh-v1.5/tree/f03589ceff5aac7111bd60cfc7d497ca17ecac65)、[Transformers 模型加载参数](https://huggingface.co/docs/transformers/main_classes/model#transformers.PreTrainedModel.from_pretrained)、[PyTorch CPU 安装](https://pytorch.org/get-started/locally/)、[Hugging Face 固定修订下载](https://huggingface.co/docs/huggingface_hub/guides/download)。
|
依据:[BGE small 官方模型说明](https://huggingface.co/BAAI/bge-small-zh-v1.5)、[BGE base 官方文件](https://huggingface.co/BAAI/bge-base-zh-v1.5/tree/f03589ceff5aac7111bd60cfc7d497ca17ecac65)、[Transformers 模型加载参数](https://huggingface.co/docs/transformers/main_classes/model#transformers.PreTrainedModel.from_pretrained)、[PyTorch CPU 安装](https://pytorch.org/get-started/locally/)、[Hugging Face 固定修订下载](https://huggingface.co/docs/huggingface_hub/guides/download)。
|
||||||
|
|||||||
@@ -70,12 +70,3 @@
|
|||||||
- 管理接口统一要求 admin 与写入 CSRF;前端只同步类型和调用层,页面仍待联调。现有关键词基线行为不变,向量、ANN、RRF 与 RAG 未在本阶段实现。
|
- 管理接口统一要求 admin 与写入 CSRF;前端只同步类型和调用层,页面仍待联调。现有关键词基线行为不变,向量、ANN、RRF 与 RAG 未在本阶段实现。
|
||||||
- 实际验证:`ruff check .`、`ruff format --check .` 通过;`pytest -q` **91 passed**(隔离 PostgreSQL 16 + pgvector),包含真实锁竞争、失败构建/发布的数据库约束回滚、迁移保留账号/知识/检索记录、原文重建与来源快照、真实 HTTP 重启后的发布指针和 chunks 持久化。修正了发布时间在 UTC 与数据库时区之间表示不一致的问题,索引接口统一 UTC 输出。前端 `npm run build` 通过。
|
- 实际验证:`ruff check .`、`ruff format --check .` 通过;`pytest -q` **91 passed**(隔离 PostgreSQL 16 + pgvector),包含真实锁竞争、失败构建/发布的数据库约束回滚、迁移保留账号/知识/检索记录、原文重建与来源快照、真实 HTTP 重启后的发布指针和 chunks 持久化。修正了发布时间在 UTC 与数据库时区之间表示不一致的问题,索引接口统一 UTC 输出。前端 `npm run build` 通过。
|
||||||
- 完整请求/响应、切分/容量、失效策略及回滚说明见 [INDEX_API.md](INDEX_API.md)。本阶段仍是有限语料的同步文本快照;Docker/浏览器验收、独立 T05 Issue 补建及后续 Embedding 方案选择仍有待完成。
|
- 完整请求/响应、切分/容量、失效策略及回滚说明见 [INDEX_API.md](INDEX_API.md)。本阶段仍是有限语料的同步文本快照;Docker/浏览器验收、独立 T05 Issue 补建及后续 Embedding 方案选择仍有待完成。
|
||||||
|
|
||||||
|
|
||||||
## 2026-10-16 T07:合并后的 RRF 混合检索
|
|
||||||
|
|
||||||
- 本地模型评估和向量持久化/精确检索分别经 PR #11/#13 合入 `dev`;#9/#10 是被替代的旧 AGit 入口。读取已确认合并结果后,从最新 `dev` 创建 `feat/backend-rrf-retrieval`,遵循每阶段 PR 合并后才开发下一阶段。
|
|
||||||
- 新增公开 chunk 正文 BM25 与 RRF 接口,三种 chunk 模式限定同一文本快照及分类;原条目关键词接口保留独立口径。固定 k=60、等权、每路候选预算,记录两路完整候选排名/原始分数、贡献、模型空间、切分/BM25/分词参数与出处。前端仅同步调用与历史类型。
|
|
||||||
- 模型编码在发布锁外,完成后重新检查审核语料、文本/向量代数与空间;两路排序和记录保存受同一事务保护。竞争、模型或写入失败不保存部分混合结果;2000 chunks 容量与原向量阶段一致。没有新增数据库迁移。完整字段、比较口径及回滚见 [HYBRID_API.md](HYBRID_API.md)。
|
|
||||||
- 常规测试 `pytest -q` **180 passed, 1 skipped**(真实模型专项须显式启用);新增 27 项覆盖手算排序/融合、候选截断/缺席/同分、相同粒度及分类、原文、历史、权限、编码期间锁释放和版本竞争、损坏/失效快照、失败回滚与容量。`ruff check`、`ruff format --check` 和前端 `npm run build` 通过。真实模型专项在独立 CPU 环境另执行 **1 passed**:固定 BGE small 经真实 HTTP 完成向量/混合/关键词 chunk 查询,进程重启后两路候选排名、结果和历史保持一致。首次专项请求因 candidate_k 小于默认 top_k 被按契约拒绝,修正测试请求后通过;未改变该输入约束。
|
|
||||||
- Issue #5 本轮复查无新评论,后端实现已合入,前端登录/CSRF/角色页面及浏览器流程仍待交接;Docker 容器、核验领域语料/问题集、正式模型冻结与领域评测同样未验收。所有新增自动化资料均为合成测试,不作为文旅事实或论文分数。下一阶段须等待本轮 PR 真正合入 `dev`。
|
|
||||||
|
|||||||
@@ -1,116 +0,0 @@
|
|||||||
# T07 向量快照、发布与精确检索
|
|
||||||
|
|
||||||
本分支 `feat/backend-vector-retrieval` 组合待审 #8 的文本版本与 #9 的固定版本 CPU Provider,继承 #7/#6。合并到 `dev` 前须按依赖审阅;不自行批准或合并。双方阶段文档均保留,独立阶段差异从组合基线 `bd0b738` 起查看。
|
|
||||||
|
|
||||||
交付 pgvector 存储、空间约束、独立发布、精确余弦检索与历史。关键词 `POST /api/retrieval` 保持原请求/分数/条目粒度;向量使用新的 `POST /api/retrieval/vector`,粒度为 chunk。本阶段尚未实现 RRF、ANN、RAG、持久化后台任务或正式领域评测。
|
|
||||||
|
|
||||||
## 配置与迁移
|
|
||||||
|
|
||||||
执行 `alembic upgrade head` 到 `0006_vectors`。迁移增加 `embedding_spaces`、`vector_versions`、`chunk_vectors`、`vector_state` 和 `vector_publications`,以及 chunks 的复合唯一约束。保留账号、目录、原文版本及既有检索日志,不预置模型或向量。
|
|
||||||
|
|
||||||
扩展类型/函数/距离算子显式引用 `public.vector`,业务表仍放在连接的当前 Schema,不把 public 加入测试搜索路径。已有 vector 扩展须位于 public;其他安装位置会明确报错,迁移不会移动共享扩展。新数据库可先由有权限的数据库管理员执行 `CREATE EXTENSION IF NOT EXISTS vector WITH SCHEMA public`。
|
|
||||||
|
|
||||||
默认 `EMBEDDING_ENABLED=false`,普通 API 启动不导入 PyTorch 或加载权重。模型相关请求返回 503,不回退占位向量或伪造空结果;关键词和管理元数据接口可继续使用。启用前按 [本地评估说明](LOCAL_EMBEDDING_EVALUATION.md) 安装独立 CPU 环境、缓存固定修订;该运行环境还需安装 `backend/requirements.txt` 的 API 依赖。
|
|
||||||
|
|
||||||
| 变量 | 默认值 | 作用 |
|
|
||||||
|---|---|---|
|
|
||||||
| `EMBEDDING_ENABLED` | false | 显式启用本地编码 |
|
|
||||||
| `EMBEDDING_MODEL` | small | 仅 small/base;完整修订、维度等由固定 Spec 决定 |
|
|
||||||
| `EMBEDDING_CACHE_DIR` | 未设置 | 固定模型缓存目录;空值使用库默认缓存 |
|
|
||||||
| `EMBEDDING_BATCH_SIZE` | 8 | 1–32 |
|
|
||||||
| `EMBEDDING_THREADS` | 2 | 1–4 个 CPU 线程 |
|
|
||||||
|
|
||||||
开发资源优先暂用 small,正式模型仍待已核验领域资料/问题集评测。API 始终只从本机固定修订缓存读取,禁止在请求中下载权重,不接收模型路径、任意仓库名或用户向量。CPU Provider 首次使用才加载,单进程同时只允许一个编码调用,繁忙立即返回 503;当前建议单 worker 本机运行,不能据此宣称已实现跨进程队列或生产并发容量。
|
|
||||||
|
|
||||||
Linux 仓库根目录示例(先完成模型准备及数据库迁移;本机 Cookie/Origin 配置仍按认证说明设置):
|
|
||||||
|
|
||||||
```bash
|
|
||||||
task_vector_root=$(git rev-parse --absolute-git-dir)
|
|
||||||
uv pip install --python "$task_vector_root/embedding-eval-venv/bin/python" \
|
|
||||||
-r backend/requirements.txt
|
|
||||||
export EMBEDDING_ENABLED=true EMBEDDING_MODEL=small
|
|
||||||
export EMBEDDING_CACHE_DIR="$task_vector_root/embedding-model-cache"
|
|
||||||
export HF_HUB_OFFLINE=1 HF_HUB_DISABLE_TELEMETRY=1 HF_HUB_DISABLE_IMPLICIT_TOKEN=1
|
|
||||||
export TOKENIZERS_PARALLELISM=false
|
|
||||||
cd backend
|
|
||||||
"$task_vector_root/embedding-eval-venv/bin/python" -m uvicorn app.main:app --host 127.0.0.1
|
|
||||||
```
|
|
||||||
|
|
||||||
环境、模型权重和实际日志只留本机,不提交。普通 API 环境不自动增加 PyTorch 依赖。
|
|
||||||
|
|
||||||
## 构建与发布协议
|
|
||||||
|
|
||||||
1. 通过 `/api/indexes` 构建并发布当前有效审核语料的文本版本。建议开发先用 `chunk_size=200, overlap=20`;实际 tokenizer 仍须检查,字符数不保证 token 数。
|
|
||||||
2. 管理员向 `POST /api/vector-indexes` 提交 `{"text_index_version_id":"<已发布文本版本 UUID>"}`。服务仅接受当前有效的文本发布版本;整份快照最多 2000 chunks,超限返回 503,不截断语料。
|
|
||||||
3. 服务短暂锁住目录/文本状态,复核实际审核语料指纹、原文窗口、偏移、哈希、出处和完整数量;复制输入后提交并释放事务。模型加载与正文编码全部在锁外运行。
|
|
||||||
4. 编码成功且向量数量/维度/有限数/L2 范数通过后,再取得同一状态锁,重新校验文本代数、目录修订、当前指针、语料指纹及实际 chunks。变化则 409,整批不入库。成功只保存 ready 向量版本,不自动发布。
|
|
||||||
5. 管理员向 `POST /api/vector-indexes/{id}/publish` 提交 `{"expected_generation":0}`。向量代数独立于文本代数;原子切换当前指针、前后版本状态和发布历史。旧代数、过期文本或不完整向量集均拒绝。相同当前版本和当前代数为幂等读取。
|
|
||||||
|
|
||||||
上述构建同步执行,没有后台任务 ID 或自动重试。超过模型 token 上限返回 422,应重新切分并发布文本,再重建;不会静默截断。编码不可用/繁忙为 503;数据库约束或版本竞争为 409。失败不会改写旧指针或留下部分向量版本。空的有效审核语料允许生成/发布 0 chunks 的版本;随后检索如实返回 `no_evidence`。
|
|
||||||
|
|
||||||
目录变化会使既有文本和向量版本过期;文本切换即使内容相同也要求对应新文本版本的向量。旧指针和日志保留为历史,公开检索对过期状态返回 409,不继续使用旧审核依据。切换模型配置后,若当前空间不一致,也须重新构建/发布相应空间或恢复匹配的运行配置。
|
|
||||||
|
|
||||||
## 数据库空间隔离
|
|
||||||
|
|
||||||
`space_id` 为固定 Spec 的 SHA-256,包含模型名、完整 revision、维度、token 上限、查询指令、编码实现(CLS/L2/CPU FP32)及权重格式。同维度的不同修订/指令/配置仍是不同空间。
|
|
||||||
|
|
||||||
数据库同时约束:
|
|
||||||
|
|
||||||
- 版本的 `(space_id, dimension)` 必须引用同一 EmbeddingSpace。
|
|
||||||
- 每条向量的 `(version_id, text_index_version_id, space_id, dimension)` 必须引用同一向量版本。
|
|
||||||
- `(chunk_id, text_index_version_id)` 必须引用对应文本版本中的 chunk。
|
|
||||||
- `public.vector_dims` 必须等于声明维度;`public.vector_norm` 须在 `0.9999–1.0001`,pgvector 拒绝非有限元素。
|
|
||||||
|
|
||||||
使用无固定维度的 vector 列容纳独立空间,查询显式限定版本、空间、文本版本和维度,然后使用余弦距离。当前为有限语料精确扫描,没有 HNSW/IVFFlat;模型空间和向量发布均不由客户端参数任意指定。
|
|
||||||
|
|
||||||
## 接口与字段交接
|
|
||||||
|
|
||||||
`/api/vector-indexes` 下所有读写要求 admin;写入沿用当前 Cookie、CSRF 和明确 Origin。元数据不要求加载模型。分页使用 `{items,total,page,page_size}`,`page_size` 最大 100。
|
|
||||||
|
|
||||||
| 方法与路径 | 请求/用途 | 响应 |
|
|
||||||
|---|---|---|
|
|
||||||
| POST `/api/vector-indexes` | `text_index_version_id` | 201,VectorVersion |
|
|
||||||
| GET `/api/vector-indexes` | page/page_size | 版本分页 |
|
|
||||||
| GET `/api/vector-indexes/{id}` | UUID | VectorVersion,缺失 404 |
|
|
||||||
| GET `/api/vector-indexes/active` | 当前元数据 | generation/text_generation/catalog_revision/is_current/active_version |
|
|
||||||
| GET `/api/vector-indexes/spaces` | page/page_size | 空间 id/dimension/spec |
|
|
||||||
| POST `/api/vector-indexes/{id}/publish` | expected_generation | 当前向量元数据 |
|
|
||||||
| GET `/api/vector-indexes/publications` | page/page_size | 代数、前后向量版本、actor_id、UTC 时间 |
|
|
||||||
| POST `/api/retrieval/vector` | 下方请求 | VectorRetrievalResponse |
|
|
||||||
|
|
||||||
VectorVersion 字段:`id/text_index_version_id/space_id/dimension/chunk_count/status/created_by/created_at/published_at`,状态为 ready/published/superseded;时间统一 UTC ISO 8601。发布历史的 `version_id` 是向量版本,不能与文本发布历史混用。
|
|
||||||
|
|
||||||
公开检索请求示例:
|
|
||||||
|
|
||||||
```json
|
|
||||||
{"query":"待检索问题","mode":"vector","top_k":5,"category":"交通"}
|
|
||||||
```
|
|
||||||
|
|
||||||
`mode` 可省略,固定为 vector;query 1–500 字符,top_k 1–20,category 可省略。未知字段拒绝。流程在查询编码前后均复核当前发布/审核状态和完整向量集;编码时不持有目录发布锁,期间任何发布代数变化都会返回 409,重试使用新快照。
|
|
||||||
|
|
||||||
响应记录:`run_id/query/mode/status/top_k/category/vector_version_id/text_index_version_id/generation/text_generation/corpus_fingerprint/corpus_size/total_hits/results/elapsed_ms/created_at/config`。`config` 固定 scorer=`pgvector-cosine-exact-v1`、ranking_unit=`chunk`,并保存完整空间 Spec。
|
|
||||||
|
|
||||||
每个 result 沿用关键词出处字段,再加 `chunk_id/chunk_ordinal`;snippet 为完整原文 chunk,snippet_start/end 是原正文的 Unicode 字符偏移(左闭右开),matched_terms 为空数组。score 为 `1 - cosine_distance`,不是正确率或置信度,也未设领域无依据阈值。按实际距离、knowledge_id、chunk_ordinal、chunk_id 排序;同一知识可出现多个 chunks,不能把结果数量当作命中的知识数。
|
|
||||||
|
|
||||||
`corpus_size/total_hits` 为分类筛选后的候选 chunk 数,不是超过置信阈值的数量;corpus_fingerprint 是整个已发布文本语料指纹,分类另行保存。耗时包括两次审核/快照检查、查询编码(首次可含模型加载)、SQL 距离排序与片段生成,不含最终日志提交及网络。不同于关键词基线的条目粒度,正式对照前必须统一评测单位。
|
|
||||||
|
|
||||||
成功和零候选均保存 `retrieval_runs` 不可变快照。既有历史接口现在返回 keyword/vector 判别联合,仍限 admin;不改写过去结果,也不因历史可读取就代表资料仍有效。前端调用层已同步 `services/vectors.ts` 与历史联合类型,页面和浏览器仍待接入验收。
|
|
||||||
|
|
||||||
## 回滚与验证
|
|
||||||
|
|
||||||
备份后 `alembic downgrade 0005_chunk_indexes` 删除本阶段空间、向量、向量状态和发布历史,保留文本 chunks/发布、账号、知识及所有检索结果快照。切回对应旧版后端前须处理其历史接口只认识 keyword 响应的兼容性;重升后向量指针为空,需要重新构建,不会复原被删权重结果。当前 API 启动会拒绝旧 Schema。
|
|
||||||
|
|
||||||
默认测试用明确的三维坐标 fixture 验证 SQL 数学、空间隔离、并发与回滚,绝不代表实际模型质量。真实模型贯通用例显式启用,在独立模型环境运行:
|
|
||||||
|
|
||||||
```bash
|
|
||||||
task_vector_root=$(git rev-parse --absolute-git-dir)
|
|
||||||
uv pip install --python "$task_vector_root/embedding-eval-venv/bin/python" \
|
|
||||||
-r backend/requirements-dev.txt pgserver
|
|
||||||
RUN_LOCAL_EMBEDDING_TEST=1 EMBEDDING_CACHE_DIR="$task_vector_root/embedding-model-cache" \
|
|
||||||
"$task_vector_root/embedding-eval-venv/bin/python" -m pytest -q \
|
|
||||||
backend/tests/test_vector_runtime.py
|
|
||||||
```
|
|
||||||
|
|
||||||
该用例用合成测试文字,验证真实 small 的 512 维编码、HTTP 构建/发布/查询、pgvector 持久化及新进程重启后的空间、发布、命中和历史一致;不是领域分数或生产性能实验。Docker 容器、前端浏览器、RRF 和领域验收保持待办。
|
|
||||||
|
|
||||||
实现依据:[pgvector 精确查询与余弦算子](https://github.com/pgvector/pgvector#querying)、[混合维度存储约束](https://github.com/pgvector/pgvector#can-i-store-vectors-with-different-dimensions-in-the-same-column)、[SQLAlchemy 集成](https://github.com/pgvector/pgvector-python#sqlalchemy)。
|
|
||||||
Generated
+1
-23
@@ -9,8 +9,7 @@
|
|||||||
"version": "0.1.0",
|
"version": "0.1.0",
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"lucide-vue-next": "^0.468.0",
|
"lucide-vue-next": "^0.468.0",
|
||||||
"vue": "^3.5.13",
|
"vue": "^3.5.13"
|
||||||
"vue-router": "^4.6.4"
|
|
||||||
},
|
},
|
||||||
"devDependencies": {
|
"devDependencies": {
|
||||||
"@types/node": "^22.10.2",
|
"@types/node": "^22.10.2",
|
||||||
@@ -1044,12 +1043,6 @@
|
|||||||
"he": "^1.2.0"
|
"he": "^1.2.0"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"node_modules/@vue/devtools-api": {
|
|
||||||
"version": "6.6.4",
|
|
||||||
"resolved": "https://registry.npmmirror.com/@vue/devtools-api/-/devtools-api-6.6.4.tgz",
|
|
||||||
"integrity": "sha512-sGhTPMuXqZ1rVOk32RylztWkfXTRhuS7vgAKv0zjqk8gbsHkJ7xfFf+jbySxt7tWObEJwyKaHMikV/WGDiQm8g==",
|
|
||||||
"license": "MIT"
|
|
||||||
},
|
|
||||||
"node_modules/@vue/language-core": {
|
"node_modules/@vue/language-core": {
|
||||||
"version": "2.2.12",
|
"version": "2.2.12",
|
||||||
"resolved": "https://registry.npmmirror.com/@vue/language-core/-/language-core-2.2.12.tgz",
|
"resolved": "https://registry.npmmirror.com/@vue/language-core/-/language-core-2.2.12.tgz",
|
||||||
@@ -1572,21 +1565,6 @@
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"node_modules/vue-router": {
|
|
||||||
"version": "4.6.4",
|
|
||||||
"resolved": "https://registry.npmmirror.com/vue-router/-/vue-router-4.6.4.tgz",
|
|
||||||
"integrity": "sha512-Hz9q5sa33Yhduglwz6g9skT8OBPii+4bFn88w6J+J4MfEo4KRRpmiNG/hHHkdbRFlLBOqxN8y8gf2Fb0MTUgVg==",
|
|
||||||
"license": "MIT",
|
|
||||||
"dependencies": {
|
|
||||||
"@vue/devtools-api": "^6.6.4"
|
|
||||||
},
|
|
||||||
"funding": {
|
|
||||||
"url": "https://github.com/sponsors/posva"
|
|
||||||
},
|
|
||||||
"peerDependencies": {
|
|
||||||
"vue": "^3.5.0"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"node_modules/vue-tsc": {
|
"node_modules/vue-tsc": {
|
||||||
"version": "2.2.12",
|
"version": "2.2.12",
|
||||||
"resolved": "https://registry.npmmirror.com/vue-tsc/-/vue-tsc-2.2.12.tgz",
|
"resolved": "https://registry.npmmirror.com/vue-tsc/-/vue-tsc-2.2.12.tgz",
|
||||||
|
|||||||
+4
-21
@@ -1,23 +1,6 @@
|
|||||||
{
|
{
|
||||||
"name": "nalati-rag-frontend",
|
"name": "nalati-rag-frontend", "private": true, "version": "0.1.0", "type": "module",
|
||||||
"private": true,
|
"scripts": {"dev": "vite", "build": "vue-tsc -b && vite build", "preview": "vite preview"},
|
||||||
"version": "0.1.0",
|
"dependencies": {"lucide-vue-next": "^0.468.0", "vue": "^3.5.13"},
|
||||||
"type": "module",
|
"devDependencies": {"@types/node": "^22.10.2", "@vitejs/plugin-vue": "^5.2.1", "typescript": "~5.7.2", "vite": "^6.0.5", "vue-tsc": "^2.2.0"}
|
||||||
"scripts": {
|
|
||||||
"dev": "vite",
|
|
||||||
"build": "vue-tsc -b && vite build",
|
|
||||||
"preview": "vite preview"
|
|
||||||
},
|
|
||||||
"dependencies": {
|
|
||||||
"lucide-vue-next": "^0.468.0",
|
|
||||||
"vue": "^3.5.13",
|
|
||||||
"vue-router": "^4.6.4"
|
|
||||||
},
|
|
||||||
"devDependencies": {
|
|
||||||
"@types/node": "^22.10.2",
|
|
||||||
"@vitejs/plugin-vue": "^5.2.1",
|
|
||||||
"typescript": "~5.7.2",
|
|
||||||
"vite": "^6.0.5",
|
|
||||||
"vue-tsc": "^2.2.0"
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
+6
-11
@@ -1,30 +1,25 @@
|
|||||||
<script setup lang="ts">
|
<script setup lang="ts">
|
||||||
import { computed, onMounted, ref, watch } from 'vue'
|
import { computed, onMounted, ref, watch } from 'vue'
|
||||||
import { Activity, BookOpen, Bot, CheckCircle2, ChevronRight, Database, FileText, Globe2, Link2, MessageSquareText, Plus, Search, Send, ShieldCheck, Sparkles, AlertCircle } from 'lucide-vue-next'
|
import { Activity, BookOpen, Bot, CheckCircle2, ChevronRight, Database, FileText, Globe2, Link2, MessageSquareText, Plus, Search, Send, ShieldCheck, Sparkles, AlertCircle } from 'lucide-vue-next'
|
||||||
import { createKnowledge, createSource, getHealth, getKnowledge, getSources, type SourceItem } from './api'
|
import { createKnowledge, createSource, getHealth, getKnowledge, getSources, retrieveKnowledge, retrieveVectorKnowledge, type SourceItem } from './api'
|
||||||
import { retrieveHybrid } from './services/hybrid'
|
|
||||||
import { getCurrentUser } from './services/auth'
|
|
||||||
import { useRouter } from 'vue-router'
|
|
||||||
import type { SessionUser } from './types/auth'
|
|
||||||
type Page='chat'|'knowledge'|'sources'|'evaluation'
|
type Page='chat'|'knowledge'|'sources'|'evaluation'
|
||||||
const router=useRouter(),currentUser=ref<SessionUser>(),page=ref<Page>('chat'), apiStatus=ref('检测中'), question=ref(''), busy=ref(false), search=ref(''), showForm=ref(false), error=ref(''), notice=ref('')
|
const page=ref<Page>('chat'), apiStatus=ref('检测中'), question=ref(''), busy=ref(false), retrievalMode=ref<'keyword'|'vector'>('keyword'), search=ref(''), showForm=ref(false), error=ref(''), notice=ref('')
|
||||||
const messages=ref([{role:'assistant',content:'你好!我是边疆红驿文旅问答原型。当前后端仅为基础接口,尚未接入知识检索和大语言模型。'}] as {role:string;content:string;sources?:{title:string;source_url?:string|null}[]}[])
|
const messages=ref([{role:'assistant',content:'你好!我是边疆红驿文旅问答原型。当前后端仅为基础接口,尚未接入知识检索和大语言模型。'}] as {role:string;content:string;sources?:{title:string;source_url?:string|null}[]}[])
|
||||||
const items=ref<any[]>([]), form=ref({title:'',category:'旅游资源',content:'',source_url:''})
|
const items=ref<any[]>([]), form=ref({title:'',category:'旅游资源',content:'',source_url:''})
|
||||||
const sources=ref<SourceItem[]>([]), sourceSearch=ref(''), showSourceForm=ref(false), sourceForm=ref({name:'',source_url:'',notes:''})
|
const sources=ref<SourceItem[]>([]), sourceSearch=ref(''), showSourceForm=ref(false), sourceForm=ref({name:'',source_url:'',notes:''})
|
||||||
const nav=[{id:'chat',label:'智能问答',icon:MessageSquareText},{id:'knowledge',label:'知识库管理',icon:Database,adminOnly:true},{id:'sources',label:'来源与更新',icon:Globe2,adminOnly:true},{id:'evaluation',label:'实验评估',icon:Activity,adminOnly:true}] as const
|
const nav=[{id:'chat',label:'智能问答',icon:MessageSquareText},{id:'knowledge',label:'知识库管理',icon:Database},{id:'sources',label:'来源与更新',icon:Globe2},{id:'evaluation',label:'实验评估',icon:Activity}] as const
|
||||||
const visibleNav=computed(()=>nav.filter(item=>!('adminOnly' in item)||!item.adminOnly||currentUser.value?.role==='admin'))
|
|
||||||
const filtered=computed(()=>items.value.filter(x=>`${x.title} ${x.category} ${x.content}`.toLowerCase().includes(search.value.toLowerCase())))
|
const filtered=computed(()=>items.value.filter(x=>`${x.title} ${x.category} ${x.content}`.toLowerCase().includes(search.value.toLowerCase())))
|
||||||
async function refresh(){try{items.value=await getKnowledge();error.value=''}catch{error.value='无法读取后端知识库,请确认 API 和 PostgreSQL 已启动。'}}
|
async function refresh(){try{items.value=await getKnowledge();error.value=''}catch{error.value='无法读取后端知识库,请确认 API 和 PostgreSQL 已启动。'}}
|
||||||
async function refreshSources(){try{sources.value=(await getSources(sourceSearch.value)).items;error.value=''}catch{error.value='无法读取来源列表,请确认 API 和 PostgreSQL 已启动。'}}
|
async function refreshSources(){try{sources.value=(await getSources(sourceSearch.value)).items;error.value=''}catch{error.value='无法读取来源列表,请确认 API 和 PostgreSQL 已启动。'}}
|
||||||
async function saveSource(){error.value='';notice.value='';if(!sourceForm.value.name.trim()){error.value='请填写来源名称。';return}if(!validUrl(sourceForm.value.source_url)){error.value='来源网址必须是有效的 http(s) 地址。';return}try{await createSource({...sourceForm.value,name:sourceForm.value.name.trim(),...(sourceForm.value.source_url?{source_url:sourceForm.value.source_url}:{})});sourceForm.value={name:'',source_url:'',notes:''};showSourceForm.value=false;notice.value='来源已登记,可继续创建关联文档和知识条目。';await refreshSources()}catch(e){error.value=e instanceof Error?e.message:'保存失败'}}
|
async function saveSource(){error.value='';notice.value='';if(!sourceForm.value.name.trim()){error.value='请填写来源名称。';return}if(!validUrl(sourceForm.value.source_url)){error.value='来源网址必须是有效的 http(s) 地址。';return}try{await createSource({...sourceForm.value,name:sourceForm.value.name.trim(),...(sourceForm.value.source_url?{source_url:sourceForm.value.source_url}:{})});sourceForm.value={name:'',source_url:'',notes:''};showSourceForm.value=false;notice.value='来源已登记,可继续创建关联文档和知识条目。';await refreshSources()}catch(e){error.value=e instanceof Error?e.message:'保存失败'}}
|
||||||
let sourceSearchTimer:number|undefined
|
let sourceSearchTimer:number|undefined
|
||||||
watch(sourceSearch,()=>{window.clearTimeout(sourceSearchTimer);sourceSearchTimer=window.setTimeout(refreshSources,260)})
|
watch(sourceSearch,()=>{window.clearTimeout(sourceSearchTimer);sourceSearchTimer=window.setTimeout(refreshSources,260)})
|
||||||
onMounted(async()=>{try{currentUser.value=await getCurrentUser()}catch{router.replace('/login');return}try{await getHealth();apiStatus.value='已连接'}catch{apiStatus.value='未连接'}if(currentUser.value.role==='admin')await Promise.all([refresh(),refreshSources()])})
|
onMounted(async()=>{try{await getHealth();apiStatus.value='已连接'}catch{apiStatus.value='未连接'}await Promise.all([refresh(),refreshSources()])})
|
||||||
async function send(){const q=question.value.trim();if(!q||busy.value)return;messages.value.push({role:'user',content:q});question.value='';busy.value=true;try{const r=await retrieveHybrid({query:q,top_k:5,candidate_k:50});const content=r.status==='no_evidence'?'未检索到可用的已核验知识依据。请调整问题或补充审核后的资料。':`已通过 RRF 混合检索命中 ${r.total_hits} 条依据(关键词 ${r.lexical_total_hits},向量 ${r.vector_total_hits},${r.elapsed_ms.toFixed(1)} ms)。当前仅展示检索证据,尚未接入大语言模型生成回答。`;messages.value.push({role:'assistant',content,sources:r.results.map(hit=>({title:`${hit.title}:${hit.snippet}`,source_url:hit.source_url}))})}catch(e){messages.value.push({role:'assistant',content:e instanceof Error?e.message:'暂时无法连接混合检索服务。'})}finally{busy.value=false}}
|
async function send(){const q=question.value.trim();if(!q||busy.value)return;messages.value.push({role:'user',content:q});question.value='';busy.value=true;try{const r=retrievalMode.value==='vector'?await retrieveVectorKnowledge({query:q,top_k:5}):await retrieveKnowledge({query:q,top_k:5});const content=r.status==='no_evidence'?'未检索到可用的已核验知识依据。请调整问题或补充审核后的资料。':`已通过${r.mode==='vector'?'向量':'关键词'}检索命中 ${r.total_hits} 条已核验依据(${r.elapsed_ms.toFixed(1)} ms)。当前仅展示检索证据,尚未接入大语言模型生成回答。`;messages.value.push({role:'assistant',content,sources:r.results.map(hit=>({title:`${hit.title}:${hit.snippet}`,source_url:hit.source_url}))})}catch(e){messages.value.push({role:'assistant',content:e instanceof Error?e.message:'暂时无法连接检索服务。'})}finally{busy.value=false}}
|
||||||
function validUrl(value:string){try{if(value)new URL(value);return true}catch{return false}}
|
function validUrl(value:string){try{if(value)new URL(value);return true}catch{return false}}
|
||||||
async function save(){error.value='';notice.value='';if(!form.value.title.trim()||!form.value.content.trim()){error.value='请填写知识标题和正文。';return}if(!validUrl(form.value.source_url)){error.value='来源网址必须是有效的 http(s) 地址。';return}try{await createKnowledge({...form.value,title:form.value.title.trim(),content:form.value.content.trim(),...(form.value.source_url?{source_url:form.value.source_url}:{})});form.value={title:'',category:'旅游资源',content:'',source_url:''};showForm.value=false;notice.value='已保存为待审核条目,尚未进入检索索引。';await refresh()}catch(e){error.value=e instanceof Error?e.message:'保存失败'}}
|
async function save(){error.value='';notice.value='';if(!form.value.title.trim()||!form.value.content.trim()){error.value='请填写知识标题和正文。';return}if(!validUrl(form.value.source_url)){error.value='来源网址必须是有效的 http(s) 地址。';return}try{await createKnowledge({...form.value,title:form.value.title.trim(),content:form.value.content.trim(),...(form.value.source_url?{source_url:form.value.source_url}:{})});form.value={title:'',category:'旅游资源',content:'',source_url:''};showForm.value=false;notice.value='已保存为待审核条目,尚未进入检索索引。';await refresh()}catch(e){error.value=e instanceof Error?e.message:'保存失败'}}
|
||||||
</script>
|
</script>
|
||||||
<template><div class="shell"><aside class="sidebar"><div class="brand"><div class="mark"><Bot :size="23"/></div><div><b>边疆红驿</b><small>NALATI KNOWLEDGE HUB</small></div></div><p class="muted label">功能列表</p><nav><button v-for="n in visibleNav" :key="n.id" :class="{active:page===n.id}" @click="page=n.id"><component :is="n.icon" :size="18"/>{{n.label}}<ChevronRight v-if="page===n.id" class="chev" :size="15"/></button><button @click="router.push('/profile')"><ShieldCheck :size="18"/>个人中心</button><button v-if="currentUser?.role==='admin'" @click="router.push('/admin')"><Activity :size="18"/>用户管理</button></nav><div class="quick-actions"><p class="muted label">快捷操作</p><button v-if="page==='knowledge'" class="side-action" @click="showForm=true"><Plus :size="16"/>新增知识</button><button v-if="page==='sources'" class="side-action" @click="showSourceForm=true"><Plus :size="16"/>新增来源</button><p v-if="page==='chat'" class="side-hint">选择示例问题,或直接开始提问。</p><p v-if="page==='evaluation'" class="side-hint">实验结果只展示真实运行数据。</p></div><div class="side-bottom"><div class="tip"><ShieldCheck :size="17"/><div><b>{{currentUser?.role==='admin'?'管理员':'已登录用户'}}</b><p>{{currentUser?.nickname||currentUser?.username||'正在载入账户'}}</p></div></div><p class="api">● API {{apiStatus}} <span>v0.1.0</span></p></div></aside><main><header><span>边疆红驿 / {{nav.find(n=>n.id===page)?.label}}</span><div><span class="pill">● {{currentUser?.role==='admin'?'管理员':'用户'}}</span><i>研</i></div></header><section class="content">
|
<template><div class="shell"><aside class="sidebar"><div class="brand"><div class="mark"><Bot :size="23"/></div><div><b>边疆红驿</b><small>NALATI KNOWLEDGE HUB</small></div></div><p class="muted label">功能列表</p><nav><button v-for="n in nav" :key="n.id" :class="{active:page===n.id}" @click="page=n.id"><component :is="n.icon" :size="18"/>{{n.label}}<ChevronRight v-if="page===n.id" class="chev" :size="15"/></button></nav><div class="quick-actions"><p class="muted label">快捷操作</p><button v-if="page==='knowledge'" class="side-action" @click="showForm=true"><Plus :size="16"/>新增知识</button><button v-if="page==='sources'" class="side-action" @click="showSourceForm=true"><Plus :size="16"/>新增来源</button><p v-if="page==='chat'" class="side-hint">选择示例问题,或直接开始提问。</p><p v-if="page==='evaluation'" class="side-hint">实验结果只展示真实运行数据。</p></div><div class="side-bottom"><div class="tip"><ShieldCheck :size="17"/><div><b>可信知识服务</b><p>优先展示可核验来源,信息不足时明确提示。</p></div></div><p class="api">● API {{apiStatus}} <span>v0.1.0</span></p></div></aside><main><header><span>边疆红驿 / {{nav.find(n=>n.id===page)?.label}}</span><div><span class="pill">● 原型阶段</span><i>研</i></div></header><section class="content">
|
||||||
<div v-if="page==='chat'"><div class="heading"><div><small class="eyebrow"><Sparkles :size="14"/> LOCAL KNOWLEDGE ASSISTANT</small><h1>文旅知识,问得明白</h1><p>面向那拉提文旅与民宿服务的知识问答原型</p></div><span class="chip"><ShieldCheck :size="15"/> 来源可追溯 · 设计目标</span></div><div class="chat-layout"><div class="card chat"><div class="chat-head"><div class="mark"><Bot :size="20"/></div><div><b>边疆红驿助手</b><small>问答服务原型</small></div></div><div class="messages"><div v-for="(m,i) in messages" :key="i" class="message" :class="m.role"><small>{{m.role==='assistant'?'智能助手':'你'}}</small><p>{{m.content}}</p><div v-if="m.sources?.length" class="sources"><b><Link2 :size="14"/> 参考来源</b><a v-for="s in m.sources" :key="s.title" :href="s.source_url ?? undefined" target="_blank">{{s.title}}</a></div></div><p v-if="busy">正在请求后端…</p></div><form class="composer" @submit.prevent="send"><textarea v-model="question" placeholder="输入你想了解的文旅问题…" rows="2" @keydown.enter.exact.prevent="send"/><div><small>当前为基础接口,尚未接入真实检索</small><button class="primary" :disabled="busy||!question.trim()"><Send :size="15"/> 发送</button></div></form></div><aside class="suggest"><b><Sparkles :size="16"/> 你可以这样问</b><button @click="question='那拉提有哪些值得了解的旅游资源?'"><Globe2/>旅游资源 <ChevronRight/></button><button @click="question='有哪些经过核验的红色文化资料?'"><BookOpen/>红色文化 <ChevronRight/></button><button @click="question='民宿服务信息应如何核验?'"><ShieldCheck/>民宿服务 <ChevronRight/></button><div class="notice"><b><AlertCircle :size="15"/> 信息说明</b><p>本项目仍处于原型阶段。演示内容不代表已核验的地方文旅事实。</p></div></aside></div></div>
|
<div v-if="page==='chat'"><div class="heading"><div><small class="eyebrow"><Sparkles :size="14"/> LOCAL KNOWLEDGE ASSISTANT</small><h1>文旅知识,问得明白</h1><p>面向那拉提文旅与民宿服务的知识问答原型</p></div><span class="chip"><ShieldCheck :size="15"/> 来源可追溯 · 设计目标</span></div><div class="chat-layout"><div class="card chat"><div class="chat-head"><div class="mark"><Bot :size="20"/></div><div><b>边疆红驿助手</b><small>问答服务原型</small></div></div><div class="messages"><div v-for="(m,i) in messages" :key="i" class="message" :class="m.role"><small>{{m.role==='assistant'?'智能助手':'你'}}</small><p>{{m.content}}</p><div v-if="m.sources?.length" class="sources"><b><Link2 :size="14"/> 参考来源</b><a v-for="s in m.sources" :key="s.title" :href="s.source_url ?? undefined" target="_blank">{{s.title}}</a></div></div><p v-if="busy">正在请求后端…</p></div><form class="composer" @submit.prevent="send"><textarea v-model="question" placeholder="输入你想了解的文旅问题…" rows="2" @keydown.enter.exact.prevent="send"/><div><small>当前为基础接口,尚未接入真实检索</small><button class="primary" :disabled="busy||!question.trim()"><Send :size="15"/> 发送</button></div></form></div><aside class="suggest"><b><Sparkles :size="16"/> 你可以这样问</b><button @click="question='那拉提有哪些值得了解的旅游资源?'"><Globe2/>旅游资源 <ChevronRight/></button><button @click="question='有哪些经过核验的红色文化资料?'"><BookOpen/>红色文化 <ChevronRight/></button><button @click="question='民宿服务信息应如何核验?'"><ShieldCheck/>民宿服务 <ChevronRight/></button><div class="notice"><b><AlertCircle :size="15"/> 信息说明</b><p>本项目仍处于原型阶段。演示内容不代表已核验的地方文旅事实。</p></div></aside></div></div>
|
||||||
<div v-else-if="page==='knowledge'"><div class="heading"><div><small class="eyebrow"><Database :size="14"/> KNOWLEDGE MANAGEMENT</small><h1>知识库管理</h1><p>集中管理领域知识、来源和审核状态。</p></div><button class="primary" @click="showForm=!showForm"><Plus :size="16"/> 新增知识</button></div><div v-if="showForm" class="card form-card"><b>新增知识条目</b><form class="knowledge-form" @submit.prevent="save"><label>标题<input v-model="form.title" required maxlength="250"/></label><label>分类<select v-model="form.category"><option>红色文化</option><option>旅游资源</option><option>民宿服务</option><option>交通出行</option><option>其他</option></select></label><label class="wide">正文<textarea v-model="form.content" required rows="3"/></label><label class="wide">来源网址(可选)<input v-model="form.source_url" type="url" placeholder="https://..."/></label><div class="wide actions"><button type="button" class="secondary" @click="showForm=false">取消</button><button class="primary">保存条目</button></div></form></div><div class="stats"><div class="card stat"><small>知识条目</small><strong>{{items.length}}</strong><span>来自当前数据库</span></div><div class="card stat"><small>已核验</small><strong>{{items.filter(x=>x.verification_status==='verified').length}}</strong><span>需经过真实审核</span></div><div class="card stat"><small>待审核</small><strong>{{items.filter(x=>x.verification_status==='pending').length}}</strong><span>尚未进入检索索引</span></div></div><p v-if="notice" class="success">{{notice}}</p><p v-if="error" class="error">{{error}}</p><div class="card table-card"><div class="table-head"><div><b>知识条目</b><p>列表来自后端接口,不含预设真实资料。</p></div><label class="search"><Search :size="15"/><input v-model="search" placeholder="搜索知识"/></label></div><div class="scroll"><table><thead><tr><th>标题</th><th>分类</th><th>来源</th><th>状态</th><th>创建时间</th></tr></thead><tbody><tr v-for="x in filtered" :key="x.id"><td><b>{{x.title}}</b><small>{{x.content.slice(0,45)}}</small></td><td>{{x.category}}</td><td><a v-if="x.source_url" :href="x.source_url" target="_blank">打开来源</a><span v-else>未填写</span></td><td><em>{{x.verification_status==='pending'?'待审核':x.verification_status}}</em></td><td>{{new Date(x.created_at).toLocaleDateString()}}</td></tr><tr v-if="!filtered.length"><td colspan="5" class="empty">暂无条目,点击“新增知识”添加。</td></tr></tbody></table></div></div></div>
|
<div v-else-if="page==='knowledge'"><div class="heading"><div><small class="eyebrow"><Database :size="14"/> KNOWLEDGE MANAGEMENT</small><h1>知识库管理</h1><p>集中管理领域知识、来源和审核状态。</p></div><button class="primary" @click="showForm=!showForm"><Plus :size="16"/> 新增知识</button></div><div v-if="showForm" class="card form-card"><b>新增知识条目</b><form class="knowledge-form" @submit.prevent="save"><label>标题<input v-model="form.title" required maxlength="250"/></label><label>分类<select v-model="form.category"><option>红色文化</option><option>旅游资源</option><option>民宿服务</option><option>交通出行</option><option>其他</option></select></label><label class="wide">正文<textarea v-model="form.content" required rows="3"/></label><label class="wide">来源网址(可选)<input v-model="form.source_url" type="url" placeholder="https://..."/></label><div class="wide actions"><button type="button" class="secondary" @click="showForm=false">取消</button><button class="primary">保存条目</button></div></form></div><div class="stats"><div class="card stat"><small>知识条目</small><strong>{{items.length}}</strong><span>来自当前数据库</span></div><div class="card stat"><small>已核验</small><strong>{{items.filter(x=>x.verification_status==='verified').length}}</strong><span>需经过真实审核</span></div><div class="card stat"><small>待审核</small><strong>{{items.filter(x=>x.verification_status==='pending').length}}</strong><span>尚未进入检索索引</span></div></div><p v-if="notice" class="success">{{notice}}</p><p v-if="error" class="error">{{error}}</p><div class="card table-card"><div class="table-head"><div><b>知识条目</b><p>列表来自后端接口,不含预设真实资料。</p></div><label class="search"><Search :size="15"/><input v-model="search" placeholder="搜索知识"/></label></div><div class="scroll"><table><thead><tr><th>标题</th><th>分类</th><th>来源</th><th>状态</th><th>创建时间</th></tr></thead><tbody><tr v-for="x in filtered" :key="x.id"><td><b>{{x.title}}</b><small>{{x.content.slice(0,45)}}</small></td><td>{{x.category}}</td><td><a v-if="x.source_url" :href="x.source_url" target="_blank">打开来源</a><span v-else>未填写</span></td><td><em>{{x.verification_status==='pending'?'待审核':x.verification_status}}</em></td><td>{{new Date(x.created_at).toLocaleDateString()}}</td></tr><tr v-if="!filtered.length"><td colspan="5" class="empty">暂无条目,点击“新增知识”添加。</td></tr></tbody></table></div></div></div>
|
||||||
<div v-else-if="page==='sources'"><div class="heading"><div><small class="eyebrow"><Globe2 :size="14"/> SOURCE OPERATIONS</small><h1>来源与更新</h1><p>登记可追溯来源,为文档采集和知识审核建立基础。</p></div><button class="primary" @click="showSourceForm=!showSourceForm"><Plus :size="16"/> 新增来源</button></div><div v-if="showSourceForm" class="card form-card"><b>登记来源</b><form class="knowledge-form" @submit.prevent="saveSource"><label>来源名称<input v-model="sourceForm.name" required maxlength="250" placeholder="例如:那拉提景区官网"/></label><label>来源网址(可选)<input v-model="sourceForm.source_url" type="url" placeholder="https://..."/></label><label class="wide">备注<textarea v-model="sourceForm.notes" rows="3" placeholder="记录来源用途、可靠性判断或待核验事项"/></label><div class="wide actions"><button type="button" class="secondary" @click="showSourceForm=false">取消</button><button class="primary">保存来源</button></div></form></div><p v-if="notice" class="success">{{notice}}</p><p v-if="error" class="error">{{error}}</p><div class="stats"><div class="card stat"><small>来源总数</small><strong>{{sources.length}}</strong><span>来自当前数据库</span></div><div class="card stat"><small>关联文档</small><strong>—</strong><span>文档工作台待接入</span></div><div class="card stat"><small>自动更新</small><strong class="pending">未启用</strong><span>当前不会自动抓取网页</span></div></div><div class="card table-card"><div class="table-head"><div><b>已登记来源</b><p>来源不是审核通过的证明,关键事实仍需交叉核验。</p></div><label class="search"><Search :size="15"/><input v-model="sourceSearch" placeholder="搜索名称或备注" @keyup.enter="refreshSources"/></label></div><div class="scroll"><table><thead><tr><th>名称</th><th>网址</th><th>备注</th><th>登记时间</th></tr></thead><tbody><tr v-for="source in sources" :key="source.id"><td><b>{{source.name}}</b></td><td><a v-if="source.source_url" :href="source.source_url" target="_blank" rel="noreferrer">打开来源</a><span v-else>未填写</span></td><td>{{source.notes||'—'}}</td><td>{{new Date(source.created_at).toLocaleDateString()}}</td></tr><tr v-if="!sources.length"><td colspan="4" class="empty">暂无来源,点击“新增来源”登记。</td></tr></tbody></table></div></div><div class="card roadmap"><b>来源处理边界</b><div><span>01 登记来源</span><ChevronRight/><span>02 人工核验</span><ChevronRight/><span>03 关联文档</span><ChevronRight/><span>04 后续检索</span></div></div></div>
|
<div v-else-if="page==='sources'"><div class="heading"><div><small class="eyebrow"><Globe2 :size="14"/> SOURCE OPERATIONS</small><h1>来源与更新</h1><p>登记可追溯来源,为文档采集和知识审核建立基础。</p></div><button class="primary" @click="showSourceForm=!showSourceForm"><Plus :size="16"/> 新增来源</button></div><div v-if="showSourceForm" class="card form-card"><b>登记来源</b><form class="knowledge-form" @submit.prevent="saveSource"><label>来源名称<input v-model="sourceForm.name" required maxlength="250" placeholder="例如:那拉提景区官网"/></label><label>来源网址(可选)<input v-model="sourceForm.source_url" type="url" placeholder="https://..."/></label><label class="wide">备注<textarea v-model="sourceForm.notes" rows="3" placeholder="记录来源用途、可靠性判断或待核验事项"/></label><div class="wide actions"><button type="button" class="secondary" @click="showSourceForm=false">取消</button><button class="primary">保存来源</button></div></form></div><p v-if="notice" class="success">{{notice}}</p><p v-if="error" class="error">{{error}}</p><div class="stats"><div class="card stat"><small>来源总数</small><strong>{{sources.length}}</strong><span>来自当前数据库</span></div><div class="card stat"><small>关联文档</small><strong>—</strong><span>文档工作台待接入</span></div><div class="card stat"><small>自动更新</small><strong class="pending">未启用</strong><span>当前不会自动抓取网页</span></div></div><div class="card table-card"><div class="table-head"><div><b>已登记来源</b><p>来源不是审核通过的证明,关键事实仍需交叉核验。</p></div><label class="search"><Search :size="15"/><input v-model="sourceSearch" placeholder="搜索名称或备注" @keyup.enter="refreshSources"/></label></div><div class="scroll"><table><thead><tr><th>名称</th><th>网址</th><th>备注</th><th>登记时间</th></tr></thead><tbody><tr v-for="source in sources" :key="source.id"><td><b>{{source.name}}</b></td><td><a v-if="source.source_url" :href="source.source_url" target="_blank" rel="noreferrer">打开来源</a><span v-else>未填写</span></td><td>{{source.notes||'—'}}</td><td>{{new Date(source.created_at).toLocaleDateString()}}</td></tr><tr v-if="!sources.length"><td colspan="4" class="empty">暂无来源,点击“新增来源”登记。</td></tr></tbody></table></div></div><div class="card roadmap"><b>来源处理边界</b><div><span>01 登记来源</span><ChevronRight/><span>02 人工核验</span><ChevronRight/><span>03 关联文档</span><ChevronRight/><span>04 后续检索</span></div></div></div>
|
||||||
|
|||||||
@@ -1,2 +0,0 @@
|
|||||||
<script setup lang="ts">import { RouterView } from 'vue-router';</script>
|
|
||||||
<template><RouterView/></template>
|
|
||||||
+5
-5
@@ -1,6 +1,4 @@
|
|||||||
import { apiRequest } from './services/session'
|
import { apiRequest } from './services/session'
|
||||||
import type { VectorRetrievalResult } from './services/vectors'
|
|
||||||
import type { ChunkKeywordResult, HybridResult } from './services/hybrid'
|
|
||||||
|
|
||||||
export const getHealth = () => apiRequest<{status:string}>('/api/health')
|
export const getHealth = () => apiRequest<{status:string}>('/api/health')
|
||||||
export const askQuestion = (question:string) => apiRequest<{answer:string;status:string;sources:{title:string;source_url?:string|null}[]}>('/api/chat',{method:'POST',body:JSON.stringify({question})})
|
export const askQuestion = (question:string) => apiRequest<{answer:string;status:string;sources:{title:string;source_url?:string|null}[]}>('/api/chat',{method:'POST',body:JSON.stringify({question})})
|
||||||
@@ -16,6 +14,8 @@ export function reviewKnowledge(id:number,payload:{expected_version:number;verif
|
|||||||
export type RetrievalHit={rank:number;knowledge_id:number;knowledge_version:number;document_id:number|null;document_version:number|null;source_id:number|null;source_version:number|null;source_name:string|null;source_url:string;title:string;category:string;snippet:string;snippet_start:number;snippet_end:number;score:number;matched_terms:string[]}
|
export type RetrievalHit={rank:number;knowledge_id:number;knowledge_version:number;document_id:number|null;document_version:number|null;source_id:number|null;source_version:number|null;source_name:string|null;source_url:string;title:string;category:string;snippet:string;snippet_start:number;snippet_end:number;score:number;matched_terms:string[]}
|
||||||
export type RetrievalResult={run_id:string;query:string;mode:'keyword';status:'ok'|'no_evidence';top_k:number;category:string|null;query_terms:string[];corpus_size:number;corpus_fingerprint:string;total_hits:number;results:RetrievalHit[];elapsed_ms:number;created_at:string;config:{tokenizer:string;scorer:string;k1:number;b:number;title_weight:number;stopwords_version:string}}
|
export type RetrievalResult={run_id:string;query:string;mode:'keyword';status:'ok'|'no_evidence';top_k:number;category:string|null;query_terms:string[];corpus_size:number;corpus_fingerprint:string;total_hits:number;results:RetrievalHit[];elapsed_ms:number;created_at:string;config:{tokenizer:string;scorer:string;k1:number;b:number;title_weight:number;stopwords_version:string}}
|
||||||
export function retrieveKnowledge(payload:{query:string;mode?:'keyword';top_k?:number;category?:string}){return apiRequest<RetrievalResult>('/api/retrieval',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify(payload)})}
|
export function retrieveKnowledge(payload:{query:string;mode?:'keyword';top_k?:number;category?:string}){return apiRequest<RetrievalResult>('/api/retrieval',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify(payload)})}
|
||||||
export type RetrievalRunResult = RetrievalResult | VectorRetrievalResult | ChunkKeywordResult | HybridResult
|
export function getRetrievalRun(runId:string){return apiRequest<RetrievalResult>(`/api/retrieval/runs/${encodeURIComponent(runId)}`)}
|
||||||
export function getRetrievalRun(runId:string){return apiRequest<RetrievalRunResult>(`/api/retrieval/runs/${encodeURIComponent(runId)}`)}
|
export function getRetrievalRuns(page=1,pageSize=20){return apiRequest<{items:RetrievalResult[];total:number;page:number;page_size:number}>(`/api/retrieval/runs?page=${page}&page_size=${pageSize}`)}
|
||||||
export function getRetrievalRuns(page=1,pageSize=20){return apiRequest<{items:RetrievalRunResult[];total:number;page:number;page_size:number}>(`/api/retrieval/runs?page=${page}&page_size=${pageSize}`)}
|
|
||||||
|
export type VectorRetrievalResult=Omit<RetrievalResult,'mode'|'results'|'config'> & {mode:'vector';vector_version_id:string;text_index_version_id:string;generation:number;text_generation:number;results:(RetrievalHit & {chunk_id:string;chunk_ordinal:number})[];config:{scorer:'pgvector-cosine-exact-v1';ranking_unit:'chunk';space:{id:string;dimension:number}}}
|
||||||
|
export function retrieveVectorKnowledge(payload:{query:string;top_k?:number;category?:string}){return apiRequest<VectorRetrievalResult>('/api/retrieval/vector',{method:'POST',headers:{'Content-Type':'application/json'},body:JSON.stringify({mode:'vector',top_k:5,...payload})})}
|
||||||
|
|||||||
@@ -1,6 +1,5 @@
|
|||||||
import { createApp } from 'vue'
|
import { createApp } from 'vue'
|
||||||
import './style.css'
|
import './style.css'
|
||||||
import './enhancements.css'
|
import './enhancements.css'
|
||||||
import Root from './Root.vue'
|
import App from './App.vue'
|
||||||
import router from './router'
|
createApp(App).mount('#app')
|
||||||
createApp(Root).use(router).mount('#app')
|
|
||||||
|
|||||||
@@ -1,16 +0,0 @@
|
|||||||
import { createRouter, createWebHistory } from 'vue-router'
|
|
||||||
import App from './App.vue'
|
|
||||||
import { getCurrentUser } from './services/auth'
|
|
||||||
import type { UserRole } from './types/auth'
|
|
||||||
|
|
||||||
let role:UserRole|undefined
|
|
||||||
const router=createRouter({history:createWebHistory(),routes:[
|
|
||||||
{path:'/',component:App},
|
|
||||||
{path:'/login',component:()=>import('./views/LoginView.vue'),meta:{public:true}},
|
|
||||||
{path:'/register',component:()=>import('./views/RegisterView.vue'),meta:{public:true}},
|
|
||||||
{path:'/profile',component:()=>import('./views/ProfileView.vue')},
|
|
||||||
{path:'/settings',component:()=>import('./views/AccountSettingsView.vue')},
|
|
||||||
{path:'/admin',component:()=>import('./views/AdminView.vue'),meta:{role:'admin'}},
|
|
||||||
]})
|
|
||||||
router.beforeEach(async to=>{if(to.meta.public)return true;try{role=(await getCurrentUser()).role}catch{return {path:'/login',query:{redirect:to.fullPath}}}if(to.meta.role&&role!==to.meta.role)return {path:'/profile',query:{forbidden:'1'}};return true})
|
|
||||||
export default router
|
|
||||||
@@ -1,19 +0,0 @@
|
|||||||
import { apiRequest } from './session'
|
|
||||||
import type { RetrievalHit, RetrievalResult } from '../api'
|
|
||||||
import type { IndexVersion } from './indexes'
|
|
||||||
import type { EmbeddingSpace } from './vectors'
|
|
||||||
|
|
||||||
type ChunkHit = RetrievalHit & { chunk_id: string; chunk_ordinal: number }
|
|
||||||
type ChunkConfig = { ranking_unit: 'chunk'; lexical: RetrievalResult['config']; chunking: IndexVersion['config']; tie_break: 'knowledge_id,chunk_ordinal,chunk_id' }
|
|
||||||
type ChunkSnapshot = { run_id: string; query: string; status: 'ok' | 'no_evidence'; top_k: number; category: string | null; query_terms: string[]; text_index_version_id: string; text_generation: number; catalog_revision: number; corpus_fingerprint: string; corpus_size: number; total_hits: number; elapsed_ms: number; created_at: string }
|
|
||||||
export type ChunkKeywordResult = ChunkSnapshot & { mode: 'keyword_chunks'; results: ChunkHit[]; config: ChunkConfig }
|
|
||||||
export type RankedCandidate = { chunk_id: string; rank: number; score: number }
|
|
||||||
export type HybridResult = ChunkSnapshot & {
|
|
||||||
mode: 'hybrid'; vector_version_id: string; generation: number; lexical_total_hits: number; vector_total_hits: number;
|
|
||||||
lexical_candidates: RankedCandidate[]; vector_candidates: RankedCandidate[];
|
|
||||||
results: (ChunkHit & { lexical_rank: number | null; lexical_score: number | null; vector_rank: number | null; vector_score: number | null; lexical_contribution: number; vector_contribution: number })[];
|
|
||||||
config: ChunkConfig & { scorer: 'rrf-v1'; rrf_k: 60; weights: [1, 1]; candidate_k: number; vector_scorer: 'pgvector-cosine-exact-v1'; space: EmbeddingSpace }
|
|
||||||
}
|
|
||||||
type Request = { query: string; top_k?: number; category?: string }
|
|
||||||
export const retrieveKeywordChunks = (payload: Request & { mode?: 'keyword_chunks' }) => apiRequest<ChunkKeywordResult>('/api/retrieval/chunks', { method: 'POST', body: JSON.stringify(payload) })
|
|
||||||
export const retrieveHybrid = (payload: Request & { mode?: 'hybrid'; candidate_k?: number }) => apiRequest<HybridResult>('/api/retrieval/hybrid', { method: 'POST', body: JSON.stringify(payload) })
|
|
||||||
@@ -1,18 +0,0 @@
|
|||||||
import { apiRequest } from './session'
|
|
||||||
import type { RetrievalHit } from '../api'
|
|
||||||
import type { IndexPublication } from './indexes'
|
|
||||||
|
|
||||||
type Page<T> = { items: T[]; total: number; page: number; page_size: number }
|
|
||||||
export type EmbeddingSpec = { model: string; revision: string; dimension: number; token_limit: number; query_instruction: string; implementation: string; checkpoint_format: string }
|
|
||||||
export type EmbeddingSpace = { id: string; dimension: number; spec: EmbeddingSpec }
|
|
||||||
export type VectorVersion = { id: string; text_index_version_id: string; space_id: string; dimension: number; chunk_count: number; status: 'ready' | 'published' | 'superseded'; created_by: number; created_at: string; published_at: string | null }
|
|
||||||
export type ActiveVectors = { generation: number; text_generation: number; catalog_revision: number; is_current: boolean; active_version: VectorVersion | null }
|
|
||||||
export type VectorRetrievalResult = { run_id: string; query: string; mode: 'vector'; status: 'ok' | 'no_evidence'; top_k: number; category: string | null; vector_version_id: string; text_index_version_id: string; generation: number; text_generation: number; corpus_fingerprint: string; corpus_size: number; total_hits: number; results: (RetrievalHit & { chunk_id: string; chunk_ordinal: number })[]; elapsed_ms: number; created_at: string; config: { scorer: 'pgvector-cosine-exact-v1'; ranking_unit: 'chunk'; space: EmbeddingSpace } }
|
|
||||||
export const buildVectorIndex = (textIndexVersionId: string) => apiRequest<VectorVersion>('/api/vector-indexes', { method: 'POST', body: JSON.stringify({ text_index_version_id: textIndexVersionId }) }, true)
|
|
||||||
export const getVectorVersions = (page = 1, pageSize = 20) => apiRequest<Page<VectorVersion>>(`/api/vector-indexes?page=${page}&page_size=${pageSize}`)
|
|
||||||
export const getVectorVersion = (id: string) => apiRequest<VectorVersion>(`/api/vector-indexes/${encodeURIComponent(id)}`)
|
|
||||||
export const getActiveVectors = () => apiRequest<ActiveVectors>('/api/vector-indexes/active')
|
|
||||||
export const getEmbeddingSpaces = (page = 1, pageSize = 20) => apiRequest<Page<EmbeddingSpace>>(`/api/vector-indexes/spaces?page=${page}&page_size=${pageSize}`)
|
|
||||||
export const publishVectorIndex = (id: string, expectedGeneration: number) => apiRequest<ActiveVectors>(`/api/vector-indexes/${encodeURIComponent(id)}/publish`, { method: 'POST', body: JSON.stringify({ expected_generation: expectedGeneration }) }, true)
|
|
||||||
export const getVectorPublications = (page = 1, pageSize = 20) => apiRequest<Page<IndexPublication>>(`/api/vector-indexes/publications?page=${page}&page_size=${pageSize}`)
|
|
||||||
export const retrieveVectors = (payload: { query: string; mode?: 'vector'; top_k?: number; category?: string }) => apiRequest<VectorRetrievalResult>('/api/retrieval/vector', { method: 'POST', body: JSON.stringify(payload) })
|
|
||||||
@@ -1,96 +0,0 @@
|
|||||||
<script setup lang="ts">
|
|
||||||
import { onMounted, ref } from 'vue'
|
|
||||||
import { useRouter } from 'vue-router'
|
|
||||||
import { changePassword, getCurrentUser, updateMyProfile } from '../services/auth'
|
|
||||||
import type { SessionUser } from '../types/auth'
|
|
||||||
|
|
||||||
const router = useRouter()
|
|
||||||
const user = ref<SessionUser>()
|
|
||||||
const profile = ref({ nickname: '', email: '' })
|
|
||||||
const password = ref({ current_password: '', new_password: '', confirm_password: '' })
|
|
||||||
const loading = ref(true)
|
|
||||||
const savingProfile = ref(false)
|
|
||||||
const savingPassword = ref(false)
|
|
||||||
const error = ref('')
|
|
||||||
const success = ref('')
|
|
||||||
|
|
||||||
onMounted(async () => {
|
|
||||||
try {
|
|
||||||
user.value = await getCurrentUser()
|
|
||||||
profile.value = { nickname: user.value.nickname ?? '', email: user.value.email ?? '' }
|
|
||||||
} catch (cause) {
|
|
||||||
error.value = cause instanceof Error ? cause.message : '无法读取账户资料'
|
|
||||||
} finally {
|
|
||||||
loading.value = false
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
async function saveProfile() {
|
|
||||||
error.value = ''
|
|
||||||
success.value = ''
|
|
||||||
if (profile.value.email && !/^\S+@\S+\.\S+$/.test(profile.value.email)) {
|
|
||||||
error.value = '请输入有效邮箱地址。'
|
|
||||||
return
|
|
||||||
}
|
|
||||||
savingProfile.value = true
|
|
||||||
try {
|
|
||||||
const updated = await updateMyProfile({ nickname: profile.value.nickname || null, email: profile.value.email || null })
|
|
||||||
if (user.value) user.value = { ...user.value, ...updated }
|
|
||||||
success.value = '基本资料已保存。'
|
|
||||||
} catch (cause) {
|
|
||||||
error.value = cause instanceof Error ? cause.message : '保存资料失败'
|
|
||||||
} finally {
|
|
||||||
savingProfile.value = false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async function savePassword() {
|
|
||||||
error.value = ''
|
|
||||||
success.value = ''
|
|
||||||
if (password.value.new_password.length < 15) {
|
|
||||||
error.value = '新密码至少需要 15 个字符。'
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if (password.value.new_password !== password.value.confirm_password) {
|
|
||||||
error.value = '两次输入的新密码不一致。'
|
|
||||||
return
|
|
||||||
}
|
|
||||||
savingPassword.value = true
|
|
||||||
try {
|
|
||||||
await changePassword({ current_password: password.value.current_password, new_password: password.value.new_password })
|
|
||||||
password.value = { current_password: '', new_password: '', confirm_password: '' }
|
|
||||||
router.replace('/login?changed=1')
|
|
||||||
} catch (cause) {
|
|
||||||
error.value = cause instanceof Error ? cause.message : '修改密码失败'
|
|
||||||
} finally {
|
|
||||||
savingPassword.value = false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
</script>
|
|
||||||
|
|
||||||
<template>
|
|
||||||
<main>
|
|
||||||
<RouterLink to="/profile">返回个人中心</RouterLink>
|
|
||||||
<h1>账户设置</h1>
|
|
||||||
<p v-if="loading">正在读取账户资料…</p>
|
|
||||||
<template v-else>
|
|
||||||
<p v-if="error" role="alert">{{ error }}</p>
|
|
||||||
<p v-if="success" role="status">{{ success }}</p>
|
|
||||||
<form @submit.prevent="saveProfile">
|
|
||||||
<h2>基本资料</h2>
|
|
||||||
<label>用户名<input :value="user?.username" disabled /></label>
|
|
||||||
<label>昵称<input v-model.trim="profile.nickname" maxlength="80" autocomplete="nickname" /></label>
|
|
||||||
<label>邮箱<input v-model.trim="profile.email" type="email" maxlength="320" autocomplete="email" /></label>
|
|
||||||
<button :disabled="savingProfile">{{ savingProfile ? '保存中…' : '保存资料' }}</button>
|
|
||||||
</form>
|
|
||||||
<form @submit.prevent="savePassword">
|
|
||||||
<h2>修改密码</h2>
|
|
||||||
<label>当前密码<input v-model="password.current_password" type="password" required autocomplete="current-password" /></label>
|
|
||||||
<label>新密码<input v-model="password.new_password" type="password" required minlength="15" autocomplete="new-password" /></label>
|
|
||||||
<label>确认新密码<input v-model="password.confirm_password" type="password" required minlength="15" autocomplete="new-password" /></label>
|
|
||||||
<p>修改密码后会退出当前会话,请使用新密码重新登录。</p>
|
|
||||||
<button :disabled="savingPassword">{{ savingPassword ? '提交中…' : '修改密码' }}</button>
|
|
||||||
</form>
|
|
||||||
</template>
|
|
||||||
</main>
|
|
||||||
</template>
|
|
||||||
@@ -1 +0,0 @@
|
|||||||
<script setup lang="ts">import { onMounted,ref } from 'vue';import { getUsers } from '../services/auth';const users=ref<any[]>([]),error=ref('');onMounted(async()=>{try{users.value=await getUsers()}catch(e){error.value=e instanceof Error?e.message:'无管理员权限'}});</script><template><main><h1>管理员用户管理</h1><p v-if="error">{{error}}</p><table v-else><tr><th>用户名</th><th>邮箱</th><th>角色</th></tr><tr v-for="u in users" :key="u.id"><td>{{u.username}}</td><td>{{u.email}}</td><td>{{u.role}}</td></tr></table></main></template>
|
|
||||||
@@ -1 +0,0 @@
|
|||||||
<script setup lang="ts">import { ref } from 'vue';import { useRoute,useRouter } from 'vue-router';import { login } from '../services/auth';const username=ref(''),password=ref(''),error=ref(''),router=useRouter(),route=useRoute();async function submit(){try{await login({username:username.value,password:password.value});await router.replace(typeof route.query.redirect==='string'?route.query.redirect:'/')}catch(e){error.value=e instanceof Error?e.message:'登录失败'}}</script><template><main><h1>登录</h1><form @submit.prevent="submit"><input v-model="username" autocomplete="username" required placeholder="用户名"><input v-model="password" type="password" autocomplete="current-password" required placeholder="密码"><p v-if="error">{{error}}</p><button>登录</button></form><RouterLink to="/register">注册账户</RouterLink></main></template>
|
|
||||||
@@ -1 +0,0 @@
|
|||||||
<script setup lang="ts">import { onMounted,ref } from 'vue';import { getCurrentUser,logout } from '../services/auth';import { useRouter } from 'vue-router';import type { SessionUser } from '../types/auth';const user=ref<SessionUser>(),error=ref(''),router=useRouter();onMounted(async()=>{try{user.value=await getCurrentUser()}catch(e){error.value=e instanceof Error?e.message:'无法读取资料'}});async function signOut(){try{await logout()}finally{router.replace('/login')}}</script><template><main><h1>个人中心</h1><p v-if="error">{{error}}</p><dl v-else-if="user"><dt>用户名</dt><dd>{{user.username}}</dd><dt>昵称</dt><dd>{{user.nickname||'未设置'}}</dd><dt>邮箱</dt><dd>{{user.email||'未设置'}}</dd><dt>角色</dt><dd>{{user.role}}</dd><dt>注册时间</dt><dd>{{new Date(user.created_at).toLocaleString()}}</dd></dl><div><RouterLink to="/settings">账户设置</RouterLink><button @click="signOut">退出登录</button></div></main></template>
|
|
||||||
@@ -1 +0,0 @@
|
|||||||
<script setup lang="ts">import { ref } from 'vue';import { useRouter } from 'vue-router';import { register } from '../services/auth';const username=ref(''),password=ref(''),error=ref(''),router=useRouter();async function submit(){try{await register({username:username.value,password:password.value});await router.push('/login')}catch(e){error.value=e instanceof Error?e.message:'注册失败'}}</script><template><main><h1>注册</h1><form @submit.prevent="submit"><input v-model="username" required minlength="3" placeholder="用户名"><input v-model="password" required minlength="15" type="password" placeholder="至少 15 位密码"><p v-if="error">{{error}}</p><button>注册</button></form></main></template>
|
|
||||||
Reference in New Issue
Block a user