noesium 0.1.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

noesium/core/__init__.py

@@ -1,4 +1,4 @@
 from noesium.core.utils.logging import setup_logging
 
-# Enable colorful logging by default for cogents
+# Enable colorful logging by default for noesium
 setup_logging(level="INFO", enable_colors=True)

noesium/core/agent/base.py

@@ -3,10 +3,18 @@ from abc import ABC, abstractmethod
 from datetime import datetime
 from typing import Any, Dict, List, Optional, Type
 
-from langchain_core.runnables import RunnableConfig
-from langgraph.graph import StateGraph
 from pydantic import BaseModel, Field
 
+try:
+    from langchain_core.runnables import RunnableConfig
+    from langgraph.graph import StateGraph
+
+    LANGCHAIN_AVAILABLE = True
+except ImportError:
+    RunnableConfig = None
+    StateGraph = None
+    LANGCHAIN_AVAILABLE = False
+
 from noesium.core.llm import get_llm_client
 from noesium.core.tracing import get_token_tracker
 from noesium.core.utils.logging import get_logger

noesium/core/goalith/decomposer/llm_decomposer.py

@@ -77,7 +77,7 @@ class LLMDecomposer(GoalDecomposer):
 
     def __init__(
         self,
-        provider: str = os.getenv("COGENTS_LLM_PROVIDER", "openrouter"),
+        provider: str = os.getenv("NOESIUM_LLM_PROVIDER", "openrouter"),
         model_name: Optional[str] = None,
         temperature: float = 0.3,
         max_tokens: int = 2000,

noesium/core/llm/__init__.py

@@ -55,7 +55,7 @@ __all__ = [
 
 
 def get_llm_client(
-    provider: str = os.getenv("COGENTS_LLM_PROVIDER", "openai"),
+    provider: str = os.getenv("NOESIUM_LLM_PROVIDER", "openai"),
     base_url: Optional[str] = None,
     api_key: Optional[str] = None,
     structured_output: bool = True,

noesium/core/llm/base.py

@@ -18,7 +18,7 @@ class BaseLLMClient(ABC):
         Args:
             **kwargs: Additional arguments to pass to the LLM client
         """
-        self.debug = os.getenv("COGENTS_DEBUG", "false").lower() == "true"
+        self.debug = os.getenv("NOESIUM_DEBUG", "false").lower() == "true"
 
     @abstractmethod
     def completion(
@@ -149,4 +149,4 @@ class BaseLLMClient(ABC):
         Returns:
             int: Expected embedding dimensions
         """
-        return int(os.getenv("COGENTS_EMBEDDING_DIMS", str(DEFAULT_EMBEDDING_DIMS)))
+        return int(os.getenv("NOESIUM_EMBEDDING_DIMS", str(DEFAULT_EMBEDDING_DIMS)))

noesium/core/llm/litellm.py

@@ -1,5 +1,5 @@
 """
-LiteLLM provider for Cogents.
+LiteLLM provider for Noesium.
 
 This module provides:
 - Unified interface to multiple LLM providers via LiteLLM
@@ -16,8 +16,23 @@ import time
 from pathlib import Path
 from typing import Any, Dict, List, Optional, Tuple, Type, TypeVar, Union
 
-import litellm
-from instructor import Instructor, Mode, patch
+try:
+    import litellm
+
+    LITELLM_AVAILABLE = True
+except ImportError:
+    litellm = None
+    LITELLM_AVAILABLE = False
+
+try:
+    from instructor import Instructor, Mode, patch
+
+    INSTRUCTOR_AVAILABLE = True
+except ImportError:
+    Instructor = None
+    Mode = None
+    patch = None
+    INSTRUCTOR_AVAILABLE = False
 
 from noesium.core.llm.base import BaseLLMClient
 from noesium.core.tracing import configure_opik, estimate_token_usage, get_token_tracker, is_opik_enabled
@@ -26,7 +41,7 @@ from noesium.core.utils.logging import get_logger
 # Only import OPIK if tracing is enabled
 OPIK_AVAILABLE = False
 track = lambda func: func  # Default no-op decorator
-if os.getenv("COGENTS_OPIK_TRACING", "false").lower() == "true":
+if os.getenv("NOESIUM_OPIK_TRACING", "false").lower() == "true":
     try:
         from opik import track
 
@@ -64,6 +79,9 @@ class LLMClient(BaseLLMClient):
             vision_model: Model to use for vision tasks (e.g., "gpt-4-vision-preview", "claude-3-sonnet")
             **kwargs: Additional arguments
         """
+        if not LITELLM_AVAILABLE:
+            raise ImportError("LiteLLM package is not installed. Install it with: pip install 'noesium[litellm]'")
+
         super().__init__(**kwargs)
         # Configure Opik tracing for observability only if enabled
         if OPIK_AVAILABLE:
@@ -90,22 +108,25 @@ class LLMClient(BaseLLMClient):
         # Initialize instructor if requested
         self.instructor = None
         if instructor:
-            try:
-                from openai import OpenAI
+            if not INSTRUCTOR_AVAILABLE:
+                logger.warning("Instructor package not available, structured completion will not work")
+            else:
+                try:
+                    from openai import OpenAI
 
-                # Create a mock client for instructor
-                mock_client = OpenAI(
-                    api_key="litellm",
-                    base_url="http://localhost:8000",  # LiteLLM proxy default
-                )
-                patched_client = patch(mock_client, mode=Mode.JSON)
-                self.instructor = Instructor(
-                    client=patched_client,
-                    create=patched_client.chat.completions.create,
-                    mode=Mode.JSON,
-                )
-            except ImportError:
-                logger.warning("OpenAI package not available, structured completion will not work")
+                    # Create a mock client for instructor
+                    mock_client = OpenAI(
+                        api_key="litellm",
+                        base_url="http://localhost:8000",  # LiteLLM proxy default
+                    )
+                    patched_client = patch(mock_client, mode=Mode.JSON)
+                    self.instructor = Instructor(
+                        client=patched_client,
+                        create=patched_client.chat.completions.create,
+                        mode=Mode.JSON,
+                    )
+                except ImportError:
+                    logger.warning("OpenAI package not available, structured completion will not work")
 
         # Configure LiteLLM settings
         litellm.drop_params = True  # Drop unsupported parameters
@@ -438,7 +459,7 @@ class LLMClient(BaseLLMClient):
             if len(embedding) != expected_dims:
                 logger.warning(
                     f"Embedding has {len(embedding)} dimensions, expected {expected_dims}. "
-                    f"Consider setting COGENTS_EMBEDDING_DIMS={len(embedding)} or "
+                    f"Consider setting NOESIUM_EMBEDDING_DIMS={len(embedding)} or "
                     f"using a different embedding model."
                 )
 
@@ -473,7 +494,7 @@ class LLMClient(BaseLLMClient):
                 if len(embedding) != expected_dims:
                     logger.warning(
                         f"Embedding at index {i} has {len(embedding)} dimensions, expected {expected_dims}. "
-                        f"Consider setting COGENTS_EMBEDDING_DIMS={len(embedding)} or "
+                        f"Consider setting NOESIUM_EMBEDDING_DIMS={len(embedding)} or "
                         f"using a different embedding model."
                     )
 

noesium/core/llm/llamacpp.py

@@ -3,8 +3,21 @@ import os
 from pathlib import Path
 from typing import Any, Dict, List, Optional, Tuple, Type, TypeVar, Union
 
-from huggingface_hub import snapshot_download
-from llama_cpp import Llama
+try:
+    from huggingface_hub import snapshot_download
+
+    HUGGINGFACE_HUB_AVAILABLE = True
+except ImportError:
+    snapshot_download = None
+    HUGGINGFACE_HUB_AVAILABLE = False
+
+try:
+    from llama_cpp import Llama
+
+    LLAMA_CPP_AVAILABLE = True
+except ImportError:
+    Llama = None
+    LLAMA_CPP_AVAILABLE = False
 
 from noesium.core.llm.base import BaseLLMClient
 from noesium.core.tracing import estimate_token_usage, get_token_tracker
@@ -14,7 +27,7 @@ from noesium.core.utils.logging import get_logger
 # Only import OPIK if tracing is enabled
 OPIK_AVAILABLE = False
 track = lambda func: func  # Default no-op decorator
-if os.getenv("COGENTS_OPIK_TRACING", "false").lower() == "true":
+if os.getenv("NOESIUM_OPIK_TRACING", "false").lower() == "true":
     try:
         from opik import track
 
@@ -39,6 +52,9 @@ def _download_default_model() -> str:
     Returns:
         Path to the downloaded model file
     """
+    if not HUGGINGFACE_HUB_AVAILABLE:
+        raise ImportError("huggingface-hub package is not installed. Install it with: pip install 'noesium[local-llm]'")
+
     try:
         logger.info(f"No model path provided, downloading default model: {DEFAULT_MODEL_REPO}")
 
@@ -101,6 +117,11 @@ class LLMClient(BaseLLMClient):
             n_gpu_layers: Number of layers to offload to GPU (-1 for all)
             **kwargs: Additional arguments to pass to Llama constructor
         """
+        if not LLAMA_CPP_AVAILABLE:
+            raise ImportError(
+                "llama-cpp-python package is not installed. Install it with: pip install 'noesium[local-llm]'"
+            )
+
         super().__init__(**kwargs)
         # Configure Opik tracing for observability only if enabled
         if OPIK_AVAILABLE:
@@ -411,7 +432,7 @@ class LLMClient(BaseLLMClient):
                 if len(embedding_vector) != expected_dims:
                     logger.warning(
                         f"Embedding has {len(embedding_vector)} dimensions, expected {expected_dims}. "
-                        f"Consider setting COGENTS_EMBEDDING_DIMS={len(embedding_vector)} or "
+                        f"Consider setting NOESIUM_EMBEDDING_DIMS={len(embedding_vector)} or "
                         f"using a different embedding model."
                     )
 

noesium/core/llm/ollama.py

@@ -1,5 +1,5 @@
 """
-Ollama LLM provider for Cogents.
+Ollama LLM provider for Noesium.
 
 This module provides:
 - Chat completion using Ollama models
@@ -13,8 +13,23 @@ import time
 from pathlib import Path
 from typing import Any, Dict, List, Optional, Tuple, Type, TypeVar, Union
 
-import ollama
-from instructor import Instructor, Mode, patch
+try:
+    import ollama
+
+    OLLAMA_AVAILABLE = True
+except ImportError:
+    ollama = None
+    OLLAMA_AVAILABLE = False
+
+try:
+    from instructor import Instructor, Mode, patch
+
+    INSTRUCTOR_AVAILABLE = True
+except ImportError:
+    Instructor = None
+    Mode = None
+    patch = None
+    INSTRUCTOR_AVAILABLE = False
 
 from noesium.core.llm.base import BaseLLMClient
 from noesium.core.tracing import estimate_token_usage, get_token_tracker
@@ -24,7 +39,7 @@ from noesium.core.utils.logging import get_logger
 # Only import OPIK if tracing is enabled
 OPIK_AVAILABLE = False
 track = lambda func: func  # Default no-op decorator
-if os.getenv("COGENTS_OPIK_TRACING", "false").lower() == "true":
+if os.getenv("NOESIUM_OPIK_TRACING", "false").lower() == "true":
     try:
         from opik import track
 
@@ -62,6 +77,9 @@ class LLMClient(BaseLLMClient):
             vision_model: Model to use for vision tasks (defaults to gemma3:4b)
             **kwargs: Additional arguments
         """
+        if not OLLAMA_AVAILABLE:
+            raise ImportError("Ollama package is not installed. Install it with: pip install 'noesium[local-llm]'")
+
         super().__init__(**kwargs)
         # Configure Opik tracing for observability only if enabled
         if OPIK_AVAILABLE:
@@ -84,23 +102,26 @@ class LLMClient(BaseLLMClient):
         # Initialize instructor if requested
         self.instructor = None
         if instructor:
-            # Create a mock OpenAI-compatible client for instructor
-            try:
-                from openai import OpenAI
-
-                # Create a mock client that uses Ollama through OpenAI-compatible API
-                mock_client = OpenAI(
-                    base_url=f"{self.base_url}/v1",
-                    api_key="ollama",  # Ollama doesn't require real API key
-                )
-                patched_client = patch(mock_client, mode=Mode.JSON)
-                self.instructor = Instructor(
-                    client=patched_client,
-                    create=patched_client.chat.completions.create,
-                    mode=Mode.JSON,
-                )
-            except ImportError:
-                logger.warning("OpenAI package not available, structured completion will not work")
+            if not INSTRUCTOR_AVAILABLE:
+                logger.warning("Instructor package not available, structured completion will not work")
+            else:
+                # Create a mock OpenAI-compatible client for instructor
+                try:
+                    from openai import OpenAI
+
+                    # Create a mock client that uses Ollama through OpenAI-compatible API
+                    mock_client = OpenAI(
+                        base_url=f"{self.base_url}/v1",
+                        api_key="ollama",  # Ollama doesn't require real API key
+                    )
+                    patched_client = patch(mock_client, mode=Mode.JSON)
+                    self.instructor = Instructor(
+                        client=patched_client,
+                        create=patched_client.chat.completions.create,
+                        mode=Mode.JSON,
+                    )
+                except ImportError:
+                    logger.warning("OpenAI package not available, structured completion will not work")
 
     @track
     def completion(
@@ -394,7 +415,7 @@ class LLMClient(BaseLLMClient):
             if len(embedding) != expected_dims:
                 logger.warning(
                     f"Embedding has {len(embedding)} dimensions, expected {expected_dims}. "
-                    f"Consider setting COGENTS_EMBEDDING_DIMS={len(embedding)} or "
+                    f"Consider setting NOESIUM_EMBEDDING_DIMS={len(embedding)} or "
                     f"using a different embedding model."
                 )
 

noesium/core/llm/openai.py

@@ -16,8 +16,23 @@ from pathlib import Path
 from typing import Any, Dict, List, Optional, Tuple, Type, TypeVar, Union
 
 # Import instructor for structured output
-from instructor import Instructor, Mode, patch
-from openai import OpenAI
+try:
+    from instructor import Instructor, Mode, patch
+
+    INSTRUCTOR_AVAILABLE = True
+except ImportError:
+    Instructor = None
+    Mode = None
+    patch = None
+    INSTRUCTOR_AVAILABLE = False
+
+try:
+    from openai import OpenAI
+
+    OPENAI_AVAILABLE = True
+except ImportError:
+    OpenAI = None
+    OPENAI_AVAILABLE = False
 
 from noesium.core.llm.base import BaseLLMClient
 from noesium.core.tracing import (
@@ -33,7 +48,7 @@ from noesium.core.utils.logging import get_logger
 OPIK_AVAILABLE = False
 track = lambda func: func  # Default no-op decorator
 track_openai = lambda client: client  # Default no-op function
-if os.getenv("COGENTS_OPIK_TRACING", "false").lower() == "true":
+if os.getenv("NOESIUM_OPIK_TRACING", "false").lower() == "true":
     try:
         from opik import track
         from opik.integrations.openai import track_openai
@@ -73,6 +88,9 @@ class LLMClient(BaseLLMClient):
             embed_model: Model to use for embeddings (defaults to text-embedding-3-small)
             **kwargs: Additional arguments to pass to the LLM client
         """
+        if not OPENAI_AVAILABLE:
+            raise ImportError("OpenAI package is not installed. Install it with: pip install 'noesium[openai]'")
+
         super().__init__(**kwargs)
         # Configure Opik tracing for observability only if enabled
         if OPIK_AVAILABLE:
@@ -109,6 +127,8 @@ class LLMClient(BaseLLMClient):
         # Initialize instructor if requested
         self.instructor = None
         if instructor:
+            if not INSTRUCTOR_AVAILABLE:
+                raise ImportError("Instructor package is not installed. Install it with: pip install 'noesium[openai]'")
             # Create instructor instance for structured output
             patched_client = patch(self.client, mode=Mode.JSON)
             self.instructor = Instructor(
@@ -412,7 +432,7 @@ class LLMClient(BaseLLMClient):
             if len(embedding) != expected_dims:
                 logger.warning(
                     f"Embedding has {len(embedding)} dimensions, expected {expected_dims}. "
-                    f"Consider setting COGENTS_EMBEDDING_DIMS={len(embedding)} or "
+                    f"Consider setting NOESIUM_EMBEDDING_DIMS={len(embedding)} or "
                     f"using a different embedding model."
                 )
 
@@ -465,7 +485,7 @@ class LLMClient(BaseLLMClient):
                 if len(embedding) != expected_dims:
                     logger.warning(
                         f"Embedding at index {i} has {len(embedding)} dimensions, expected {expected_dims}. "
-                        f"Consider setting COGENTS_EMBEDDING_DIMS={len(embedding)} or "
+                        f"Consider setting NOESIUM_EMBEDDING_DIMS={len(embedding)} or "
                         f"using a different embedding model."
                     )
 

noesium/core/llm/openrouter.py

@@ -20,7 +20,7 @@ from noesium.core.utils.logging import get_logger
 # Only import OPIK if tracing is enabled
 OPIK_AVAILABLE = False
 track = lambda func: func  # Default no-op decorator
-if os.getenv("COGENTS_OPIK_TRACING", "false").lower() == "true":
+if os.getenv("NOESIUM_OPIK_TRACING", "false").lower() == "true":
     try:
         pass
 

noesium/core/toolify/base.py

@@ -9,13 +9,20 @@ import abc
 import asyncio
 from typing import Any, Callable, Dict, List, Optional, Union
 
-from langchain_core.tools import BaseTool, tool
-
 from noesium.core.llm import BaseLLMClient
 from noesium.core.utils.logging import get_logger
 
 from .config import ToolkitConfig
 
+try:
+    from langchain_core.tools import BaseTool, tool
+
+    LANGCHAIN_AVAILABLE = True
+except ImportError:
+    BaseTool = None
+    tool = None
+    LANGCHAIN_AVAILABLE = False
+
 try:
     import mcp.types as mcp_types
 

noesium/core/toolify/config.py

@@ -32,7 +32,7 @@ class ToolkitConfig(BaseModel):
     """Toolkit-specific configuration parameters"""
 
     # LLM Integration
-    llm_provider: str = Field(default_factory=lambda: os.getenv("COGENTS_LLM_PROVIDER", "openai"))
+    llm_provider: str = Field(default_factory=lambda: os.getenv("NOESIUM_LLM_PROVIDER", "openai"))
     """LLM provider to use (openrouter, openai, ollama, llamacpp, litellm)"""
 
     llm_model: Optional[str] = None
@@ -55,7 +55,7 @@ class ToolkitConfig(BaseModel):
     log_level: str = Field(default_factory=lambda: os.getenv("LOG_LEVEL", "INFO"))
     """Logging level for this toolkit"""
 
-    enable_tracing: bool = Field(default_factory=lambda: os.getenv("COGENTS_ENABLE_TRACING", "false").lower() == "true")
+    enable_tracing: bool = Field(default_factory=lambda: os.getenv("NOESIUM_ENABLE_TRACING", "false").lower() == "true")
     """Enable detailed tracing for debugging"""
 
     class Config:

noesium/core/toolify/registry.py

@@ -188,10 +188,21 @@ def _discover_builtin_toolkits():
     import pkgutil
     from pathlib import Path
 
-    # Get the toolkits directory
-    toolkits_dir = Path(__file__).parent / "toolkits"
-
-    if not toolkits_dir.exists():
+    # Get the toolkits directory - look in noesium.toolkits
+    # Try multiple possible locations
+    possible_toolkits_dirs = [
+        Path(__file__).parent.parent.parent / "toolkits",  # noesium/toolkits
+        Path(__file__).parent / "toolkits",  # noesium/core/toolify/toolkits
+    ]
+
+    toolkits_dir = None
+    for dir_path in possible_toolkits_dirs:
+        if dir_path.exists() and (dir_path / "__init__.py").exists():
+            toolkits_dir = dir_path
+            break
+
+    if not toolkits_dir:
+        logger.warning("No toolkits directory found")
         return
 
     # Import all toolkit modules
@@ -199,8 +210,13 @@ def _discover_builtin_toolkits():
         if module_info.name.startswith("_"):
             continue
 
-        try:
+        # Determine the correct module name based on the location
+        if "core/toolify/toolkits" in str(toolkits_dir):
             module_name = f"noesium.core.toolify.toolkits.{module_info.name}"
+        else:
+            module_name = f"noesium.toolkits.{module_info.name}"
+
+        try:
             importlib.import_module(module_name)
             logger.debug(f"Discovered toolkit module: {module_name}")
         except ImportError as e:

noesium/core/tracing/opik_tracing.py

@@ -17,7 +17,7 @@ def configure_opik() -> bool:
     Configure Opik tracing based on environment variables.
 
     Environment variables:
-        COGENTS_OPIK_TRACING: Global toggle for Opik tracing (default: false)
+        NOESIUM_OPIK_TRACING: Global toggle for Opik tracing (default: false)
         OPIK_USE_LOCAL: Use local Opik deployment (default: true)
         OPIK_LOCAL_URL: Local Opik URL (default: http://localhost:5173)
         OPIK_API_KEY: API key for Comet ML/Opik (only needed for cloud)
@@ -31,10 +31,10 @@ def configure_opik() -> bool:
     """
     try:
         # Check global Noesium Opik tracing toggle first
-        cogents_opik_enabled = os.getenv("COGENTS_OPIK_TRACING", "false").lower() == "true"
+        noesium_opik_enabled = os.getenv("NOESIUM_OPIK_TRACING", "false").lower() == "true"
 
-        if not cogents_opik_enabled:
-            logger.debug("Opik tracing disabled via COGENTS_OPIK_TRACING=false")
+        if not noesium_opik_enabled:
+            logger.debug("Opik tracing disabled via NOESIUM_OPIK_TRACING=false")
             return False
 
         # Check if using local deployment
@@ -43,7 +43,7 @@ def configure_opik() -> bool:
         # Configuration variables
         opik_api_key = os.getenv("OPIK_API_KEY")
         os.getenv("OPIK_WORKSPACE")
-        opik_project = os.getenv("OPIK_PROJECT_NAME", "cogents-llm")
+        opik_project = os.getenv("OPIK_PROJECT_NAME", "noesium-llm")
         opik_tracing = os.getenv("OPIK_TRACING", "true").lower() == "true"
 
         if not opik_tracing:
@@ -88,9 +88,9 @@ def is_opik_enabled() -> bool:
     """
     try:
         # Check global Noesium Opik tracing toggle first
-        cogents_opik_enabled = os.getenv("COGENTS_OPIK_TRACING", "false").lower() == "true"
+        noesium_opik_enabled = os.getenv("NOESIUM_OPIK_TRACING", "false").lower() == "true"
 
-        if not cogents_opik_enabled:
+        if not noesium_opik_enabled:
             return False
 
         import opik

noesium/core/vector_store/__init__.py

@@ -33,9 +33,9 @@ __all__ = [
 
 
 def get_vector_store(
-    provider: str = os.getenv("COGENTS_VECTOR_STORE_PROVIDER", "weaviate"),
+    provider: str = os.getenv("NOESIUM_VECTOR_STORE_PROVIDER", "weaviate"),
     collection_name: str = "default_collection",
-    embedding_model_dims: int = int(os.getenv("COGENTS_EMBEDDING_DIMS", "768")),
+    embedding_model_dims: int = int(os.getenv("NOESIUM_EMBEDDING_DIMS", "768")),
     **kwargs,
 ):
     """

noesium/core/vector_store/base.py

@@ -37,7 +37,7 @@ class BaseVectorStore(ABC):
                 raise ValueError(
                     f"Vector at index {i} has {len(vector)} dimensions, "
                     f"expected {self.embedding_model_dims}. "
-                    f"Check that your embedding model matches COGENTS_EMBEDDING_DIMS={self.embedding_model_dims}"
+                    f"Check that your embedding model matches NOESIUM_EMBEDDING_DIMS={self.embedding_model_dims}"
                 )
 
     @abstractmethod

noesium/core/vector_store/pgvector.py

@@ -6,7 +6,10 @@ try:
     import psycopg2
     from psycopg2.extras import execute_values
 except ImportError:
-    raise ImportError("The 'psycopg2' library is required. Please install it using 'pip install psycopg2'.")
+    raise ImportError(
+        "The 'psycopg2' library is required for PGVector support. "
+        "Install it with 'pip install noesium[postgres]' or 'pip install psycopg2-binary'."
+    )
 
 from .base import BaseVectorStore, OutputData
 
@@ -63,36 +66,30 @@ class PGVectorStore(BaseVectorStore):
             distance (str): Distance metric (not used in PGVector, for compatibility).
         """
         self.cur.execute("CREATE EXTENSION IF NOT EXISTS vector")
-        self.cur.execute(
-            f"""
+        self.cur.execute(f"""
             CREATE TABLE IF NOT EXISTS {self.collection_name} (
                 id UUID PRIMARY KEY,
                 vector vector({vector_size}),
                 payload JSONB
             );
-        """
-        )
+        """)
 
         if self.use_diskann and vector_size < 2000:
             # Check if vectorscale extension is installed
             self.cur.execute("SELECT * FROM pg_extension WHERE extname = 'vectorscale'")
             if self.cur.fetchone():
                 # Create DiskANN index if extension is installed for faster search
-                self.cur.execute(
-                    f"""
+                self.cur.execute(f"""
                     CREATE INDEX IF NOT EXISTS {self.collection_name}_diskann_idx
                     ON {self.collection_name}
                     USING diskann (vector);
-                """
-                )
+                """)
         elif self.use_hnsw:
-            self.cur.execute(
-                f"""
+            self.cur.execute(f"""
                 CREATE INDEX IF NOT EXISTS {self.collection_name}_hnsw_idx
                 ON {self.collection_name}
                 USING hnsw (vector vector_cosine_ops)
-            """
-            )
+            """)
 
         self.conn.commit()
 

noesium/core/vector_store/weaviate.py

@@ -6,7 +6,8 @@ try:
     import weaviate
 except ImportError:
     raise ImportError(
-        "The 'weaviate' library is required. Please install it using 'pip install weaviate-client weaviate'."
+        "The 'weaviate' library is required for Weaviate support. "
+        "Install it with 'pip install noesium[weaviate]' or 'pip install weaviate-client'."
     )
 
 import weaviate.classes.config as wvcc

noesium/toolkits/__init__.py

@@ -0,0 +1 @@
+# A collection of toolkits inheriting `noesium.core.toolify` module.