truefoundry 0.2.0rc3__tar.gz → 0.2.0rc4__tar.gz

{truefoundry-0.2.0rc3 → truefoundry-0.2.0rc4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.2.0rc3
+Version: 0.2.0rc4
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhichoudhary06@gmail.com

{truefoundry-0.2.0rc3 → truefoundry-0.2.0rc4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "truefoundry"
-version = "0.2.0rc3"
+version = "0.2.0rc4"
 description = "Truefoundry CLI"
 authors = ["Abhishek Choudhary <abhichoudhary06@gmail.com>"]
 readme = "README.md"

{truefoundry-0.2.0rc3 → truefoundry-0.2.0rc4}/truefoundry/autodeploy/cli.py

@@ -41,7 +41,7 @@ from truefoundry.autodeploy.tools.commit import CommitConfirmation
 from truefoundry.autodeploy.tools.docker_run import DockerRun, DockerRunLog
 
 
-def get_openai_client() -> OpenAI:
+def _get_openai_client() -> OpenAI:
     session = ServiceFoundrySession()
     if AUTODEPLOY_OPENAI_BASE_URL is not None and AUTODEPLOY_OPENAI_API_KEY is not None:
         return OpenAI(
@@ -190,11 +190,22 @@ def _get_default_project_name(project_root_path: str):
     return name
 
 
+def _get_docker(console: Console) -> docker.DockerClient:
+    try:
+        return docker.from_env()
+    except Exception:
+        console.print_exception(extra_lines=0, max_frames=1)
+        console.print(
+            "[bold red]Error:[/] Could not connect to Docker, please check whether the Docker daemon is running."
+        )
+        sys.exit(1)
+
+
 def cli(project_root_path: str, deploy: bool):
-    openai_client = get_openai_client()
-    docker_client = docker.from_env()
-    project_root_path = os.path.abspath(project_root_path)
     console = Console()
+    openai_client = _get_openai_client()
+    docker_client = _get_docker(console)
+    project_root_path = os.path.abspath(project_root_path)
     console.print(
         "[bold magenta]TFY-Agent[/]: A tool for building and deploying [magenta]Jobs/Services[/] to the Truefoundry platform."
     )