knowledge-master 0.4.0__tar.gz → 0.5.0__tar.gz

{knowledge_master-0.4.0 → knowledge_master-0.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: knowledge-master
-Version: 0.4.0
+Version: 0.5.0
 Summary: Local-first knowledge graph for developers. Your AI agent's permanent memory.
 Author: Milenko Mitrovic
 License: MIT

{knowledge_master-0.4.0 → knowledge_master-0.5.0}/knowledge_master.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: knowledge-master
-Version: 0.4.0
+Version: 0.5.0
 Summary: Local-first knowledge graph for developers. Your AI agent's permanent memory.
 Author: Milenko Mitrovic
 License: MIT

{knowledge_master-0.4.0 → knowledge_master-0.5.0}/knowledge_master.egg-info/SOURCES.txt

@@ -29,5 +29,10 @@ knowledge_master/parsers/markdown.py
 tests/test_api.py
 tests/test_chunking.py
 tests/test_cli.py
+tests/test_connectors.py
+tests/test_dedup.py
 tests/test_intelligence.py
-tests/test_static_analysis.py
+tests/test_migrations.py
+tests/test_reranker.py
+tests/test_static_analysis.py
+tests/test_ts_parsers.py

{knowledge_master-0.4.0 → knowledge_master-0.5.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "knowledge-master"
-version = "0.4.0"
+version = "0.5.0"
 description = "Local-first knowledge graph for developers. Your AI agent's permanent memory."
 requires-python = ">=3.11"
 license = {text = "MIT"}

knowledge_master-0.5.0/tests/test_connectors.py

@@ -0,0 +1,58 @@
+"""Tests for connector framework."""
+from knowledge_master.connectors import MCPSource, SOURCES, add_custom_source, _parse_mcp_result
+
+
+def test_preconfigured_sources_exist():
+    assert "outlook" in SOURCES
+    assert "slack" in SOURCES
+
+
+def test_source_has_required_fields():
+    for name, source in SOURCES.items():
+        assert source.name, f"{name} missing name"
+        assert source.command, f"{name} missing command"
+        assert source.tool_name, f"{name} missing tool_name"
+        assert source.source_type, f"{name} missing source_type"
+
+
+def test_add_custom_source():
+    add_custom_source("test-source", ["echo", "hi"], "get_data", {"limit": 10}, "custom")
+    assert "test-source" in SOURCES
+    assert SOURCES["test-source"].tool_name == "get_data"
+    # Cleanup
+    del SOURCES["test-source"]
+
+
+def test_parse_mcp_result_json_list():
+    class FakeContent:
+        text = '[{"title": "hello", "body": "world"}]'
+
+    class FakeResult:
+        content = [FakeContent()]
+
+    items = _parse_mcp_result(FakeResult())
+    assert len(items) == 1
+    assert items[0]["title"] == "hello"
+
+
+def test_parse_mcp_result_json_dict_with_results():
+    class FakeContent:
+        text = '{"results": [{"text": "a"}, {"text": "b"}]}'
+
+    class FakeResult:
+        content = [FakeContent()]
+
+    items = _parse_mcp_result(FakeResult())
+    assert len(items) == 2
+
+
+def test_parse_mcp_result_plain_text():
+    class FakeContent:
+        text = "just plain text"
+
+    class FakeResult:
+        content = [FakeContent()]
+
+    items = _parse_mcp_result(FakeResult())
+    assert len(items) == 1
+    assert items[0]["text"] == "just plain text"

knowledge_master-0.5.0/tests/test_dedup.py

@@ -0,0 +1,24 @@
+"""Tests for store deduplication and content hashing."""
+from knowledge_master.store import content_hash
+
+
+def test_content_hash_deterministic():
+    h1 = content_hash("hello world")
+    h2 = content_hash("hello world")
+    assert h1 == h2
+
+
+def test_content_hash_different_for_different_content():
+    h1 = content_hash("hello world")
+    h2 = content_hash("hello world!")
+    assert h1 != h2
+
+
+def test_content_hash_length():
+    h = content_hash("test")
+    assert len(h) == 16  # sha256 truncated to 16 chars
+
+
+def test_content_hash_hex():
+    h = content_hash("test")
+    assert all(c in "0123456789abcdef" for c in h)

knowledge_master-0.5.0/tests/test_migrations.py

@@ -0,0 +1,23 @@
+"""Tests for schema migrations."""
+from knowledge_master.migrations import (
+    CURRENT_SCHEMA_VERSION,
+    MIGRATIONS,
+    _migrate_to_v1,
+    _migrate_to_v3,
+    _migrate_to_v4,
+)
+
+
+def test_current_schema_version():
+    assert CURRENT_SCHEMA_VERSION == 4
+
+
+def test_all_migrations_defined():
+    for v in range(1, CURRENT_SCHEMA_VERSION + 1):
+        assert v in MIGRATIONS, f"Migration to v{v} not defined"
+
+
+def test_migrations_are_callable():
+    for v, fn in MIGRATIONS.items():
+        assert callable(fn)
+        assert fn.__doc__ is not None, f"Migration v{v} has no docstring"

knowledge_master-0.5.0/tests/test_reranker.py

@@ -0,0 +1,43 @@
+"""Benchmark test for re-ranker — compare raw cosine vs re-ranked ordering.
+
+This is an integration test (requires Ollama) but demonstrates the value of re-ranking.
+Run with: pytest tests/integration/test_reranker_benchmark.py -v
+"""
+import pytest
+from knowledge_master.rerank import rerank, _cosine_sim
+
+
+def test_cosine_sim_identical():
+    a = [1.0, 0.0, 0.0]
+    assert _cosine_sim(a, a) == pytest.approx(1.0)
+
+
+def test_cosine_sim_orthogonal():
+    a = [1.0, 0.0, 0.0]
+    b = [0.0, 1.0, 0.0]
+    assert _cosine_sim(a, b) == pytest.approx(0.0)
+
+
+def test_cosine_sim_opposite():
+    a = [1.0, 0.0]
+    b = [-1.0, 0.0]
+    assert _cosine_sim(a, b) == pytest.approx(-1.0)
+
+
+def test_rerank_empty():
+    result = rerank("test query", [], top_k=5)
+    assert result == []
+
+
+def test_rerank_preserves_fields():
+    """Re-rank should keep all original result fields."""
+    results = [
+        {"text": "hello world", "source": "a.py", "score": 0.5, "extra": "keep"},
+    ]
+    # This would normally call Ollama — skip if not available
+    try:
+        ranked = rerank("hello", results, top_k=1)
+        assert "extra" in ranked[0]
+        assert "rerank_score" in ranked[0]
+    except Exception:
+        pytest.skip("Ollama not available")

knowledge_master-0.5.0/tests/test_ts_parsers.py

@@ -0,0 +1,96 @@
+"""Tests for tree-sitter based language parsers."""
+import tempfile
+from pathlib import Path
+
+from knowledge_master.ts_parsers import (
+    extract_typescript_graph,
+    extract_go_graph,
+    extract_rust_graph,
+    resolve_ts_import,
+)
+
+
+def test_typescript_import_extraction():
+    with tempfile.NamedTemporaryFile(suffix=".ts", mode="w", delete=False) as f:
+        f.write('import { useState } from "react";\n')
+        f.write('import { helper } from "./utils";\n')
+        f.write("export function App() { return null; }\n")
+        f.flush()
+        result = extract_typescript_graph(f.name)
+
+    assert len(result["imports"]) == 2
+    assert result["imports"][0]["module"] == "react"
+    assert result["imports"][1]["module"] == "./utils"
+    assert len(result["exports"]) >= 1
+    assert any(e["name"] == "App" for e in result["exports"])
+
+
+def test_typescript_require_extraction():
+    with tempfile.NamedTemporaryFile(suffix=".js", mode="w", delete=False) as f:
+        f.write('const fs = require("fs");\n')
+        f.flush()
+        result = extract_typescript_graph(f.name)
+
+    assert len(result["imports"]) >= 1
+    assert result["imports"][0]["module"] == "fs"
+
+
+def test_go_import_extraction():
+    with tempfile.NamedTemporaryFile(suffix=".go", mode="w", delete=False) as f:
+        f.write('package main\n\nimport (\n\t"fmt"\n\t"os"\n)\n\nfunc Main() {}\n')
+        f.flush()
+        result = extract_go_graph(f.name)
+
+    assert len(result["imports"]) >= 2
+    modules = [i["module"] for i in result["imports"]]
+    assert "fmt" in modules
+    assert "os" in modules
+
+
+def test_go_exported_functions():
+    with tempfile.NamedTemporaryFile(suffix=".go", mode="w", delete=False) as f:
+        f.write('package pkg\n\nfunc PublicFunc() {}\nfunc privateFunc() {}\n')
+        f.flush()
+        result = extract_go_graph(f.name)
+
+    names = [e["name"] for e in result["exports"]]
+    assert "PublicFunc" in names
+    assert "privateFunc" not in names
+
+
+def test_rust_use_extraction():
+    with tempfile.NamedTemporaryFile(suffix=".rs", mode="w", delete=False) as f:
+        f.write("use std::collections::HashMap;\nuse crate::utils;\n\npub fn hello() {}\nfn private() {}\n")
+        f.flush()
+        result = extract_rust_graph(f.name)
+
+    assert len(result["imports"]) >= 2
+    modules = [i["module"] for i in result["imports"]]
+    assert "std" in modules
+    assert any(e["name"] == "hello" for e in result["exports"])
+    assert not any(e["name"] == "private" for e in result["exports"])
+
+
+def test_rust_mod_detection():
+    with tempfile.NamedTemporaryFile(suffix=".rs", mode="w", delete=False) as f:
+        f.write("mod parser;\nmod utils;\n")
+        f.flush()
+        result = extract_rust_graph(f.name)
+
+    mod_imports = [i for i in result["imports"] if i.get("is_mod")]
+    assert len(mod_imports) == 2
+    assert mod_imports[0]["module"] == "parser"
+
+
+def test_ts_import_resolution():
+    with tempfile.TemporaryDirectory() as tmp:
+        (Path(tmp) / "src").mkdir()
+        (Path(tmp) / "src" / "utils.ts").write_text("export const x = 1;")
+        (Path(tmp) / "src" / "index.ts").write_text("")
+
+        result = resolve_ts_import("./utils", "src/index.ts", tmp)
+        assert result == "src/utils.ts"
+
+        # External package — should return None
+        result = resolve_ts_import("react", "src/index.ts", tmp)
+        assert result is None