truefoundry 0.2.0rc3__py3-none-any.whl → 0.2.0rc5__py3-none-any.whl

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/deploy/__init__.py

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

truefoundry/langchain/__init__.py

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

truefoundry/ml/__init__.py

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

{truefoundry-0.2.0rc3.dist-info → truefoundry-0.2.0rc5.dist-info}/METADATA

@@ -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.dist-info → truefoundry-0.2.0rc5.dist-info}/RECORD

@@ -5,7 +5,7 @@ truefoundry/autodeploy/agents/base.py,sha256=woCry229x7WX0IZmBopW_e_NB76aCLQRLXZ
 truefoundry/autodeploy/agents/developer.py,sha256=u1BfjaMDX6Ad32vWYOEFZPWFK33ux0Ro8JztrRAFviY,4126
 truefoundry/autodeploy/agents/project_identifier.py,sha256=m8xg7U5D2zSLEKp5QyPZRldw4VPuTFM8omPCRC0Ri-w,4428
 truefoundry/autodeploy/agents/tester.py,sha256=iSnWHubzHPO36Je2nYQYsdxL64xVQ3DxvEnakEJAW1Q,2348
-truefoundry/autodeploy/cli.py,sha256=QWOf3YrUzgfvglaS43MYyi3LnmWLBbZF8ntZsvVKTN0,10572
+truefoundry/autodeploy/cli.py,sha256=nr5FmwFbE5l0dKhsMr2BSkQ0vGRqGb92PfMIDogXmH0,11713
 truefoundry/autodeploy/constants.py,sha256=4ZHn08CCs_6ti-8Y6zaM_5IhI9mrB-1eDALhbn36oIw,435
 truefoundry/autodeploy/exception.py,sha256=BHqbkN-RdEsrfqPHQhWwSHiWnwQ6eaBc7PGGEXjJ3RY,54
 truefoundry/autodeploy/logger.py,sha256=tkV2UKcOTFl5nz0cn4eRbzxF-2CZd8b7MK9vnhaflYw,325
@@ -24,10 +24,10 @@ truefoundry/autodeploy/utils/diff.py,sha256=Ef8Y-VffDKel_-q-GxRam6gqiv8qTLMcqVg6
 truefoundry/autodeploy/utils/pydantic_compat.py,sha256=hEAUy5kLjhPdzw7yGZ2iXGMXbbMVXVlGzIofmyHafXQ,412
 truefoundry/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/cli/__main__.py,sha256=gSKVAG41klzNpe0IrURtRlR7zDo430tX0nEqR02L2SU,1585
-truefoundry/deploy/__init__.py,sha256=rY-AcjgSmu0r072rdUNpRkMmDFC7hH9drF5_pIN5jnw,29
-truefoundry/langchain/__init__.py,sha256=R1h2DkLLsVLPh3xVVzU2H_kpN4LNgzx3mPCcGVr2u5g,39
-truefoundry/ml/__init__.py,sha256=SJiI5QimXJETfME2x0iXz8gPn6C22YNiLiEdHIV1g7c,151
-truefoundry-0.2.0rc3.dist-info/METADATA,sha256=CuNSdxJPNngMDunPDi-FFb4hULsoPeqZ9_a85jgwbp0,1838
-truefoundry-0.2.0rc3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-truefoundry-0.2.0rc3.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.2.0rc3.dist-info/RECORD,,
+truefoundry/deploy/__init__.py,sha256=wDQq3y-t9zOEOVA61t15fRpAVQSpBtiWTyW5hVn1iz4,43
+truefoundry/langchain/__init__.py,sha256=kQrM0qr2EoB-EFaPkVVVcmSBSJ8KsQzv0nx-LGzmCRo,53
+truefoundry/ml/__init__.py,sha256=yFjvF-e1RW488vLHgn5M7TXoajqww6grkKHb3mhqDEw,179
+truefoundry-0.2.0rc5.dist-info/METADATA,sha256=OaBUuxH2on9rDV1Z3jdY38wg4GAUSE3t0xU0TIxcw78,1838
+truefoundry-0.2.0rc5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+truefoundry-0.2.0rc5.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.2.0rc5.dist-info/RECORD,,