mirror of
https://github.com/TencentCloud/Octop.git
synced 2026-10-21 06:29:24 +08:00
jubaoliangandCursor
e8a824bb1e
Introduce document indexing, embedding (ONNX/remote), and chat/IM injection with default-open selection in the dashboard composer. Co-authored-by: Cursor <cursoragent@cursor.com>
32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
"""Unit tests for per-knowledge-base sidecar indexes."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from octop.infra.knowledge.index import KnowledgeIndex
|
|
|
|
|
|
def test_index_replaces_document_chunks_and_returns_cosine_top_k(tmp_path, monkeypatch) -> None:
|
|
monkeypatch.setenv("OCTOP_HOME", str(tmp_path))
|
|
index = KnowledgeIndex("kb-1")
|
|
|
|
index.replace_doc_chunks(
|
|
"doc-1",
|
|
["first", "second"],
|
|
[[1.0, 0.0], [0.0, 1.0]],
|
|
metadata=[{"page": 1}, {"page": 2}],
|
|
)
|
|
index.replace_doc_chunks("doc-2", ["third"], [[0.8, 0.2]])
|
|
|
|
hits = index.search([1.0, 0.0], k=2)
|
|
|
|
assert [hit.text for hit in hits] == ["first", "third"]
|
|
assert hits[0].doc_id == "doc-1"
|
|
assert hits[0].ordinal == 0
|
|
assert hits[0].metadata == {"page": 1}
|
|
|
|
index.replace_doc_chunks("doc-1", ["replacement"], [[0.0, 1.0]])
|
|
assert [hit.text for hit in index.search([1.0, 0.0], k=5)] == ["third", "replacement"]
|
|
|
|
index.delete_doc("doc-2")
|
|
assert [hit.doc_id for hit in index.search([1.0, 0.0], k=5)] == ["doc-1"]
|