local-knowledge 0.2.2__tar.gz → 0.2.4__tar.gz

{local_knowledge-0.2.2 → local_knowledge-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: local-knowledge
-Version: 0.2.2
+Version: 0.2.4
 Summary: Local file/folder indexer + grounded-Q&A MCP service — same engine as confluence-knowledge, no credentials needed
 License: MIT
 Project-URL: Repository, https://github.com/nkunal1992/agents
@@ -20,7 +20,7 @@ Requires-Dist: openpyxl>=3.1
 Requires-Dist: mcp<2,>=1.6
 Requires-Dist: pip-system-certs>=5.0
 Provides-Extra: models
-Requires-Dist: confluence-knowledge[models]>=0.5.2; extra == "models"
+Requires-Dist: confluence-knowledge[models]>=0.5.4; extra == "models"
 Provides-Extra: dev
 Requires-Dist: pytest>=8.0; extra == "dev"
 Requires-Dist: pytest-cov>=5.0; extra == "dev"

{local_knowledge-0.2.2 → local_knowledge-0.2.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "local-knowledge"
-version = "0.2.2"
+version = "0.2.4"
 description = "Local file/folder indexer + grounded-Q&A MCP service — same engine as confluence-knowledge, no credentials needed"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -27,7 +27,7 @@ dependencies = [
 
 [project.optional-dependencies]
 models = [
-    "confluence-knowledge[models]>=0.5.2",
+    "confluence-knowledge[models]>=0.5.4",
 ]
 dev = [
     "pytest>=8.0",

{local_knowledge-0.2.2 → local_knowledge-0.2.4}/src/local_knowledge/crawl_local.py

@@ -133,6 +133,7 @@ class LocalCrawler:
 
         queue: list[str] = job["queue"]
         errors: list[str] = job["errors"]
+        keep_failed: set[str] = set()  # transient read fails keep old content
         # Mid-crawl model change: already-embedded pages live in the OLD
         # vector space — finishing would publish mixed dims. Wipe + fail;
         # start_crawl rebuilds under the new model.
@@ -186,6 +187,7 @@ class LocalCrawler:
                 page, chunks = self._page(path)
             except SetupError as e:
                 errors.append(str(e))
+                keep_failed.add(self._rel(path))
                 continue
             rel = page["page_id"]
             if rel in seen:
@@ -208,7 +210,7 @@ class LocalCrawler:
         # persist final seen/levels/warns too — the in-loop checkpoint runs
         # before each item is processed, so the last item would undercount
         self._checkpoint(job_id, queue, errors, seen, levels, warns)
-        self.index.publish(seen)
+        self.index.publish(seen | keep_failed)
         self.index.set_model_id(_MODEL_ID)
         self.index.save_job(job_id, "done" if not queue else "running",
                             queue, errors)

{local_knowledge-0.2.2 → local_knowledge-0.2.4}/src/local_knowledge/server_local.py

@@ -12,6 +12,7 @@ import socket
 import threading
 
 from confluence_knowledge.client import AccessError, SecurityException
+from confluence_knowledge.feedback import FeedbackStore
 from confluence_knowledge.index import Index
 from confluence_knowledge.retrieval import (
     MODEL_MARKER as EMBED_MODEL_ID,
@@ -32,8 +33,9 @@ from .settings_local import APP_NAME, LocalProfile, home, index_root, workdir
 mcp = FastMCP("local-knowledge")
 
 _scopes: dict[str, dict] = {}
-_evidence_sets: dict[str, list[dict]] = {}
+_evidence_sets: dict[str, dict] = {}  # set_id → {scope, ev}
 _indexes: dict[str, Index] = {}
+_feedback: dict[str, FeedbackStore] = {}
 _EMBEDDER = None
 _RERANKER = None
 _SPARSE = None
@@ -51,6 +53,13 @@ def _index_for(scope_key: str) -> Index:
         idx = _indexes[scope_key] = Index(index_root() / f"{scope_key}.db")
     return idx
 
+def _feedback_for(scope_key: str) -> FeedbackStore:
+    st = _feedback.get(scope_key)
+    if st is None:
+        st = _feedback[scope_key] = FeedbackStore(
+            index_root(), scope_key, "local-knowledge")
+    return st
+
 
 def _embedder():
     global _EMBEDDER
@@ -267,17 +276,25 @@ def search(scope_handle: str, question: str) -> dict:
         return {"evidence": [], "coverage": "stale_model",
                 "note": "this index was built with an older embedding model "
                         "— run start_crawl once to rebuild it, then search again"}
-    ev = retrieve(idx, question, _embedder(), _reranker(), sparse=_sparse())
+    emb = _embedder()
+    ev = retrieve(idx, question, emb, _reranker(), sparse=_sparse())
+    corrections = _feedback_for(s["key"]).match(emb.embed([question])[0])
     set_id = f"ev:{len(_evidence_sets)}:{s['key'][:8]}"
-    _evidence_sets[set_id] = ev
+    # bound the session cache — sets are never reused after the turn ends
+    while len(_evidence_sets) >= 32:
+        _evidence_sets.pop(next(iter(_evidence_sets)))
+    _evidence_sets[set_id] = {"scope": s["key"], "ev": ev}
     return {
         "evidence_set": set_id,
         "coverage": _coverage(idx),
         "as_of": st["as_of"],
+        "user_corrections": corrections,
         "evidence": [
             {"chunk_id": e["chunk_id"], "page_title": e["page_title"],
              "page_url": e["page_url"], "heading": e["heading"],
-             "score": e.get("rerank_score"), "text": e["text"][:2000]}
+             "score": e.get("rerank_score"), "text": e["text"][:12000],
+             "low_confidence": e.get("low_confidence", False),
+             "truncated": len(e["text"]) > 12000}
             for e in ev
         ],
     }
@@ -300,15 +317,39 @@ def get_evidence(scope_handle: str, chunk_ids: list[str]) -> dict:
     return {"chunks": out}
 
 
+@mcp.tool()
+
+
+@mcp.tool()
+def submit_feedback(scope_handle: str, question: str, correction: str,
+                    note: str = "") -> dict:
+    """Store a colleague correction — disclosed alongside the docs on
+    similar future questions (never overrides them)."""
+    s = _scope(scope_handle)
+    if not _models_loaded(0):
+        return {"stored": False, "note": LOADING_NOTE}
+    qvec = _embedder().embed([question])[0]
+    entry = _feedback_for(s["key"]).add(question, qvec, correction, note)
+    return {"stored": entry["id"],
+            "note": "will be disclosed on similar questions"}
+
+
+@mcp.tool()
+def feedback_report(scope_handle: str) -> dict:
+    """Aggregated corrections — which topics keep getting corrected points
+    at outdated/confusing documentation."""
+    s = _scope(scope_handle)
+    return _feedback_for(s["key"]).report()
+
 @mcp.tool()
 def verify_citations(scope_handle: str, evidence_set: str,
                      mappings: list[dict]) -> dict:
-    _scope(scope_handle)
-    ev = _evidence_sets.get(evidence_set)
-    if ev is None or not evidence_set.endswith(_scopes[scope_handle]["key"][:8]):
+    s = _scope(scope_handle)
+    rec = _evidence_sets.get(evidence_set)
+    if rec is None or rec["scope"] != s["key"]:
         raise SecurityException(
             "unknown or cross-scope evidence set — run search again and cite its evidence_set")
-    return check_citations(ev, mappings)
+    return check_citations(rec["ev"], mappings)
 
 
 def main() -> None:

{local_knowledge-0.2.2 → local_knowledge-0.2.4}/src/local_knowledge.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: local-knowledge
-Version: 0.2.2
+Version: 0.2.4
 Summary: Local file/folder indexer + grounded-Q&A MCP service — same engine as confluence-knowledge, no credentials needed
 License: MIT
 Project-URL: Repository, https://github.com/nkunal1992/agents
@@ -20,7 +20,7 @@ Requires-Dist: openpyxl>=3.1
 Requires-Dist: mcp<2,>=1.6
 Requires-Dist: pip-system-certs>=5.0
 Provides-Extra: models
-Requires-Dist: confluence-knowledge[models]>=0.5.2; extra == "models"
+Requires-Dist: confluence-knowledge[models]>=0.5.4; extra == "models"
 Provides-Extra: dev
 Requires-Dist: pytest>=8.0; extra == "dev"
 Requires-Dist: pytest-cov>=5.0; extra == "dev"

{local_knowledge-0.2.2 → local_knowledge-0.2.4}/src/local_knowledge.egg-info/requires.txt

@@ -12,4 +12,4 @@ pytest-cov>=5.0
 ruff>=0.6
 
 [models]
-confluence-knowledge[models]>=0.5.2
+confluence-knowledge[models]>=0.5.4

{local_knowledge-0.2.2 → local_knowledge-0.2.4}/tests/test_agent_doc_consistency.py

@@ -29,3 +29,24 @@ def test_agent_references_repo_files_that_exist():
     for m in re.findall(r"prompts/([\w.-]+)", doc):
         assert (REPO / "prompts" / m).exists(), \
             f"agent references missing prompts/{m}"
+
+
+def test_agent_frontmatter_tools_exist():
+    """Every tool named in the agent's frontmatter must be a real
+    @mcp.tool() in the server source."""
+    if not AGENT.exists():
+        import pytest
+        pytest.skip("agent file not in this checkout layout")
+    doc = AGENT.read_text()
+    tools_line = re.search(r'tools:\s*\[([^\]]+)\]', doc).group(1)
+    app = AGENT.stem.replace(".agent", "")
+    declared = {t.strip().strip('"').split("/")[1]
+                for t in tools_line.split(",") if f"{app}/" in t}
+    srv_name = {"local-knowledge": "server_local",
+                "sharepoint-knowledge": "server_sp",
+                "confluence-knowledge": "server"}[app]
+    pkg = app.replace("-", "_")
+    srv = (Path(__file__).parents[1] / "src" / pkg / f"{srv_name}.py").read_text()
+    real = set(re.findall(r"@mcp\.tool\(\)\s*\ndef\s+(\w+)", srv))
+    missing = declared - real
+    assert not missing, f"agent frontmatter names tools not in server: {missing}"