quantalogic 0.50.3__py3-none-any.whl → 0.50.6__py3-none-any.whl

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/tools/composio/composio.py

@@ -1,28 +1,31 @@
 """Tool for interacting with Composio API services."""
 
+import importlib
+import importlib.util
 import json
 import os
 import traceback
 from datetime import datetime
-from typing import Any, Dict, Optional, Union
+from typing import Any, Dict, Optional, Union, TYPE_CHECKING
+
+# Type checking imports - these are only used for type hints and not at runtime
+if TYPE_CHECKING:
+    from composio import ComposioToolSet
 
 from loguru import logger
 from pydantic import BaseModel, ConfigDict, Field
 
-# Try to import composio, but don't fail if it's not installed
+from quantalogic.tools.tool import Tool, ToolArgument
+
+# Flag to check if composio is available
+COMPOSIO_AVAILABLE = False
+
+# Try to check if composio is importable without actually importing it
 try:
-    from composio import Action, ComposioToolSet
-    COMPOSIO_AVAILABLE = True
+    spec = importlib.util.find_spec("composio")
+    COMPOSIO_AVAILABLE = spec is not None
 except ImportError:
-    logger.warning("composio module not installed. ComposioTool will not be available.")
-    COMPOSIO_AVAILABLE = False
-    # Create dummy classes to avoid errors
-    class Action:
-        pass
-    class ComposioToolSet:
-        pass
-
-from quantalogic.tools.tool import Tool, ToolArgument
+    pass
 
 
 def setup_logger():
@@ -84,7 +87,7 @@ class ComposioTool(Tool):
     )
     action: str = Field(default="")  # Single action per tool instance
     action_schema: Optional[ActionSchema] = None
-    toolset: Optional[ComposioToolSet] = None
+    toolset: Optional['ComposioToolSet'] = None
     
     # Performance tracking
     _last_execution_time: Optional[datetime] = None
@@ -113,6 +116,14 @@ class ComposioTool(Tool):
             logger.error("ComposioTool cannot be initialized: composio module is not installed")
             raise ImportError("The 'composio' package is required to use ComposioTool. "
                              "Install it with 'pip install composio' or include it in your dependencies.")
+        
+        # Import composio only when needed
+        try:
+            from composio import Action, ComposioToolSet
+        except ImportError:
+            logger.error("Failed to import composio module")
+            raise ImportError("The 'composio' package is required to use ComposioTool. "
+                             "Install it with 'pip install composio' or include it in your dependencies.")
             
         logger.info(f"Initializing ComposioTool for action: {action}")
         start_time = datetime.now()
@@ -168,6 +179,14 @@ class ComposioTool(Tool):
 
     def _setup_action(self) -> None:
         """Set up the Composio action with its schema and parameters."""
+        # Import composio only when needed
+        try:
+            from composio import Action, ComposioToolSet
+        except ImportError:
+            logger.error("Failed to import composio module")
+            raise ImportError("The 'composio' package is required to use ComposioTool. "
+                             "Install it with 'pip install composio' or include it in your dependencies.")
+            
         logger.debug(f"Setting up action: {self.action}")
         start_time = datetime.now()
         
@@ -363,6 +382,14 @@ class ComposioTool(Tool):
 
     def execute(self, **kwargs: Any) -> str:
         """Execute the Composio action with the given parameters."""
+        # Import composio only when needed
+        try:
+            from composio import Action, ComposioToolSet
+        except ImportError:
+            logger.error("Failed to import composio module")
+            raise ImportError("The 'composio' package is required to use ComposioTool. "
+                             "Install it with 'pip install composio' or include it in your dependencies.")
+        
         start_time = datetime.now()
         self._execution_count += 1
         self._last_execution_time = start_time

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/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/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/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)

{quantalogic-0.50.3.dist-info → quantalogic-0.50.6.dist-info}/METADATA

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

{quantalogic-0.50.3.dist-info → quantalogic-0.50.6.dist-info}/RECORD

@@ -38,11 +38,11 @@ quantalogic/server/templates/index.html,sha256=nDnXJoQEm1vXbhXtgaYk0G5VXj0wwzE6K
 quantalogic/task_file_reader.py,sha256=oPcB4vTxJ__Y8o7VVABIPOkVw3tGDMdQYwdK27PERlE,1440
 quantalogic/task_runner.py,sha256=c2QVGKZfHA0wZIBUvehpjMvtRaKZuhuYQerjIiCC9iY,10053
 quantalogic/tool_manager.py,sha256=vNA7aBKgdU3wpw_goom6i9rg_64pNZapNxvg4cUhhCI,6983
-quantalogic/tools/__init__.py,sha256=Hpgt8RslYAwAfaKYunTFz_tsNdIZUXH5ovssNZu0H0c,5142
+quantalogic/tools/__init__.py,sha256=0ZwUnszv8GFW_Ml56ih_V0ilUkALmRfTK22vengmP5s,5139
 quantalogic/tools/agent_tool.py,sha256=MXCXxWHRch7VK4UWhtRP1jeI8Np9Ne2CUGo8vm1oZiM,3064
-quantalogic/tools/composio/composio.py,sha256=1FJ8lyZUslQpewX7WUF-sanF0f52gmL0-ZQFBXkiqaE,18373
+quantalogic/tools/composio/composio.py,sha256=l12notWc2okQYq_vrXRirLZyfc4WiTwY1AyLP96qidQ,19657
 quantalogic/tools/database/generate_database_report_tool.py,sha256=lGgER2VuHqnxliPM806tbwJh7WRW9HJcbBLL7QMvVBQ,1861
-quantalogic/tools/database/sql_query_tool_advanced.py,sha256=Vale-5nR3e-Mm4joOBII8T1PbLcshXYbM8Mp4TJsS30,9425
+quantalogic/tools/database/sql_query_tool_advanced.py,sha256=kOe0u24r1GIsBjs3glpEgZtkP6H7FGJ6Iz9k9gILZIc,10027
 quantalogic/tools/document_tools/markdown_to_docx_tool.py,sha256=r5AOyQfpcDoqQ6_g2uYyVjTEPtfeNn4mDmmefNZmJyM,22553
 quantalogic/tools/document_tools/markdown_to_epub_tool.py,sha256=iFIUTty7bIuOV7nbSfCuzBvOWJBmSP1TBFxf74184X4,14031
 quantalogic/tools/document_tools/markdown_to_html_tool.py,sha256=ngwU3RAj05Ghua2yqmpkpws09XYfjDpcg4MuONHNvO8,12030
@@ -109,7 +109,7 @@ quantalogic/tools/safe_python_interpreter_tool.py,sha256=yibIVmtkcfU4dS6KNcbJPxE
 quantalogic/tools/search_definition_names.py,sha256=zqtaqq8aS5jdDQOkbd4wMUPFyL6Bq-4q9NWyLKdXL1E,18771
 quantalogic/tools/sequence_tool.py,sha256=Hb2FSjWWACvXZX7rmJXPk5lnnnqaDeRTbhQQRtCd8hI,11169
 quantalogic/tools/serpapi_search_tool.py,sha256=sX-Noch77kGP2XiwislPNFyy3_4TH6TwMK6C81L3q9Y,5316
-quantalogic/tools/sql_query_tool.py,sha256=jEDZLlxOsB2bzsWlEqsqvTKiyovnRuk0XvgtwW7-WSQ,6055
+quantalogic/tools/sql_query_tool.py,sha256=CWDNdqHJDWfmL_XSMWPbQPl-RlIu6y_FuAYkNavrg2A,6768
 quantalogic/tools/task_complete_tool.py,sha256=L8tuyVoN07Q2hOsxx17JTW0C5Jd_N-C0i_0PtCUQUKU,929
 quantalogic/tools/tool.py,sha256=2XlveQf7NVuDlPz-IgE0gURye8rIp9iH3YhWhoOsnog,11232
 quantalogic/tools/unified_diff_tool.py,sha256=o7OiYnCM5MjbPlQTpB2OmkMQRI9zjdToQmgVkhiTvOI,14148
@@ -117,8 +117,8 @@ quantalogic/tools/utilities/csv_processor_tool.py,sha256=Mu_EPVj6iYAclNaVX_vbkek
 quantalogic/tools/utilities/download_file_tool.py,sha256=hw_tO6RD0tA_LH46Tathf-LzSxRl7kgEiiP76rTGdds,6616
 quantalogic/tools/utilities/mermaid_validator_tool.py,sha256=Brd6pt8OxpUgf8dm6kHqJJs_IU8V4Kc-8VDP-n1eCUM,25886
 quantalogic/tools/utils/__init__.py,sha256=-NtMSwxRt_G79Oo_DcDaCdLU2vLvuXIoCd34TOYEUmI,334
-quantalogic/tools/utils/create_sample_database.py,sha256=h5c_uxv3eztQvHlloTZxzWt5gEzai8zfnR8-_QrUqxU,3724
-quantalogic/tools/utils/generate_database_report.py,sha256=3PT34ClMvZ2O62-24cp_5lOyZHY_pBjVObMHpfyVi-s,10140
+quantalogic/tools/utils/create_sample_database.py,sha256=xc0I1oWr_XYSQ79Ar8Mpr93dqooQOxtEXZPyoXbv8kM,5517
+quantalogic/tools/utils/generate_database_report.py,sha256=MbImXDX2j7KRQrCo6kvW-WZRnhrQQjVGQPFVhI25uZE,11294
 quantalogic/tools/wikipedia_search_tool.py,sha256=LXQSPH8961Efw2QNxKe-cD5ZiIYD3ufEgrxH4y5uB74,5180
 quantalogic/tools/write_file_tool.py,sha256=_mx9_Zjg2oMAAVzlcHEKjZVZUxQVgbRfcoMKgWnoZcg,3764
 quantalogic/utils/__init__.py,sha256=hsS3hXH5lsBQcZh2QBANY1Af2Zs1jtrgxA7kXJEWi58,680
@@ -141,8 +141,8 @@ quantalogic/version_check.py,sha256=JyQFTNMDWtpHCLnN-BiakzB2cyXf6kUFsTjvmSruZi4,
 quantalogic/welcome_message.py,sha256=o4tHdgabNuIV9kbIDPgS3_2yzJhayK30oKad2UouYDc,3020
 quantalogic/xml_parser.py,sha256=AKuMdJC3QAX3Z_tErHVlZ-msjPemWaZUFiTwkHz76jg,11614
 quantalogic/xml_tool_parser.py,sha256=Vz4LEgDbelJynD1siLOVkJ3gLlfHsUk65_gCwbYJyGc,3784
-quantalogic-0.50.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-quantalogic-0.50.3.dist-info/METADATA,sha256=8yPjleuhCSPRiv7lwPRB5vObHFWMyrU0Ih7A4W2PA1Y,22785
-quantalogic-0.50.3.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-quantalogic-0.50.3.dist-info/entry_points.txt,sha256=h74O_Q3qBRCrDR99qvwB4BpBGzASPUIjCfxHq6Qnups,183
-quantalogic-0.50.3.dist-info/RECORD,,
+quantalogic-0.50.6.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+quantalogic-0.50.6.dist-info/METADATA,sha256=eVBh6BPUPZPaaeJ9b8JQVdk0ntzzAeXAajF-l5FAAn0,22785
+quantalogic-0.50.6.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+quantalogic-0.50.6.dist-info/entry_points.txt,sha256=h74O_Q3qBRCrDR99qvwB4BpBGzASPUIjCfxHq6Qnups,183
+quantalogic-0.50.6.dist-info/RECORD,,