modaic 0.1.1__tar.gz → 0.2.0__tar.gz

{modaic-0.1.1/src/modaic.egg-info → modaic-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: modaic
-Version: 0.1.1
+Version: 0.2.0
 Summary: Modular Agent Infrastructure Collective, a python framework for managing and sharing DSPy agents
 Author-email: Tyrin <tytodd@mit.edu>, Farouk <farouk@modaic.dev>
 License: MIT License
@@ -48,7 +48,8 @@ Requires-Dist: langchain-community>=0.3.29
 Requires-Dist: langchain-core>=0.3.72
 Requires-Dist: langchain-text-splitters>=0.3.9
 Requires-Dist: more-itertools>=10.8.0
-Requires-Dist: opik>=1.8.42
+Requires-Dist: openpyxl>=3.1.5
+Requires-Dist: opik==1.8.42
 Requires-Dist: pillow>=11.3.0
 Requires-Dist: pymilvus>=2.5.14
 Requires-Dist: sqlalchemy>=2.0.42
@@ -58,8 +59,11 @@ Requires-Dist: pinecone>=7.3.0; extra == "pinecone"
 Dynamic: license-file
 
 [![Docs](https://img.shields.io/badge/docs-available-brightgreen.svg)](https://docs.modaic.dev)
+[![PyPI](https://img.shields.io/pypi/v/modaic)](https://pypi.org/project/modaic/)
+
+
 # Modaic 🐙
-**Mod**ular **A**gent **I**nfrastructure **C**ollective, a Python framework for building AI agents with structured context management, database integration, and retrieval-augmented generation (RAG) capabilities.
+**Mod**ular **A**gent **I**nfrastructure **C**ollection, a Python framework for building AI agents with structured context management, database integration, and retrieval-augmented generation (RAG) capabilities.
 
 ## Overview
 
@@ -255,7 +259,6 @@ from modaic.databases import VectorDatabase, SQLDatabase
 from modaic.types import Indexer
 
 class TableRAGConfig(PrecompiledConfig):
-    agent_type = "TableRAGAgent"
     k_recall: int = 50
     k_rerank: int = 5
 

{modaic-0.1.1 → modaic-0.2.0}/README.md

@@ -1,6 +1,9 @@
 [![Docs](https://img.shields.io/badge/docs-available-brightgreen.svg)](https://docs.modaic.dev)
+[![PyPI](https://img.shields.io/pypi/v/modaic)](https://pypi.org/project/modaic/)
+
+
 # Modaic 🐙
-**Mod**ular **A**gent **I**nfrastructure **C**ollective, a Python framework for building AI agents with structured context management, database integration, and retrieval-augmented generation (RAG) capabilities.
+**Mod**ular **A**gent **I**nfrastructure **C**ollection, a Python framework for building AI agents with structured context management, database integration, and retrieval-augmented generation (RAG) capabilities.
 
 ## Overview
 
@@ -196,7 +199,6 @@ from modaic.databases import VectorDatabase, SQLDatabase
 from modaic.types import Indexer
 
 class TableRAGConfig(PrecompiledConfig):
-    agent_type = "TableRAGAgent"
     k_recall: int = 50
     k_rerank: int = 5
 

{modaic-0.1.1 → modaic-0.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "modaic"
-version = "0.1.1"
+version = "0.2.0"
 description = "Modular Agent Infrastructure Collective, a python framework for managing and sharing DSPy agents"
 authors = [{ name = "Tyrin", email = "tytodd@mit.edu" }, {name = "Farouk", email = "farouk@modaic.dev"}]
 readme = "README.md"
@@ -17,7 +17,8 @@ dependencies = [
     "langchain-core>=0.3.72",
     "langchain-text-splitters>=0.3.9",
     "more-itertools>=10.8.0",
-    "opik>=1.8.42",
+    "openpyxl>=3.1.5",
+    "opik==1.8.42",
     "pillow>=11.3.0",
     "pymilvus>=2.5.14",
     "sqlalchemy>=2.0.42",
@@ -34,7 +35,6 @@ dev = [
     "pdoc>=15.0.4",
     "pytest>=8.4.1",
     "ruff>=0.12.7",
-    "opik>=1.8.42",
     "matplotlib>=3.10.6",
 ]
 examples = [
@@ -61,7 +61,7 @@ exclude = [
 [tool.ruff]
 line-length = 120
 fix = true
-exclude = ["*scratchpad*.py"]
+exclude = ["*scratchpad*.py", "**/*.ipynb"]
 
 [tool.ruff.lint]
 select = ["E", "F", "I", "B", "T20", "ERA", "ANN", "N"]
@@ -85,5 +85,6 @@ members = [
     "tests/artifacts/test_repos/simple_repo_with_compile",
     "tests/artifacts/test_repos/nested_repo",
     "tests/artifacts/test_repos/nested_repo_2",
+    "tests/artifacts/test_repos/nested_repo_3",
     "tests/artifacts/test_repos/multi_module_repo",
 ]

{modaic-0.1.1 → modaic-0.2.0}/src/modaic/__init__.py

@@ -1,4 +1,4 @@
-from .auto_agent import AutoAgent, AutoConfig, AutoRetriever
+from .auto import AutoAgent, AutoConfig, AutoRetriever
 from .indexing import Embedder
 from .observability import Trackable, configure, track, track_modaic_obj
 from .precompiled import Indexer, PrecompiledAgent, PrecompiledConfig, Retriever
@@ -22,4 +22,16 @@ __all__ = [
     "Prop",
     "Value",
     "parse_modaic_filter",
+    "Condition",
 ]
+_configured = False
+
+
+def _auto_configure():
+    global _configured
+    if not _configured:
+        configure()
+        _configured = True
+
+
+_auto_configure()

modaic-0.1.1/src/modaic/auto_agent.py → modaic-0.2.0/src/modaic/auto.py

@@ -4,24 +4,38 @@ import os
 import sys
 from functools import lru_cache
 from pathlib import Path
-from typing import Literal, Optional, Type
+from typing import Callable, Literal, Optional, Type, TypedDict
 
-from .hub import load_repo
+from .hub import AGENTS_CACHE, load_repo
 from .precompiled import PrecompiledAgent, PrecompiledConfig, Retriever, is_local_path
 
 MODAIC_TOKEN = os.getenv("MODAIC_TOKEN")
 
 
-_REGISTRY = {}  # maps model_type string -> (ConfigCls, ModelCls)
+class RegisteredRepo(TypedDict, total=False):
+    AutoConfig: Type[PrecompiledConfig]
+    AutoAgent: Type[PrecompiledAgent]
+    AutoRetriever: Type[Retriever]
 
 
-def register(model_type: str, config_cls: Type[PrecompiledConfig], model_cls: Type[PrecompiledAgent]):
-    _REGISTRY[model_type] = (config_cls, model_cls)
+_REGISTRY: dict[str, RegisteredRepo] = {}
 
 
+def register(
+    name: str,
+    auto_type: Literal["AutoConfig", "AutoAgent", "AutoRetriever"],
+    cls: Type[PrecompiledConfig | PrecompiledAgent | Retriever],
+):
+    if name in _REGISTRY:
+        _REGISTRY[name][auto_type] = cls
+    else:
+        _REGISTRY[name] = {auto_type: cls}
+
+
+# TODO: Cleanup code still using parent_mdoule
 @lru_cache
 def _load_dynamic_class(
-    repo_dir: str, class_path: str, parent_module: Optional[str] = None
+    repo_dir: Path, class_path: str, hub_path: str = None
 ) -> Type[PrecompiledConfig | PrecompiledAgent | Retriever]:
     """
     Load a class from a given repository directory and fully qualified class path.
@@ -29,27 +43,24 @@ def _load_dynamic_class(
     Args:
       repo_dir: Absolute path to a local repository directory containing the code.
       class_path: Dotted path to the target class (e.g., "pkg.module.Class").
-      parent_module: Optional dotted module prefix (e.g., "swagginty.TableRAG"). If provided,
-                     class_path is treated as relative to this module and only the agents cache
-                     root is added to sys.path.
+      hub_path: The path to the repo on modaic hub (if its a hub repo) *Must be specified if its a hub repo*
 
     Returns:
       The resolved class object.
     """
-
-    repo_path = Path(repo_dir)
-
-    repo_dir_str = str(repo_path)
-    print(f"repo_dir_str: {repo_dir_str}")
-    print(f"sys.path: {sys.path}")
-    if repo_dir_str not in sys.path:
-        # print(f"Inserting {repo_dir_str} into sys.path")
-        sys.path.insert(0, repo_dir_str)
-    full_path = (
-        f"{parent_module}.{class_path}"
-        if parent_module and not class_path.startswith(parent_module + ".")
-        else class_path
-    )
+    if hub_path is None:
+        # Local folder case
+        repo_dir_str = str(repo_dir)
+        if repo_dir_str not in sys.path:
+            sys.path.insert(0, repo_dir_str)
+        full_path = f"{class_path}"
+    else:
+        # loaded hub repo case
+        agents_cache_str = str(AGENTS_CACHE)
+        if agents_cache_str not in sys.path:
+            sys.path.insert(0, agents_cache_str)
+        parent_module = hub_path.replace("/", ".")
+        full_path = f"{parent_module}.{class_path}"
 
     module_name, _, attr = full_path.rpartition(".")
     module = importlib.import_module(module_name)
@@ -62,27 +73,31 @@ class AutoConfig:
     """
 
     @staticmethod
-    def from_precompiled(repo_path: str, *, parent_module: Optional[str] = None, **kwargs) -> PrecompiledConfig:
+    def from_precompiled(repo_path: str, **kwargs) -> PrecompiledConfig:
+        local = is_local_path(repo_path)
+        repo_dir = load_repo(repo_path, local)
+        return AutoConfig._from_precompiled(repo_dir, hub_path=repo_path if not local else None, **kwargs)
+
+    @staticmethod
+    def _from_precompiled(repo_dir: Path, hub_path: str = None, **kwargs) -> PrecompiledConfig:
         """
         Load a config for an agent or retriever from a precompiled repo.
 
         Args:
-          repo_path: Hub path ("user/repo") or a local directory.
-          parent_module: Optional dotted module prefix (e.g., "swagginty.TableRAG") to use to import classes from repo_path. If provided, overides default parent_module behavior.
+          repo_dir: The path to the repo directory. the loaded local repository directory.
+          hub_path: The path to the repo on modaic hub (if its a hub repo) *Must be specified if its a hub repo*
 
         Returns:
           A config object constructed via the resolved config class.
         """
-        local = is_local_path(repo_path)
-        repo_dir = load_repo(repo_path, local)
 
         cfg_path = repo_dir / "config.json"
         if not cfg_path.exists():
-            raise FileNotFoundError(f"Failed to load AutoConfig, config.json not found in {repo_path}")
+            raise FileNotFoundError(f"Failed to load AutoConfig, config.json not found in {hub_path or str(repo_dir)}")
         with open(cfg_path, "r") as fp:
             cfg = json.load(fp)
 
-        ConfigClass = _load_auto_class(repo_path, repo_dir, "AutoConfig", parent_module=parent_module)  # noqa: N806
+        ConfigClass = _load_auto_class(repo_dir, "AutoConfig", hub_path=hub_path)  # noqa: N806
         return ConfigClass(**{**cfg, **kwargs})
 
 
@@ -96,7 +111,6 @@ class AutoAgent:
         repo_path: str,
         *,
         config_options: Optional[dict] = None,
-        parent_module: Optional[str] = None,
         project: Optional[str] = None,
         **kw,
     ) -> PrecompiledAgent:
@@ -105,23 +119,25 @@ class AutoAgent:
 
         Args:
           repo_path: Hub path ("user/repo") or local directory.
-          parent_module: Optional dotted module prefix (e.g., "swagginty.TableRAG") to use to import classes from repo_path. If provided, overides default parent_module behavior.
           project: Optional project name. If not provided and repo_path is a hub path, defaults to the repo name.
           **kw: Additional keyword arguments forwarded to the Agent constructor.
 
         Returns:
           An instantiated Agent subclass.
         """
+        # TODO: fast lookups via registry
         local = is_local_path(repo_path)
         repo_dir = load_repo(repo_path, local)
+        hub_path = repo_path if not local else None
 
         if config_options is None:
             config_options = {}
 
-        cfg = AutoConfig.from_precompiled(repo_dir, local=True, parent_module=parent_module, **config_options)
-        AgentClass = _load_auto_class(repo_path, repo_dir, "AutoAgent", parent_module=parent_module)  # noqa: N806
+        cfg = AutoConfig._from_precompiled(repo_dir, hub_path=hub_path, **config_options)
+        AgentClass = _load_auto_class(repo_dir, "AutoAgent", hub_path=hub_path)  # noqa: N806
 
         # automatically configure repo and project from repo_path if not provided
+        # TODO: redundant checks in if statement. Investigate removing.
         if not local and "/" in repo_path and not repo_path.startswith("/"):
             parts = repo_path.split("/")
             if len(parts) >= 2:
@@ -146,7 +162,6 @@ class AutoRetriever:
         repo_path: str,
         *,
         config_options: Optional[dict] = None,
-        parent_module: Optional[str] = None,
         project: Optional[str] = None,
         **kw,
     ) -> Retriever:
@@ -155,7 +170,6 @@ class AutoRetriever:
 
         Args:
           repo_path: hub path ("user/repo"), or local directory.
-          parent_module: Optional dotted module prefix (e.g., "swagginty.TableRAG") to use to import classes from repo_path. If provided, overides default parent_module behavior.
           project: Optional project name. If not provided and repo_path is a hub path, defaults to the repo name.
           **kw: Additional keyword arguments forwarded to the Retriever constructor.
 
@@ -164,14 +178,16 @@ class AutoRetriever:
         """
         local = is_local_path(repo_path)
         repo_dir = load_repo(repo_path, local)
+        hub_path = repo_path if not local else None
 
         if config_options is None:
             config_options = {}
 
-        cfg = AutoConfig.from_precompiled(repo_dir, local=True, parent_module=parent_module, **config_options)
-        RetrieverClass = _load_auto_class(repo_path, repo_dir, "AutoRetriever", parent_module=parent_module)  # noqa: N806
+        cfg = AutoConfig._from_precompiled(repo_dir, hub_path=hub_path, **config_options)
+        RetrieverClass = _load_auto_class(repo_dir, "AutoRetriever", hub_path=hub_path)  # noqa: N806
 
         # automatically configure repo and project from repo_path if not provided
+        # TODO: redundant checks in if statement. Investigate removing.
         if not local and "/" in repo_path and not repo_path.startswith("/"):
             parts = repo_path.split("/")
             if len(parts) >= 2:
@@ -186,27 +202,25 @@ class AutoRetriever:
 
 
 def _load_auto_class(
-    repo_path: str,
     repo_dir: Path,
     auto_name: Literal["AutoConfig", "AutoAgent", "AutoRetriever"],
-    parent_module: Optional[str] = None,
+    hub_path: str = None,
 ) -> Type[PrecompiledConfig | PrecompiledAgent | Retriever]:
     """
     Load a class from the auto_classes.json file.
 
     Args:
-        repo_path: The path to the repo. (local or hub path)
         repo_dir: The path to the repo directory. the loaded local repository directory.
         auto_name: The name of the auto class to load. (AutoConfig, AutoAgent, AutoRetriever)
-        parent_module: The parent module to use to import the class.
+        hub_path: The path to the repo on modaic hub (if its a hub repo) *Must be specified if its a hub repo*
     """
     # determine if the repo was loaded from local or hub
-    local = is_local_path(repo_path)
+    local = hub_path is None
     auto_classes_path = repo_dir / "auto_classes.json"
 
     if not auto_classes_path.exists():
         raise FileNotFoundError(
-            f"Failed to load {auto_name}, auto_classes.json not found in {repo_path}, if this is your repo, make sure you push_to_hub() with `with_code=True`"
+            f"Failed to load {auto_name}, auto_classes.json not found in {hub_path or str(repo_dir)}, if this is your repo, make sure you push_to_hub() with `with_code=True`"
         )
 
     with open(auto_classes_path, "r") as fp:
@@ -214,15 +228,33 @@ def _load_auto_class(
 
     if not (auto_class_path := auto_classes.get(auto_name)):
         raise KeyError(
-            f"{auto_name} not found in {repo_path}/auto_classes.json. Please check that the auto_classes.json file is correct."
+            f"{auto_name} not found in {hub_path or str(repo_dir)}/auto_classes.json. Please check that the auto_classes.json file is correct."
         ) from None
 
-    if auto_class_path in _REGISTRY:
-        _, LoadedClass = _REGISTRY[auto_class_path]  # noqa: N806
-    else:
-        if parent_module is None and not local:
-            parent_module = str(repo_path).replace("/", ".")
-
-        repo_dir = repo_dir.parent.parent if not local else repo_dir
-        LoadedClass = _load_dynamic_class(repo_dir, auto_class_path, parent_module=parent_module)  # noqa: N806
+    repo_dir = repo_dir.parent.parent if not local else repo_dir
+    LoadedClass = _load_dynamic_class(repo_dir, auto_class_path, hub_path=hub_path)  # noqa: N806
     return LoadedClass
+
+
+def builtin_agent(name: str) -> Callable[[Type], Type]:
+    def _wrap(cls: Type) -> Type:
+        register(name, "AutoAgent", cls)
+        return cls
+
+    return _wrap
+
+
+def builtin_indexer(name: str) -> Callable[[Type], Type]:
+    def _wrap(cls: Type) -> Type:
+        register(name, "AutoRetriever", cls)
+        return cls
+
+    return _wrap
+
+
+def builtin_config(name: str) -> Callable[[Type], Type]:
+    def _wrap(cls: Type) -> Type:
+        register(name, "AutoConfig", cls)
+        return cls
+
+    return _wrap

{modaic-0.1.1 → modaic-0.2.0}/src/modaic/context/__init__.py

@@ -12,7 +12,7 @@ from .table import (
     Table,
     TableFile,
 )
-from .text import Text
+from .text import Text, TextFile
 
 __all__ = [
     "MultiTabbedTable",
@@ -31,4 +31,5 @@ __all__ = [
     "Prop",
     "HydratedAttr",
     "requires_hydration",
+    "TextFile",
 ]

{modaic-0.1.1 → modaic-0.2.0}/src/modaic/context/base.py

@@ -22,7 +22,7 @@ from pydantic._internal._model_construction import ModelMetaclass
 from pydantic.fields import ModelPrivateAttr
 from pydantic.main import IncEx
 from pydantic.v1 import Field as V1Field
-from pydantic_core import CoreSchema, SchemaSerializer
+from pydantic_core import SchemaSerializer
 
 from ..query_language import Prop
 from ..storage.file_store import FileStore

{modaic-0.1.1 → modaic-0.2.0}/src/modaic/context/table.py

@@ -38,7 +38,7 @@ class BaseTable(Context, ABC):
         """
         Return up to 3 distinct sample values from the given column.
 
-        Picks at most three unique, non-null, short (<64 chars) values from
+        Picks at most three unique, non-null, short (&lt;64 chars) values from
         the column, favoring speed by sampling after de-duplicating values.
 
         Args:

{modaic-0.1.1 → modaic-0.2.0}/src/modaic/context/text.py

@@ -47,7 +47,7 @@ class TextFile(Context):
     file_type: Literal["txt"] = "txt"
 
     def hydrate(self, file_store: FileStore) -> None:
-        file = file_store.get(self.file_ref)
+        file = file_store.get(self.file_ref).file
         if isinstance(file, Path):
             file = file.read_text()
         else:
@@ -91,4 +91,4 @@ class TextFile(Context):
                 chunks.append(Text(text=chunk))
             return chunks
 
-        self.apply_to_chunks(chunk_text_fn)
+        self.chunk_with(chunk_text_fn)

{modaic-0.1.1 → modaic-0.2.0}/src/modaic/databases/graph_database.py

@@ -12,12 +12,13 @@ from typing import (
     Type,
 )
 
-from dotenv import load_dotenv
+from dotenv import find_dotenv, load_dotenv
 
 from ..context.base import Context, Relation
 from ..observability import Trackable, track_modaic_obj
 
-load_dotenv()
+env_file = find_dotenv(usecwd=True)
+load_dotenv(env_file)
 
 
 if TYPE_CHECKING:

modaic-0.2.0/src/modaic/datasets.py

@@ -0,0 +1,22 @@
+import dspy
+
+
+class Dataset:
+    def __init__(self, data: list):
+        self.data = data
+
+    def to_dspy(self) -> list:
+        return dspy.Dataset(self.data)
+
+    @classmethod
+    def from_csv(cls, file_path: str) -> "Dataset":
+        with open(file_path, "r") as file:
+            data = file.read()
+        return cls(data)
+
+    @classmethod
+    def from_hub(cls, dataset_name: str) -> "Dataset":
+        from datasets import load_dataset
+
+        data = load_dataset(dataset_name)
+        return cls(data)

{modaic-0.1.1 → modaic-0.2.0}/src/modaic/hub.py

@@ -1,15 +1,17 @@
 import os
+import shutil
 from pathlib import Path
 from typing import Any, Dict, Optional
 
 import git
 import requests
-from dotenv import load_dotenv
+from dotenv import find_dotenv, load_dotenv
 
 from .exceptions import AuthenticationError, RepositoryExistsError, RepositoryNotFoundError
 from .utils import compute_cache_dir
 
-load_dotenv()
+env_file = find_dotenv(usecwd=True)
+load_dotenv(env_file)
 
 MODAIC_TOKEN = os.getenv("MODAIC_TOKEN")
 MODAIC_GIT_URL = os.getenv("MODAIC_GIT_URL", "git.modaic.dev").replace("https://", "").rstrip("/")
@@ -80,6 +82,7 @@ def create_remote_repo(repo_path: str, access_token: str, exist_ok: bool = False
         raise Exception(f"Request failed: {str(e)}") from e
 
 
+# FIXME: make faster. Currently takes ~9 seconds
 def push_folder_to_hub(
     folder: str,
     repo_path: str,
@@ -123,10 +126,11 @@ def push_folder_to_hub(
             "Modaic fast paths not yet implemented. Please load agents with 'user/repo' or 'org/repo' format"
         )
     assert repo_path.count("/") <= 1, f"Extra '/' in repo_path: {repo_path}"
-
+    # TODO: try pushing first and on error create the repo. create_remote_repo currently takes ~1.5 seconds to run
     create_remote_repo(repo_path, access_token, exist_ok=True)
     username = get_user_info(access_token)["login"]
 
+    # FIXME: takes 6 seconds
     try:
         # 1) If local folder is not a git repository, initialize it.
         local_repo = git.Repo.init(folder)
@@ -205,6 +209,7 @@ def get_repo_payload(repo_name: str) -> Dict[str, Any]:
     return payload
 
 
+# TODO: add persistent filesystem based cache mapping access_token to user_info. Currently takes ~1 second
 def get_user_info(access_token: str) -> Dict[str, Any]:
     """
     Returns the user info for the given access token.
@@ -214,12 +219,14 @@ def get_user_info(access_token: str) -> Dict[str, Any]:
         access_token: The access token to get the user info for.
 
     Returns:
+    ```python
         {
             "login": str,
             "email": str,
             "avatar_url": str,
             "name": str,
         }
+    ```
     """
     global user_info
     if user_info:
@@ -243,6 +250,7 @@ def get_user_info(access_token: str) -> Dict[str, Any]:
     return user_info
 
 
+# TODO:
 def git_snapshot(
     repo_path: str,
     *,
@@ -265,7 +273,6 @@ def git_snapshot(
     elif access_token is None:
         raise ValueError("Access token is required")
 
-    # If a local folder path is provided, just return it
     repo_dir = Path(AGENTS_CACHE) / repo_path
     username = get_user_info(access_token)["login"]
     try:
@@ -291,10 +298,26 @@ def git_snapshot(
         repo.git.reset("--hard", f"origin/{target}")
         return repo_dir
     except Exception as e:
-        repo_dir.rmdir()
+        shutil.rmtree(repo_dir)
         raise e
 
 
+def _move_to_commit_sha_folder(repo: git.Repo) -> git.Repo:
+    """
+    Moves the repo to a new path based on the commit SHA. (Unused for now)
+    Args:
+        repo: The git.Repo object.
+
+    Returns:
+        The new git.Repo object.
+    """
+    commit = repo.head.commit
+    repo_dir = Path(repo.working_dir)
+    new_path = repo_dir / commit.hexsha
+    repo_dir.rename(new_path)
+    return git.Repo(new_path)
+
+
 def load_repo(repo_path: str, is_local: bool = False) -> Path:
     if is_local:
         path = Path(repo_path)

{modaic-0.1.1 → modaic-0.2.0}/src/modaic/indexing.py

@@ -77,7 +77,7 @@ class PineconeReranker(Reranker):
         try:
             from pinecone import Pinecone
         except ImportError:
-            raise ImportError("Pinecone is not installed. Please install it with `uv add pinecone`")
+            raise ImportError("Pinecone is not installed. Please install it with `uv add pinecone`") from None
 
         if api_key is None:
             self.pinecone = Pinecone(os.getenv("PINECONE_API_KEY"))