dug-cli 0.1.3__tar.gz → 0.1.4__tar.gz

{dug_cli-0.1.3 → dug_cli-0.1.4}/.github/workflows/release.yml

@@ -73,13 +73,13 @@ jobs:
           --onefile
           --name dug
           --collect-submodules dug
+          --collect-all fastembed
+          --collect-all onnxruntime
           --hidden-import=dug.__main__
           --exclude-module torch
           --exclude-module transformers
           --exclude-module sentence_transformers
           --exclude-module tokenizers
-          --exclude-module scipy
-          --exclude-module sklearn
           --exclude-module openai
           main.py
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dug-cli
-Version: 0.1.3
+Version: 0.1.4
 Summary: Dig into any bug with full codebase context — zero LLM calls
 Project-URL: Homepage, https://github.com/ratishjain12/dug
 Project-URL: Repository, https://github.com/ratishjain12/dug
@@ -20,6 +20,7 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Software Development :: Debuggers
 Requires-Python: >=3.10
 Requires-Dist: click
+Requires-Dist: fastembed
 Requires-Dist: lancedb
 Requires-Dist: networkx
 Requires-Dist: tree-sitter-java
@@ -28,11 +29,6 @@ Requires-Dist: tree-sitter-python
 Requires-Dist: tree-sitter-typescript>=0.23.2
 Requires-Dist: tree-sitter>=0.22
 Requires-Dist: watchdog
-Provides-Extra: all
-Requires-Dist: openai; extra == 'all'
-Requires-Dist: sentence-transformers; extra == 'all'
-Provides-Extra: local
-Requires-Dist: sentence-transformers; extra == 'local'
 Provides-Extra: openai
 Requires-Dist: openai; extra == 'openai'
 Description-Content-Type: text/markdown

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

@@ -1,6 +1,6 @@
 [project]
 name = "dug-cli"
-version = "0.1.3"
+version = "0.1.4"
 description = "Dig into any bug with full codebase context — zero LLM calls"
 readme = "README.md"
 license = { text = "MIT" }
@@ -28,6 +28,7 @@ dependencies = [
   "tree-sitter-typescript>=0.23.2",
   "lancedb",
   "watchdog",
+  "fastembed",
 ]
 
 [project.urls]
@@ -36,9 +37,7 @@ Repository = "https://github.com/ratishjain12/dug"
 "Bug Tracker" = "https://github.com/ratishjain12/dug/issues"
 
 [project.optional-dependencies]
-local  = ["sentence-transformers"]
 openai = ["openai"]
-all    = ["sentence-transformers", "openai"]
 
 [project.scripts]
 dug = "dug.__main__:cli"

{dug_cli-0.1.3 → dug_cli-0.1.4}/src/dug/embeddings.py

@@ -15,14 +15,13 @@ warnings.filterwarnings("ignore", category=UserWarning, module="huggingface_hub"
 warnings.filterwarnings("ignore", category=FutureWarning, module="transformers")
 
 for _noisy in ("sentence_transformers", "huggingface_hub", "transformers",
-               "torch", "tokenizers"):
+               "torch", "tokenizers", "fastembed"):
     logging.getLogger(_noisy).setLevel(logging.ERROR)
 
 # ---------------------------------------------------------------------------
-# Dependency installer
+# Dependency installer (OpenAI only — fastembed is a base dep)
 # ---------------------------------------------------------------------------
 
-_LOCAL_DEPS  = ["sentence-transformers"]
 _OPENAI_DEPS = ["openai"]
 
 
@@ -40,14 +39,9 @@ def _ensure_installed(packages: list[str], label: str) -> None:
     if not missing:
         return
 
-    # PyInstaller binary: sys.executable is the frozen binary, not Python
     if getattr(sys, "frozen", False):
         print(f"\n[dug] Running as a standalone binary — cannot auto-install {label} packages.")
-        print(f"[dug] Switch to OpenAI embeddings instead:")
-        print(f"      dug config set embedding_mode openai")
-        print(f"      dug config set api_key <your-openai-key>")
-        print(f"[dug] Or reinstall dug via pipx for local embeddings:")
-        print(f"      pipx install dug-cli")
+        print(f"[dug] Install via pipx instead: pipx install dug-cli")
         sys.exit(1)
 
     print(f"\n[dug] {label} dependencies not found: {', '.join(missing)}")
@@ -72,12 +66,11 @@ def _ensure_installed(packages: list[str], label: str) -> None:
 
 class LocalEmbedder:
     def __init__(self):
-        _ensure_installed(_LOCAL_DEPS, "Local embedding")
-        from sentence_transformers import SentenceTransformer
-        self.model = SentenceTransformer("all-MiniLM-L6-v2")
+        from fastembed import TextEmbedding
+        self.model = TextEmbedding("sentence-transformers/all-MiniLM-L6-v2")
 
     def embed(self, text: str) -> list[float]:
-        return self.model.encode(text).tolist()
+        return list(self.model.embed([text]))[0].tolist()
 
 
 class OpenAIEmbedder: