quantalogic 0.50.5__py3-none-any.whl → 0.50.7__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/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.5.dist-info → quantalogic-0.50.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: quantalogic
-Version: 0.50.5
+Version: 0.50.7
 Summary: QuantaLogic ReAct Agents
 Author: Raphaël MANSUY
 Author-email: raphael.mansuy@gmail.com
@@ -9,6 +9,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: click (>=8.1.8,<9.0.0)
+Requires-Dist: faker (>=36.1.1,<37.0.0)
 Requires-Dist: instructor (>=1.7.2,<2.0.0)
 Requires-Dist: jinja2 (>=3.1.5,<4.0.0)
 Requires-Dist: litellm (>=1.56.4,<2.0.0)

{quantalogic-0.50.5.dist-info → quantalogic-0.50.7.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=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.5.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-quantalogic-0.50.5.dist-info/METADATA,sha256=CfJryWRRsudR8N1CYEcEpizlRo438fLxRTCCeVdIk9Q,22785
-quantalogic-0.50.5.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-quantalogic-0.50.5.dist-info/entry_points.txt,sha256=h74O_Q3qBRCrDR99qvwB4BpBGzASPUIjCfxHq6Qnups,183
-quantalogic-0.50.5.dist-info/RECORD,,
+quantalogic-0.50.7.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+quantalogic-0.50.7.dist-info/METADATA,sha256=TlYrGZR6hhDIMqndrpPcGLsBCJrbSty-J7s6iGmyhn8,22825
+quantalogic-0.50.7.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+quantalogic-0.50.7.dist-info/entry_points.txt,sha256=h74O_Q3qBRCrDR99qvwB4BpBGzASPUIjCfxHq6Qnups,183
+quantalogic-0.50.7.dist-info/RECORD,,