daita-agents 0.2.1__py3-none-any.whl → 0.2.3__py3-none-any.whl

daita/cli/main.py

@@ -59,8 +59,8 @@ def _import_core_functions():
         from .core.status import show_project_status
         from .core.logs import show_deployment_logs
         from .core.deployments import (
-            list_deployments, download_deployment, show_deployment_details,
-            rollback_deployment, delete_deployment
+            list_deployments, show_deployment_details,
+            delete_deployment
         )
         from .core.run import (
             run_remote_execution, list_remote_executions, get_execution_logs
@@ -73,9 +73,7 @@ def _import_core_functions():
             show_project_status,
             show_deployment_logs,
             list_deployments,
-            download_deployment,
             show_deployment_details,
-            rollback_deployment,
             delete_deployment,
             run_remote_execution,
             list_remote_executions,
@@ -138,11 +136,20 @@ def cli(ctx, verbose, quiet):
         log_level = logging.ERROR
     else:
         log_level = logging.INFO
-    
+
+    # Configure root logger with higher level to avoid noise from libraries
     logging.basicConfig(
-        level=log_level,
+        level=logging.WARNING,  # Only show warnings/errors from third-party libraries
         format="%(levelname)s: %(message)s"
     )
+
+    # Set our own loggers to the desired level
+    daita_logger = logging.getLogger('daita')
+    daita_logger.setLevel(log_level)
+
+    # Also set CLI logger
+    cli_logger = logging.getLogger(__name__)
+    cli_logger.setLevel(log_level)
     
     # Store options in context
     ctx.obj['verbose'] = verbose
@@ -323,28 +330,13 @@ def list_cmd(ctx, project_name, limit):
         click.echo(f" Error: {str(e)}", err=True)
         sys.exit(1)
 
-@deployments.command()
-@click.argument('deployment_id')
-@click.option('--output', '-o', help='Output file path')
-@click.pass_context
-def download(ctx, deployment_id, output):
-    """Download a deployment package."""
-    try:
-        download_deployment_func = ctx.obj['core_functions'][7]
-        asyncio.run(download_deployment_func(deployment_id, output))
-    except Exception as e:
-        if ctx.obj.get('verbose'):
-            logging.exception("Download deployment failed")
-        click.echo(f" Error: {str(e)}", err=True)
-        sys.exit(1)
-
 @deployments.command()
 @click.argument('deployment_id')
 @click.pass_context
 def show(ctx, deployment_id):
     """Show detailed deployment information."""
     try:
-        show_deployment_details_func = ctx.obj['core_functions'][8]
+        show_deployment_details_func = ctx.obj['core_functions'][7]
         asyncio.run(show_deployment_details_func(deployment_id))
     except Exception as e:
         if ctx.obj.get('verbose'):
@@ -352,20 +344,6 @@ def show(ctx, deployment_id):
         click.echo(f" Error: {str(e)}", err=True)
         sys.exit(1)
 
-@deployments.command()
-@click.argument('deployment_id')
-@click.pass_context
-def rollback(ctx, deployment_id):
-    """Rollback to a previous deployment."""
-    try:
-        rollback_deployment_func = ctx.obj['core_functions'][9]
-        asyncio.run(rollback_deployment_func(deployment_id, 'production'))
-    except Exception as e:
-        if ctx.obj.get('verbose'):
-            logging.exception("Rollback deployment failed")
-        click.echo(f" Error: {str(e)}", err=True)
-        sys.exit(1)
-
 @deployments.command()
 @click.argument('deployment_id')
 @click.option('--force', is_flag=True, help='Skip confirmation')
@@ -373,7 +351,7 @@ def rollback(ctx, deployment_id):
 def delete(ctx, deployment_id, force):
     """Delete a deployment and its Lambda functions."""
     try:
-        delete_deployment_func = ctx.obj['core_functions'][10]
+        delete_deployment_func = ctx.obj['core_functions'][8]
         asyncio.run(delete_deployment_func(deployment_id, force))
     except Exception as e:
         if ctx.obj.get('verbose'):
@@ -476,7 +454,7 @@ def run(ctx, target_name, target_type, data_file, data_json, task, follow, timeo
     """Execute an agent or workflow remotely in the cloud."""
     environment = 'production'  # Only production is supported
     try:
-        run_remote_execution_func = ctx.obj['core_functions'][11]
+        run_remote_execution_func = ctx.obj['core_functions'][9]
         success = asyncio.run(run_remote_execution_func(
             target_name=target_name,
             target_type=target_type,

daita/plugins/__init__.py

@@ -5,6 +5,7 @@ This module provides database, API, cloud storage, search, collaboration, and MC
 - PostgreSQL plugin for async database operations
 - MySQL plugin for async database operations
 - MongoDB plugin for async document database operations
+- Snowflake plugin for cloud data warehouse operations
 - REST API plugin for HTTP client functionality
 - AWS S3 plugin for cloud object storage operations
 - Slack plugin for team collaboration and notifications
@@ -16,13 +17,17 @@ without over-engineering.
 
 Usage:
     ```python
-    from daita.plugins import postgresql, mysql, mongodb, rest, s3, slack, elasticsearch, mcp
+    from daita.plugins import postgresql, mysql, mongodb, snowflake, rest, s3, slack, elasticsearch, mcp
     from daita import SubstrateAgent
 
     # Database plugins
     async with postgresql(host="localhost", database="mydb") as db:
         results = await db.query("SELECT * FROM users")
 
+    # Snowflake plugin
+    async with snowflake(account="xy12345", warehouse="COMPUTE_WH", database="MYDB", user="user", password="pass") as db:
+        results = await db.query("SELECT * FROM customers LIMIT 10")
+
     # REST API plugin
     async with rest(base_url="https://api.example.com") as api:
         data = await api.get("/users")
@@ -52,6 +57,7 @@ Usage:
 from .postgresql import PostgreSQLPlugin, postgresql
 from .mysql import MySQLPlugin, mysql
 from .mongodb import MongoDBPlugin, mongodb
+from .snowflake import SnowflakePlugin, snowflake
 
 # API plugins  
 from .rest import RESTPlugin, rest
@@ -90,7 +96,11 @@ class PluginAccess:
     def mongodb(self, **kwargs) -> MongoDBPlugin:
         """Create MongoDB plugin."""
         return mongodb(**kwargs)
-    
+
+    def snowflake(self, **kwargs) -> SnowflakePlugin:
+        """Create Snowflake plugin."""
+        return snowflake(**kwargs)
+
     def rest(self, **kwargs) -> RESTPlugin:
         """Create REST API plugin."""
         return rest(**kwargs)
@@ -117,6 +127,7 @@ __all__ = [
     'PostgreSQLPlugin',
     'MySQLPlugin',
     'MongoDBPlugin',
+    'SnowflakePlugin',
     'RESTPlugin',
     'S3Plugin',
     'SlackPlugin',
@@ -127,6 +138,7 @@ __all__ = [
     'postgresql',
     'mysql',
     'mongodb',
+    'snowflake',
     'rest',
     's3',
     'slack',