docpilot-cli 1.0.0__tar.gz → 1.0.1__tar.gz

{docpilot_cli-1.0.0 → docpilot_cli-1.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: docpilot-cli
-Version: 1.0.0
+Version: 1.0.1
 Summary: A local-first RAG pipeline CLI tool
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
@@ -20,9 +20,9 @@ Requires-Dist: pyfiglet>=0.8.0
 Requires-Dist: rich>=13.0.0
 Dynamic: license-file
 
-# 🚀 DocPilot CLI
+# 🚀 [DocPilot CLI](https://pypi.org/project/docpilot-cli/)
 
-![PyPI - Version](https://img.shields.io/pypi/v/docpilot-cli?color=blue&style=flat-square)
+[![PyPI - Version](https://img.shields.io/pypi/v/docpilot-cli?color=blue&style=flat-square)](https://pypi.org/project/docpilot-cli/)
 ![Python Version](https://img.shields.io/badge/python-3.12%2B-blue?style=flat-square)
 ![License](https://img.shields.io/badge/license-MIT-green?style=flat-square)
 
@@ -103,6 +103,13 @@ Manage your configuration, models, and local database with ease.
 ### `docpilot setup`
 Re-run the interactive setup wizard at any time to change your default models.
 
+### `docpilot project`
+Switch to a different project or check the active project. Each project has its own isolated vector database.
+```bash
+docpilot project
+docpilot project <project-name>
+```
+
 ### `docpilot clear`
 Wipe your local Chroma vector database to start fresh. Prompts for safety confirmation.
 

{docpilot_cli-1.0.0 → docpilot_cli-1.0.1}/README.md

@@ -1,6 +1,6 @@
-# 🚀 DocPilot CLI
+# 🚀 [DocPilot CLI](https://pypi.org/project/docpilot-cli/)
 
-![PyPI - Version](https://img.shields.io/pypi/v/docpilot-cli?color=blue&style=flat-square)
+[![PyPI - Version](https://img.shields.io/pypi/v/docpilot-cli?color=blue&style=flat-square)](https://pypi.org/project/docpilot-cli/)
 ![Python Version](https://img.shields.io/badge/python-3.12%2B-blue?style=flat-square)
 ![License](https://img.shields.io/badge/license-MIT-green?style=flat-square)
 
@@ -81,6 +81,13 @@ Manage your configuration, models, and local database with ease.
 ### `docpilot setup`
 Re-run the interactive setup wizard at any time to change your default models.
 
+### `docpilot project`
+Switch to a different project or check the active project. Each project has its own isolated vector database.
+```bash
+docpilot project
+docpilot project <project-name>
+```
+
 ### `docpilot clear`
 Wipe your local Chroma vector database to start fresh. Prompts for safety confirmation.
 

{docpilot_cli-1.0.0 → docpilot_cli-1.0.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "docpilot-cli"
-version = "1.0.0"
+version = "1.0.1"
 description = "A local-first RAG pipeline CLI tool"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -29,7 +29,6 @@ where = ["src"]
 [build-system]
 requires = ["setuptools>=68", "wheel"]
 build-backend = "setuptools.build_meta"
-
 [dependency-groups]
 dev = [
     "pytest>=8.0.0",

{docpilot_cli-1.0.0 → docpilot_cli-1.0.1}/src/docpilot/chat.py

@@ -1,8 +1,9 @@
 from langchain_ollama.llms import OllamaLLM as Ollama
 from langchain_core.prompts import ChatPromptTemplate
-from .embed import vectorstore
 import os
 from . import store
+from langchain_chroma import Chroma
+from langchain_ollama import OllamaEmbeddings
 
 config = store.load_config()
 retrieval_k = int(config.get("retrieval_k", 6))
@@ -17,7 +18,16 @@ model = Ollama(
     temperature=float(config.get("temperature", 0.1)),
 )
 
-retriever = vectorstore.as_retriever(search_kwargs={"k": retrieval_k})
+
+def _get_vectorstore():
+    """Get vectorstore for the active project."""
+    embeddings = OllamaEmbeddings(model=config.get("default_embed_model", "mxbai-embed-large:335m"))
+    db_location = store.get_active_project_path()
+    return Chroma(collection_name="documents", persist_directory=str(db_location), embedding_function=embeddings)
+
+
+retriever = None
+
 template = """
 You are an assistant for answering questions based on the following ingested documents. 
 Use the information in the documents to answer the question as best as you can. 
@@ -47,6 +57,8 @@ def _build_bounded_context(docs, max_total_chars: int = 6000, max_doc_chars: int
 
 
 def askai(question):
+    vectorstore = _get_vectorstore()
+    retriever = vectorstore.as_retriever(search_kwargs={"k": retrieval_k})
     rag = retriever.invoke(question)
     rag_text = _build_bounded_context(
         rag,

{docpilot_cli-1.0.0 → docpilot_cli-1.0.1}/src/docpilot/cli.py

@@ -21,11 +21,30 @@ console = Console()
 app = typer.Typer()
 
 # Project metadata
-PROJECT_VERSION = "0.0.1"
+PROJECT_VERSION = "1.0.0"
 PROJECT_NAME = "docpilot"
 PROJECT_DESCRIPTION = "A local-first RAG pipeline CLI tool"
 
 
+@app.command()
+def project(
+    action: str = typer.Argument(None, help="Action: 'status' to show active project, or project name to switch"),
+):
+    """Manage projects: show status or switch to a project."""
+    if action is None or action == "status":
+        active = store.get_active_project()
+        path = store.get_active_project_path()
+        console.print(f"[bold cyan]Active Project:[/bold cyan] {active}")
+        console.print(f"[dim]DB path:[/dim] {path}")
+    else:
+        try:
+            project_path = store.set_active_project(action)
+            console.print(f"[green]✓ Switched to project:[/green] [bold]{action}[/bold]")
+            console.print(f"[dim]DB path:[/dim] {project_path}")
+        except Exception as e:
+            console.print(f"[red]Error setting project:[/red] {e}")
+
+
 def loadpdf(file_path: Path) -> list[str]:
     try:
         PdfReader = importlib.import_module("pypdf").PdfReader
@@ -212,9 +231,8 @@ def ask(question: str):
 
 @app.command()
 def clear():
-    """Clear the local vector database."""
-    config = store.load_config()
-    db_path = Path(config.get("db_path", ""))
+    """Clear the local vector database for the active project."""
+    db_path = store.get_active_project_path()
 
     if not db_path.exists():
         console.print("[yellow]Database is already empty or does not exist.[/yellow]")

{docpilot_cli-1.0.0 → docpilot_cli-1.0.1}/src/docpilot/embed.py

@@ -88,9 +88,9 @@ def _print_progress(label: str, current: int, total: int, width: int = 28) -> No
 
 
 embeddings = OllamaEmbeddings(model=store.load_config().get("default_embed_model", "mxbai-embed-large:335m"))
-db_location = store.path / "chroma_langchain_db"
+db_location = store.get_active_project_path()
 
-vectorstore = Chroma(collection_name="documents", persist_directory=db_location, embedding_function=embeddings)
+vectorstore = Chroma(collection_name="documents", persist_directory=str(db_location), embedding_function=embeddings)
 
 
 def chunk_text(

{docpilot_cli-1.0.0 → docpilot_cli-1.0.1}/src/docpilot/store.py

@@ -6,6 +6,7 @@ import os
 
 path = Path.home() / ".docpilot"
 CONFIG_PATH = path / "config.toml"
+PROJECTS_DIR = path / "projects"
 
 
 def _get_available_models():
@@ -22,7 +23,7 @@ _AVAILABLE_MODELS = _get_available_models()
 DEFAULT_CONFIG = {
     "default_embed_model": _AVAILABLE_MODELS[0] if _AVAILABLE_MODELS else "mxbai-embed-large:335m",
     "default_model": _AVAILABLE_MODELS[1] if len(_AVAILABLE_MODELS) > 1 else "llama2",
-    "db_path": str(path / "chroma_langchain_db"),
+    "active_project": "default",
     "log_level": "info",
     "retrieval_k": 6,
     "max_context_chars": 3500,
@@ -52,6 +53,8 @@ def init_config():
     """Initialize config file with defaults if it doesn't exist."""
     if not CONFIG_PATH.exists():
         CONFIG_PATH.parent.mkdir(parents=True, exist_ok=True)
+        PROJECTS_DIR.mkdir(parents=True, exist_ok=True)
+        (PROJECTS_DIR / "default").mkdir(parents=True, exist_ok=True)
         interactive_setup(first_time=True)
     return load_config()
 
@@ -111,3 +114,26 @@ def save_config(config: dict):
     CONFIG_PATH.parent.mkdir(parents=True, exist_ok=True)
     with open(CONFIG_PATH, "wb") as f:
         tomli_w.dump(config, f)
+
+
+def get_active_project() -> str:
+    """Get the currently active project name."""
+    config = load_config()
+    return config.get("active_project", "default")
+
+
+def set_active_project(project_name: str) -> Path:
+    """Set active project and ensure its directory exists. Returns the project path."""
+    project_path = PROJECTS_DIR / project_name / "chroma_langchain_db"
+    project_path.parent.mkdir(parents=True, exist_ok=True)
+
+    config = load_config()
+    config["active_project"] = project_name
+    save_config(config)
+    return project_path
+
+
+def get_active_project_path() -> Path:
+    """Get the Chroma DB path for the currently active project."""
+    active_project = get_active_project()
+    return PROJECTS_DIR / active_project / "chroma_langchain_db"

{docpilot_cli-1.0.0 → docpilot_cli-1.0.1}/src/docpilot_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: docpilot-cli
-Version: 1.0.0
+Version: 1.0.1
 Summary: A local-first RAG pipeline CLI tool
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
@@ -20,9 +20,9 @@ Requires-Dist: pyfiglet>=0.8.0
 Requires-Dist: rich>=13.0.0
 Dynamic: license-file
 
-# 🚀 DocPilot CLI
+# 🚀 [DocPilot CLI](https://pypi.org/project/docpilot-cli/)
 
-![PyPI - Version](https://img.shields.io/pypi/v/docpilot-cli?color=blue&style=flat-square)
+[![PyPI - Version](https://img.shields.io/pypi/v/docpilot-cli?color=blue&style=flat-square)](https://pypi.org/project/docpilot-cli/)
 ![Python Version](https://img.shields.io/badge/python-3.12%2B-blue?style=flat-square)
 ![License](https://img.shields.io/badge/license-MIT-green?style=flat-square)
 
@@ -103,6 +103,13 @@ Manage your configuration, models, and local database with ease.
 ### `docpilot setup`
 Re-run the interactive setup wizard at any time to change your default models.
 
+### `docpilot project`
+Switch to a different project or check the active project. Each project has its own isolated vector database.
+```bash
+docpilot project
+docpilot project <project-name>
+```
+
 ### `docpilot clear`
 Wipe your local Chroma vector database to start fresh. Prompts for safety confirmation.
 

{docpilot_cli-1.0.0 → docpilot_cli-1.0.1}/tests/test_store.py

@@ -16,8 +16,5 @@ def test_init_config_creates_file(temp_config_path):
     with patch("docpilot.store.interactive_setup") as mock_setup:
         # Prevent it from prompting
         mock_setup.return_value = None
-        config = store.init_config()
-
-        # We don't check for file existence because we mocked the setup function
-        # that normally creates the file. We just verify it was called.
+        store.init_config()
         mock_setup.assert_called_once()