truefoundry 0.2.0rc3__tar.gz → 0.2.0rc5__tar.gz

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

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

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

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

{truefoundry-0.2.0rc3 → truefoundry-0.2.0rc5}/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,17 +190,41 @@ 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."
     )
-    console.print(
-        "[bold reverse red]DISCLAIMER:[/] The contents of the project will be sent to OpenAI."
-    )
+    if AUTODEPLOY_OPENAI_BASE_URL is not None and AUTODEPLOY_OPENAI_API_KEY is not None:
+        console.print(
+            "[bold green]OpenAI credentials found in environment variables.[/]"
+        )
+        console.print(
+            "[bold reverse red]DISCLAIMER:[/] The contents of your project will be sent to OpenAI.",
+            "This operation will use tokens from your provided OpenAI account and may incur costs.",
+        )
+    else:
+        console.print(
+            "[bold reverse red]DISCLAIMER:[/] The contents of the project will be sent to OpenAI."
+        )
+        console.print(
+            "[dim]To use your own LLM, set the environment variables [dim italic green]AUTODEPLOY_OPENAI_BASE_URL[/],[/]",
+            "[dim][dim italic green]AUTODEPLOY_OPENAI_API_KEY[/], and [dim italic green]AUTODEPLOY_MODEL_NAME[/] for URL, API key, and LLM model name respectively.[/]",
+        )
     console.print(
         "[bold cyan]Note:[/] All changes will be committed to a new branch. Please ensure you have a repository."
     )

truefoundry-0.2.0rc5/truefoundry/deploy/__init__.py

@@ -0,0 +1 @@
+from servicefoundry import *  # noqa: F403

truefoundry-0.2.0rc5/truefoundry/langchain/__init__.py

@@ -0,0 +1 @@
+from servicefoundry.langchain import *  # noqa: F403

truefoundry-0.2.0rc5/truefoundry/ml/__init__.py

@@ -0,0 +1,6 @@
+try:
+    from mlfoundry import *  # noqa: F403
+except ImportError:
+    print(
+        "To use metadata/artifact logging features, please run 'pip install truefoundry[ml]'."
+    )

truefoundry-0.2.0rc3/truefoundry/deploy/__init__.py

@@ -1 +0,0 @@
-from servicefoundry import *

truefoundry-0.2.0rc3/truefoundry/langchain/__init__.py

@@ -1 +0,0 @@
-from servicefoundry.langchain import *

truefoundry-0.2.0rc3/truefoundry/ml/__init__.py

@@ -1,4 +0,0 @@
-try:
-    from mlfoundry import *
-except ImportError:
-    print("To use metadata/artifact logging features, please run 'pip install truefoundry[ml]'.")