quantalogic 0.50.5__tar.gz → 0.50.6__tar.gz

{quantalogic-0.50.5 → quantalogic-0.50.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: quantalogic
-Version: 0.50.5
+Version: 0.50.6
 Summary: QuantaLogic ReAct Agents
 Author: Raphaël MANSUY
 Author-email: raphael.mansuy@gmail.com

{quantalogic-0.50.5 → quantalogic-0.50.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "quantalogic"
-version = "0.50.5"
+version = "0.50.6"
 description = "QuantaLogic ReAct Agents"
 authors = ["Raphaël MANSUY <raphael.mansuy@gmail.com>"]
 readme = "README.md"

{quantalogic-0.50.5 → quantalogic-0.50.6}/quantalogic/tools/__init__.py

@@ -33,8 +33,8 @@ class LazyLoader:
 _TOOL_IMPORTS = {
     "AgentTool": (".agent_tool", False),
     "ComposioTool": (".composio.composio", True),
-    "GenerateDatabaseReportTool": (".database.generate_database_report_tool", False),
-    "SQLQueryToolAdvanced": (".database.sql_query_tool_advanced", False),
+    "GenerateDatabaseReportTool": (".database.generate_database_report_tool", True),
+    "SQLQueryToolAdvanced": (".database.sql_query_tool_advanced", True),
     "MarkdownToDocxTool": (".document_tools.markdown_to_docx_tool", True),
     "MarkdownToEpubTool": (".document_tools.markdown_to_epub_tool", True),
     "MarkdownToHtmlTool": (".document_tools.markdown_to_html_tool", True),
@@ -76,7 +76,7 @@ _TOOL_IMPORTS = {
     "SearchDefinitionNames": (".search_definition_names", False),
     "SequenceTool": (".sequence_tool", False),
     "SerpApiSearchTool": (".serpapi_search_tool", False),
-    "SQLQueryTool": (".sql_query_tool", False),
+    "SQLQueryTool": (".sql_query_tool", True),
     "TaskCompleteTool": (".task_complete_tool", False),
     "Tool": (".tool", False),
     "ToolArgument": (".tool", False),

{quantalogic-0.50.5 → quantalogic-0.50.6}/quantalogic/tools/database/sql_query_tool_advanced.py

@@ -1,13 +1,27 @@
 """Tool for executing SQL queries and performing database operations with safety checks."""
 
+import importlib.util
 import json
 from enum import Enum
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Optional, TYPE_CHECKING
 
 from loguru import logger
 from pydantic import Field
-from sqlalchemy import create_engine, inspect, text
-from sqlalchemy.engine import Engine
+
+# Check if sqlalchemy is available
+SQLALCHEMY_AVAILABLE = False
+try:
+    spec = importlib.util.find_spec("sqlalchemy")
+    SQLALCHEMY_AVAILABLE = spec is not None
+    if SQLALCHEMY_AVAILABLE:
+        from sqlalchemy import create_engine, inspect, text
+        from sqlalchemy.engine import Engine
+except ImportError:
+    pass
+
+# Type checking imports
+if TYPE_CHECKING:
+    from sqlalchemy.engine import Engine
 
 from quantalogic.tools.tool import Tool, ToolArgument
 
@@ -72,6 +86,9 @@ class SQLQueryToolAdvanced(Tool):
 
     def __init__(self, **data):
         super().__init__(**data)
+        if not SQLALCHEMY_AVAILABLE:
+            raise ImportError("The 'sqlalchemy' package is required to use SQLQueryToolAdvanced. "
+                           "Install it with 'pip install sqlalchemy' or include it in your dependencies.")
         self._engine = create_engine(self.connection_string)
         logger.info(f"Initialized SQL tool with engine for {self.connection_string}")
 

{quantalogic-0.50.5 → quantalogic-0.50.6}/quantalogic/tools/sql_query_tool.py

@@ -1,10 +1,20 @@
 """Tool for executing SQL queries and returning paginated results in markdown format."""
 
-from typing import Any, Dict, List
+import importlib.util
+from typing import Any, Dict, List, TYPE_CHECKING
 
 from pydantic import Field, ValidationError
-from sqlalchemy import create_engine, text
-from sqlalchemy.exc import SQLAlchemyError
+
+# Check if sqlalchemy is available
+SQLALCHEMY_AVAILABLE = False
+try:
+    spec = importlib.util.find_spec("sqlalchemy")
+    SQLALCHEMY_AVAILABLE = spec is not None
+    if SQLALCHEMY_AVAILABLE:
+        from sqlalchemy import create_engine, text
+        from sqlalchemy.exc import SQLAlchemyError
+except ImportError:
+    pass
 
 from quantalogic.tools.tool import Tool, ToolArgument
 
@@ -63,7 +73,13 @@ class SQLQueryTool(Tool):
         Raises:
             ValueError: For invalid parameters or query errors
             RuntimeError: For database connection issues
+            ImportError: When sqlalchemy is not available
         """
+        # Check if required dependencies are available
+        if not SQLALCHEMY_AVAILABLE:
+            raise ImportError("The 'sqlalchemy' package is required to use SQLQueryTool. "
+                           "Install it with 'pip install sqlalchemy' or include it in your dependencies.")
+            
         try:
             # Convert and validate row numbers
             start = self._convert_row_number(start_row, "start_row")
@@ -105,12 +121,13 @@ class SQLQueryTool(Tool):
 
             return "\n".join(markdown)
 
-        except SQLAlchemyError as e:
-            raise ValueError(f"SQL Error: {str(e)}") from e
-        except ValidationError as e:
-            raise ValueError(f"Validation Error: {str(e)}") from e
         except Exception as e:
-            raise RuntimeError(f"Database Error: {str(e)}") from e
+            if SQLALCHEMY_AVAILABLE and isinstance(e, SQLAlchemyError):
+                raise ValueError(f"SQL Error: {str(e)}") from e
+            elif isinstance(e, ValidationError):
+                raise ValueError(f"Validation Error: {str(e)}") from e
+            else:
+                raise RuntimeError(f"Database Error: {str(e)}") from e
 
     def _convert_row_number(self, value: Any, field_name: str) -> int:
         """Convert and validate row number input."""

{quantalogic-0.50.5 → quantalogic-0.50.6}/quantalogic/tools/utils/create_sample_database.py

@@ -1,12 +1,62 @@
+import importlib.util
 import random
 from datetime import datetime, timedelta
-
-from faker import Faker
-from sqlalchemy import Column, Date, Float, ForeignKey, Integer, String, create_engine
-from sqlalchemy.orm import declarative_base, relationship, sessionmaker
-
-Base = declarative_base()
-fake = Faker()
+from typing import Optional
+
+# Check if faker is available
+FAKER_AVAILABLE = False
+try:
+    spec = importlib.util.find_spec("faker")
+    FAKER_AVAILABLE = spec is not None
+except ImportError:
+    pass
+
+# Check if sqlalchemy is available
+SQLALCHEMY_AVAILABLE = False
+try:
+    spec = importlib.util.find_spec("sqlalchemy")
+    SQLALCHEMY_AVAILABLE = spec is not None
+except ImportError:
+    pass
+
+# Only import if available
+if FAKER_AVAILABLE:
+    from faker import Faker
+else:
+    # Create a dummy Faker class
+    class Faker:
+        def __init__(self):
+            pass
+        
+        def name(self) -> str:
+            return "Sample Name"
+            
+        def email(self) -> str:
+            return "sample@example.com"
+            
+        def street_address(self) -> str:
+            return "123 Sample St"
+            
+        def city(self) -> str:
+            return "Sample City"
+            
+        def word(self) -> str:
+            return "sample"
+            
+        def date_between(self, start_date: datetime) -> datetime:
+            return start_date + timedelta(days=random.randint(1, 365))
+
+if SQLALCHEMY_AVAILABLE:
+    from sqlalchemy import Column, Date, Float, ForeignKey, Integer, String, create_engine
+    from sqlalchemy.orm import declarative_base, relationship, sessionmaker
+
+# Initialize these only if the required modules are available
+Base = None
+fake = None
+if SQLALCHEMY_AVAILABLE:
+    Base = declarative_base()
+if FAKER_AVAILABLE:
+    fake = Faker()
 
 
 def create_sample_database(db_path: str) -> None:
@@ -16,6 +66,13 @@ def create_sample_database(db_path: str) -> None:
     Args:
         db_path: Path to the SQLite database file (e.g., 'sample.db')
     """
+    # Check if required dependencies are available
+    if not SQLALCHEMY_AVAILABLE:
+        raise ImportError("The 'sqlalchemy' package is required to use create_sample_database. "
+                         "Install it with 'pip install sqlalchemy' or include it in your dependencies.")
+    if not FAKER_AVAILABLE:
+        raise ImportError("The 'faker' package is required to use create_sample_database. "
+                         "Install it with 'pip install faker' or include it in your dependencies.")
 
     # Define database schema
     class Customer(Base):

{quantalogic-0.50.5 → quantalogic-0.50.6}/quantalogic/tools/utils/generate_database_report.py

@@ -1,9 +1,34 @@
+import importlib.util
 from datetime import UTC, datetime
-from typing import Dict, List
-
-import networkx as nx
-from sqlalchemy import create_engine, inspect, text
-from sqlalchemy.engine import Inspector
+from typing import Any, Dict, List, Optional, TYPE_CHECKING
+
+# Check if networkx is available
+NETWORKX_AVAILABLE = False
+try:
+    spec = importlib.util.find_spec("networkx")
+    NETWORKX_AVAILABLE = spec is not None
+except ImportError:
+    pass
+
+# Check if sqlalchemy is available
+SQLALCHEMY_AVAILABLE = False
+try:
+    spec = importlib.util.find_spec("sqlalchemy")
+    SQLALCHEMY_AVAILABLE = spec is not None
+except ImportError:
+    pass
+
+# Only import if available
+if NETWORKX_AVAILABLE:
+    import networkx as nx
+if SQLALCHEMY_AVAILABLE:
+    from sqlalchemy import create_engine, inspect, text
+    from sqlalchemy.engine import Inspector
+
+# Type checking imports
+if TYPE_CHECKING:
+    import networkx as nx
+    from sqlalchemy.engine import Inspector
 
 
 def generate_database_report(connection_string: str) -> str:
@@ -16,6 +41,14 @@ def generate_database_report(connection_string: str) -> str:
     Returns:
         Markdown-formatted report as a string
     """
+    # Check if required dependencies are available
+    if not SQLALCHEMY_AVAILABLE:
+        raise ImportError("The 'sqlalchemy' package is required to use generate_database_report. "
+                         "Install it with 'pip install sqlalchemy' or include it in your dependencies.")
+    if not NETWORKX_AVAILABLE:
+        raise ImportError("The 'networkx' package is required to use generate_database_report. "
+                         "Install it with 'pip install networkx' or include it in your dependencies.")
+    
     # Setup database connection and inspection
     engine = create_engine(connection_string)
     inspector = inspect(engine)