codrspot-processor-mcp 0.1.3__tar.gz → 0.1.4__tar.gz

{codrspot_processor_mcp-0.1.3 → codrspot_processor_mcp-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codrspot-processor-mcp
-Version: 0.1.3
+Version: 0.1.4
 Summary: Local MCP server exposing codepreproc's code-context tools to Claude Code
 Author-email: alexlqi <alexlqi@gmail.com>
 License: Copyright (c) 2026 alexlqi (alexlqi@gmail.com)

{codrspot_processor_mcp-0.1.3 → codrspot_processor_mcp-0.1.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "codrspot-processor-mcp"
-version = "0.1.3"
+version = "0.1.4"
 description = "Local MCP server exposing codepreproc's code-context tools to Claude Code"
 readme = "README.md"
 requires-python = ">=3.12,<3.13"

{codrspot_processor_mcp-0.1.3 → codrspot_processor_mcp-0.1.4}/src/codepreproc_client/layer1_business/config.py

@@ -33,7 +33,9 @@ def apply_env_overrides(registry: Registry) -> Registry:
     new_defaults = d.model_copy(
         update={
             "embedding_model": os.getenv("CODEPREPROC_EMBEDDING_MODEL", d.embedding_model),
+            "embedding_device": os.getenv("CODEPREPROC_EMBEDDING_DEVICE", d.embedding_device),
             "reranker_model": os.getenv("CODEPREPROC_RERANKER_MODEL", d.reranker_model),
+            "reranker_device": os.getenv("CODEPREPROC_RERANKER_DEVICE", d.reranker_device),
             "llm_endpoint": os.getenv("CODEPREPROC_LLM_ENDPOINT", d.llm_endpoint),
             "llm_model": os.getenv("CODEPREPROC_LLM_MODEL", d.llm_model),
             "default_llm_policy": os.getenv("CODEPREPROC_DEFAULT_LLM_POLICY", d.default_llm_policy),

{codrspot_processor_mcp-0.1.3 → codrspot_processor_mcp-0.1.4}/src/codepreproc_client/layer2_tooling/condensation/embed_worker.py

@@ -61,8 +61,12 @@ def main() -> None:
     device = config["device"]
 
     try:
+        import torch
         from sentence_transformers import SentenceTransformer
 
+        if device.startswith("cuda") and not torch.cuda.is_available():
+            device = "cpu"
+
         model = SentenceTransformer(model_name, device=device)
         _send({"type": "worker_ready"})
     except Exception as exc:

{codrspot_processor_mcp-0.1.3 → codrspot_processor_mcp-0.1.4}/src/codepreproc_client/layer2_tooling/condensation/embeddings.py

@@ -39,8 +39,12 @@ def _coerce_batch_vectors(vectors: Any, *, expected_rows: int) -> list[list[floa
 
 def _embedding_worker(model_name: str, device: str, command_queue: Queue, event_queue: Queue) -> None:
     try:
+        import torch
         from sentence_transformers import SentenceTransformer
 
+        if device.startswith("cuda") and not torch.cuda.is_available():
+            device = "cpu"
+
         model = SentenceTransformer(model_name, device=device)
         event_queue.put({"type": "worker_ready"})
     except Exception as exc:

{codrspot_processor_mcp-0.1.3 → codrspot_processor_mcp-0.1.4}/src/codepreproc_client/layer2_tooling/retrieval/rerank_worker.py

@@ -42,8 +42,12 @@ def main() -> None:
     device = config["device"]
 
     try:
+        import torch
         from sentence_transformers import CrossEncoder
 
+        if device.startswith("cuda") and not torch.cuda.is_available():
+            device = "cpu"
+
         model = CrossEncoder(model_name, device=device)
         _send({"type": "worker_ready"})
     except Exception as exc:

{codrspot_processor_mcp-0.1.3 → codrspot_processor_mcp-0.1.4}/src/codepreproc_client/layer2_tooling/retrieval/reranker.py

@@ -87,9 +87,14 @@ class Reranker:
 
     def _ensure_model(self):
         if self._model is None:
+            import torch
             from sentence_transformers import CrossEncoder
 
-            self._model = CrossEncoder(self.defaults.reranker_model, device=self.defaults.reranker_device)
+            device = self.defaults.reranker_device
+            if device.startswith("cuda") and not torch.cuda.is_available():
+                device = "cpu"
+
+            self._model = CrossEncoder(self.defaults.reranker_model, device=device)
         return self._model
 
     # ------------------------------------------------------------------

{codrspot_processor_mcp-0.1.3 → codrspot_processor_mcp-0.1.4}/src/codrspot_processor_mcp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codrspot-processor-mcp
-Version: 0.1.3
+Version: 0.1.4
 Summary: Local MCP server exposing codepreproc's code-context tools to Claude Code
 Author-email: alexlqi <alexlqi@gmail.com>
 License: Copyright (c) 2026 alexlqi (alexlqi@gmail.com)