cortex-local 2.1.0__tar.gz → 2.1.12__tar.gz

{cortex_local-2.1.0 → cortex_local-2.1.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cortex-local
-Version: 2.1.0
+Version: 2.1.12
 Summary: Local semantic search MCP server — notes + code, embedded Qdrant, fastembed
 Requires-Python: >=3.11
 Requires-Dist: click>=8.0.0

{cortex_local-2.1.0 → cortex_local-2.1.12}/cortex/config.py

@@ -1,5 +1,6 @@
 """Paths and constants for cortex local mode."""
 
+import os
 import shutil
 from pathlib import Path
 
@@ -9,6 +10,7 @@ ACTIVE_WORKSPACE_FILE = CORTEX_DIR / "active_workspace"
 
 EMBED_MODEL = "nomic-ai/nomic-embed-text-v1.5"
 VECTOR_SIZE = 768
+PPR_THRESHOLD = float(os.environ.get("PPR_THRESHOLD", "0.03"))
 
 try:
     from importlib.metadata import version

{cortex_local-2.1.0 → cortex_local-2.1.12}/cortex/core/chunker.py

@@ -17,6 +17,86 @@ CHUNK_LINES = 30
 OVERLAP_LINES = 5
 MAX_CHUNK_LINES = 80
 
+_FRAMEWORK_FILENAMES = {
+    "hooks.server":    "sveltekit server hooks handle",
+    "hooks.client":    "sveltekit client hooks",
+    "+page":           "sveltekit page component",
+    "+layout":         "sveltekit layout component",
+    "+server":         "sveltekit api route endpoint",
+    "+page.server":    "sveltekit page server load action",
+    "+layout.server":  "sveltekit layout server load",
+    "+error":          "sveltekit error page",
+    "svelte.config":   "sveltekit",
+    "next.config":     "nextjs",
+    "nuxt.config":     "nuxt",
+    "astro.config":    "astro",
+    "remix.config":    "remix",
+    "_app":            "nextjs",
+    "_document":       "nextjs",
+}
+
+_FRAMEWORK_IMPORTS = [
+    ("@sveltejs/kit",          "sveltekit"),
+    ("from 'react'",           "react"),
+    ('from "react"',           "react"),
+    ("from 'next/",            "nextjs"),
+    ('from "next/',            "nextjs"),
+    ("from 'vue'",             "vue"),
+    ('from "vue"',             "vue"),
+    ("@angular/core",          "angular"),
+    ("from 'express'",         "express"),
+    ('from "express"',         "express"),
+    ("from 'fastify'",         "fastify"),
+    ('from "fastify"',         "fastify"),
+    ("from 'solid-js'",        "solidjs"),
+    ('from "solid-js"',        "solidjs"),
+    ("@remix-run/",            "remix"),
+    ("from 'astro'",           "astro"),
+    ('from "astro"',           "astro"),
+    ("from fastapi",           "fastapi"),
+    ("import fastapi",         "fastapi"),
+    ("from django",            "django"),
+    ("import django",          "django"),
+    ("from flask",             "flask"),
+    ("import flask",           "flask"),
+    ("from starlette",         "starlette"),
+    ("from pydantic",          "pydantic"),
+    ("from sqlalchemy",        "sqlalchemy"),
+    ("import torch",           "pytorch"),
+    ("from torch",             "pytorch"),
+    ("import tensorflow",      "tensorflow"),
+    ("from tensorflow",        "tensorflow"),
+    ("import numpy",           "numpy"),
+    ("import pandas",          "pandas"),
+    ("from sklearn",           "scikit-learn"),
+    ("from transformers",      "huggingface"),
+    ("from qdrant_client",     "qdrant"),
+    ("from fastembed",         "fastembed"),
+    ("import io.ktor",         "ktor"),
+    ("import org.springframework", "spring"),
+    ("import androidx",        "android"),
+    ("gin-gonic/gin",          "gin"),
+    ("gofiber/fiber",          "fiber"),
+    ("use actix",              "actix"),
+    ("use axum",               "axum"),
+    ("use tokio",              "tokio"),
+]
+
+
+def _extra_tokens(rel: str, lines: list) -> str:
+    stem = Path(rel).stem
+    parts = []
+    if stem in _FRAMEWORK_FILENAMES:
+        parts.append(_FRAMEWORK_FILENAMES[stem])
+    header = "\n".join(lines[:50])
+    seen = set(parts)
+    for pattern, framework in _FRAMEWORK_IMPORTS:
+        if pattern in header and framework not in seen:
+            parts.append(framework)
+            seen.add(framework)
+    return " ".join(parts)
+
+
 try:
     from tree_sitter import Language, Parser as _TSParser
     import tree_sitter_python as _tspy
@@ -49,7 +129,7 @@ except Exception:
     _TS_SEMANTIC = {}
 
 
-def _sliding_window(lines: list, start: int, end: int, repo_name: str, rel: str, language: str, github_url_base: str = "") -> list:
+def _sliding_window(lines: list, start: int, end: int, repo_name: str, rel: str, language: str, github_url_base: str = "", extra_tokens: str = "") -> list:
     step = CHUNK_LINES - OVERLAP_LINES
     chunks = []
     i = start
@@ -58,10 +138,13 @@ def _sliding_window(lines: list, start: int, end: int, repo_name: str, rel: str,
         body = "\n".join(lines[i:j]).strip()
         if body:
             url = f"{github_url_base}/{rel}#L{i+1}-L{j}" if github_url_base else ""
+            header = f"# {repo_name}/{rel} (lines {i+1}-{j})"
+            if extra_tokens:
+                header += f"\n# {extra_tokens}"
             chunks.append({
                 "repo": repo_name, "file": rel, "language": language,
                 "start_line": i + 1, "end_line": j,
-                "text": f"# {repo_name}/{rel} (lines {i+1}-{j})\n\n{body}",
+                "text": f"{header}\n\n{body}",
                 "github_url": url,
             })
         if j == end:
@@ -110,6 +193,7 @@ def chunk_file(path: Path, repo_name: str, base_dir: Path | None = None, github_
     except ValueError:
         rel = path.name
     language = LANG_MAP.get(ext, ext.lstrip("."))
+    extra = _extra_tokens(rel, lines)
 
     if _TS_AVAILABLE and ext in _TS_LANGUAGES:
         try:
@@ -123,19 +207,22 @@ def chunk_file(path: Path, repo_name: str, base_dir: Path | None = None, github_
                     s = node.start_point[0]
                     e = node.end_point[0] + 1
                     if e - s > MAX_CHUNK_LINES:
-                        chunks.extend(_sliding_window(lines, s, e, repo_name, rel, language, github_url_base))
+                        chunks.extend(_sliding_window(lines, s, e, repo_name, rel, language, github_url_base, extra))
                     else:
                         body = "\n".join(lines[s:e]).strip()
                         if body:
                             url = f"{github_url_base}/{rel}#L{s+1}-L{e}" if github_url_base else ""
+                            header = f"# {repo_name}/{rel} (lines {s+1}-{e})"
+                            if extra:
+                                header += f"\n# {extra}"
                             chunks.append({
                                 "repo": repo_name, "file": rel, "language": language,
                                 "start_line": s + 1, "end_line": e,
-                                "text": f"# {repo_name}/{rel} (lines {s+1}-{e})\n\n{body}",
+                                "text": f"{header}\n\n{body}",
                                 "github_url": url,
                             })
                 return chunks
         except Exception:
             pass
 
-    return _sliding_window(lines, 0, len(lines), repo_name, rel, language, github_url_base)
+    return _sliding_window(lines, 0, len(lines), repo_name, rel, language, github_url_base, extra)

{cortex_local-2.1.0 → cortex_local-2.1.12}/cortex/mcp_server.py

@@ -13,7 +13,7 @@ from qdrant_client.models import Prefetch, Fusion, SparseVector
 from .config import (
     qdrant_path, data_dir, VERSION, VECTOR_SIZE,
     WORKSPACES_DIR, get_active_workspace, set_active_workspace,
-    get_workspace_dir,
+    get_workspace_dir, PPR_THRESHOLD,
 )
 from .embedder import embed, sparse_embed
 from .state import _stats, get_graph_meta, save_stats, invalidate_graph_cache
@@ -59,7 +59,7 @@ def _ppr_block(matched_files: list, matched_scores: list) -> str | None:
     except Exception:
         return None
     graph_path = data_dir() / "graph_notes.json"
-    extras, reason = ppr_augment(matched_files, matched_scores, graph_path, _return_reason=True)
+    extras, reason = ppr_augment(matched_files, matched_scores, graph_path, threshold=PPR_THRESHOLD, _return_reason=True)
     if extras:
         _stats["ppr_fires"] += 1
         _stats["ppr_results_added"] += len(extras)

{cortex_local-2.1.0 → cortex_local-2.1.12}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "cortex-local"
-version = "2.1.0"
+dynamic = ["version"]
 description = "Local semantic search MCP server — notes + code, embedded Qdrant, fastembed"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -25,5 +25,10 @@ cortex = "cortex.cli:cli"
 [project.entry-points."mcp.servers"]
 cortex = "cortex.mcp_server:run"
 
+[tool.hatch.version]
+source = "regex"
+path = "../VERSION"
+pattern = "^(?P<version>[0-9]+\\.[0-9]+\\.[0-9]+)"
+
 [tool.hatch.build.targets.wheel]
 packages = ["cortex"]