[CLEANUP]

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "swarms-cloud"
-version = "0.4.0"
+version = "0.4.1"
 description = "Swarms Cloud - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <[email protected]>"]
@@ -51,6 +51,7 @@ hf_transfer = "*"
 swarms = "*"
 rich = "*"
 asyncio = "*"
+pulumi = "*"
 
 [tool.poetry.scripts]
 swarms-cloud = "swarms_cloud.cli.main:main"

requirements.txt

@@ -22,6 +22,7 @@ huggingface-hub
 optimum
 auto-gptq
 whisperx
+pulumi
 shortuuid
 hf_transfer
 swarms

swarms_cloud/cli/main.py

@@ -44,6 +44,8 @@ def show_help():
     """
     )
 
+    # [bold white]add-agent[/bold white]   : Add an agent to the marketplace under your name. Must have a Dockerfile + your agent.yaml to publish. Learn more Here: https://docs.swarms.world/en/latest/swarms_cloud/vision/
+
 
 # Fetch API key from platform
 def get_api_key():
@@ -113,6 +115,8 @@ def main():
         check_login()
     elif args.command == "read-docs":
         redirect_to_docs()
+    # elif args.command == "add-agent":
+    #     deploy_agent()
     else:
         console.print("[bold red]Unknown command! Type 'help' for usage.[/bold red]")
 

swarms_cloud/structs/deploy_gcp.py

@@ -2,9 +2,9 @@
 from typing import Any, Dict
 
 import click
-import pulumi
 import yaml
 from loguru import logger
+import pulumi
 from pulumi_gcp import cloudrun
 
 # Configure logging
@@ -71,10 +71,11 @@ def build_and_push_docker_image(
     "--dockerfile-path", default="./Dockerfile", help="Path to the Dockerfile"
 )
 @click.option("--yaml-path", default="./agent.yaml", help="Path to the agent YAML file")
-def deploy_agent(
-    project_id: str, region: str, dockerfile_path: str, yaml_path: str
-) -> None:
+def deploy_agent(dockerfile_path: str, yaml_path: str) -> None:
     try:
+        project_id = os.getenv("GCP_PROJECT_ID")
+        region = os.getenv("GCP_REGION")
+
         # Step 1: Validate existence of Dockerfile and agent.yaml
         if not check_file_exists(dockerfile_path):
             logger.error("Dockerfile is missing, exiting.")
@@ -140,5 +141,5 @@ def deploy_agent(
         raise
 
 
-if __name__ == "__main__":
-    deploy_agent()
+# if __name__ == "__main__":
+#     deploy_agent()

swarms_cloud/utils/cleanup_terminal.py

@@ -1,8 +1,11 @@
 import warnings
 
+# Suppress all warnings
 
-# Suppress specific warnings
+
+# Suppress specific warnings (if you want to keep this)
 def cleanup_logs():
+    warnings.filterwarnings("ignore")
     warnings.filterwarnings(
         "ignore", category=UserWarning, module="pydantic._internal._fields"
     )