truefoundry 0.2.10__py3-none-any.whl → 0.3.0__py3-none-any.whl

truefoundry/__init__.py

@@ -0,0 +1 @@
+from truefoundry.deploy.core import login, logout

truefoundry/autodeploy/cli.py

@@ -13,12 +13,6 @@ from openai import OpenAI
 from rich.console import Console
 from rich.prompt import Prompt
 from rich.status import Status
-from servicefoundry import Build, DockerFileBuild, Job, LocalSource, Port, Service
-from servicefoundry.cli.const import COMMAND_CLS
-from servicefoundry.lib.auth.servicefoundry_session import ServiceFoundrySession
-from servicefoundry.lib.clients.service_foundry_client import (
-    ServiceFoundryServiceClient,
-)
 
 from truefoundry.autodeploy.agents.developer import Developer
 from truefoundry.autodeploy.agents.project_identifier import (
@@ -37,6 +31,12 @@ from truefoundry.autodeploy.tools.ask import AskQuestion
 from truefoundry.autodeploy.tools.commit import CommitConfirmation
 from truefoundry.autodeploy.tools.docker_run import DockerRun, DockerRunLog
 from truefoundry.autodeploy.utils.client import get_git_binary
+from truefoundry.deploy import Build, DockerFileBuild, Job, LocalSource, Port, Service
+from truefoundry.deploy.cli.const import COMMAND_CLS
+from truefoundry.deploy.lib.auth.servicefoundry_session import ServiceFoundrySession
+from truefoundry.deploy.lib.clients.servicefoundry_client import (
+    ServiceFoundryServiceClient,
+)
 
 
 def _get_openai_client() -> OpenAI:
@@ -58,7 +58,11 @@ def _get_openai_client() -> OpenAI:
     except requests.exceptions.HTTPError as http_error:
         if http_error.response.status_code in [401, 403]:
             raise InvalidRequirementsException(
-                message='Unauthorized access to Truefoundry server. Please verify your credentials and ensure you have the necessary access rights.\nIf you wish to proceed without Truefoundry AI, you need to either have a truefoundry.yaml file in your project root or pass the path to a yaml file using the "--file file_name" option.'
+                message="Unauthorized access to TrueFoundry server. "
+                "Please verify your credentials and ensure you have the necessary access rights."
+                "\nIf you wish to proceed without TrueFoundry AI, "
+                "you need to either have a truefoundry.yaml file in your project root "
+                'or pass the path to a yaml file using the "--file file_name" option.'
             ) from http_error
 
         raise http_error
@@ -146,7 +150,8 @@ def _check_repo(project_root_path: str, console: Console):
         repo = git.Repo(path=project_root_path, search_parent_directories=True)
         if repo.is_dirty():
             console.print(
-                "[bold red]Error:[/] The repository has uncommitted changes. Please commit or stash them before proceeding."
+                "[bold red]Error:[/] The repository has uncommitted changes. "
+                "Please commit or stash them before proceeding."
             )
             sys.exit(1)
         current_active_branch = repo.active_branch.name
@@ -211,11 +216,13 @@ def _update_status(event, status: Status, component_type: ComponentType):
     if isinstance(event, DockerRunLog):
         if component_type == ComponentType.SERVICE:
             status.update(
-                "[bold cyan]Running:[/] [bold magenta]TrueFoundry[/] is running your app in a Docker container. Press ctrl+c once your app is ready for testing."
+                "[bold cyan]Running:[/] [bold magenta]TrueFoundry[/] is running your app in a Docker container. "
+                "Press ctrl+c once your app is ready for testing."
             )
         else:
             status.update(
-                "[bold cyan]Running:[/] [bold magenta]TrueFoundry[/] is running your app in a Docker container and waiting for completion."
+                "[bold cyan]Running:[/] [bold magenta]TrueFoundry[/] is running your app in a Docker container "
+                "and waiting for completion."
             )
 
 
@@ -254,8 +261,10 @@ def cli(project_root_path: str, deploy: bool, workspace_fqn: str = None):
         )
     else:
         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.[/]",
+            "[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."
@@ -264,11 +273,13 @@ def cli(project_root_path: str, deploy: bool, workspace_fqn: str = None):
     _check_repo(project_root_path=project_root_path, console=console)
 
     choices = {
-        "Service: An application that runs continuously. Example: web servers, workers polling a job queue, etc.": "SERVICE",
-        "Job: An application that runs once and then stops. Example: Training an ML model, running a script, etc.": "JOB",
+        "Service: An application that runs continuously. "
+        "Example: web servers, workers polling a job queue, etc.": "SERVICE",
+        "Job: An application that runs once and then stops. "
+        "Example: Training an ML model, running a script, etc.": "JOB",
     }
     component = questionary.select(
-        "TrueFoundry: Is your project a", choices=choices.keys()
+        "TrueFoundry: Is your project a", choices=list(choices.keys())
     ).ask()
     component_type = ComponentType[choices[component]]
     while True:
@@ -300,8 +311,10 @@ def cli(project_root_path: str, deploy: bool, workspace_fqn: str = None):
     )
     if workspace_fqn is None:
         workspace_fqn = Prompt.ask(
-            "[bold magenta]TrueFoundry:[/] Enter the Workspace FQN where you would like to deploy, [dim]Ex: cluster-name:workspace-name[/]"
+            "[bold magenta]TrueFoundry:[/] Enter the Workspace FQN where you would like to deploy, [dim]"
+            "Ex: cluster-name:workspace-name[/]"
         )
+    env = {}
     while True:
         try:
             env = _parse_env(project_root_path, env_path) if env_path else {}
@@ -340,7 +353,7 @@ def cli(project_root_path: str, deploy: bool, workspace_fqn: str = None):
                 break
 
     if deploy:
-        console.rule("[bold green]Deploying to Truefoundry[/]")
+        console.rule("[bold green]Deploying to TrueFoundry[/]")
         deploy_component(
             workspace_fqn=workspace_fqn,
             project_root_path=project_root_path,
@@ -367,7 +380,7 @@ def cli(project_root_path: str, deploy: bool, workspace_fqn: str = None):
 )
 def autodeploy_cli(path: str, deploy: bool):
     """
-    Build and deploy projects using Truefoundry
+    Build and deploy projects using TrueFoundry
     """
     cli(
         project_root_path=path,

truefoundry/deploy/__init__.py

@@ -1 +1,112 @@
-from servicefoundry import *  # noqa: F403
+from truefoundry.deploy.auto_gen.models import (
+    AppProtocol,
+    CanaryStep,
+    CapacityType,
+    ConcurrencyPolicy,
+    Kustomize,
+    ParamType,
+    Protocol,
+    WorkbenchImage,
+)
+from truefoundry.deploy.lib.dao.application import (
+    delete_application,
+    get_application,
+    get_job_run,
+    list_applications,
+    list_job_runs,
+    terminate_job_run,
+    trigger_job,
+    trigger_workflow,
+)
+from truefoundry.deploy.lib.dao.version import (
+    get_version as get_application_version,
+)
+from truefoundry.deploy.lib.dao.version import (
+    list_versions as list_application_versions,
+)
+from truefoundry.deploy.lib.dao.workspace import (
+    delete_workspace,
+    get_workspace_by_fqn,
+    list_workspaces,
+)
+from truefoundry.deploy.v2.lib.deployable_patched_models import (
+    Application,
+    ApplicationSet,
+    AsyncService,
+    Codeserver,
+    Helm,
+    Job,
+    Notebook,
+    Service,
+    SSHServer,
+    Volume,
+    Workflow,
+)
+from truefoundry.deploy.v2.lib.patched_models import (
+    AMQPInputConfig,
+    AMQPMetricConfig,
+    AMQPOutputConfig,
+    ArtifactsCacheVolume,
+    ArtifactsDownload,
+    AsyncProcessorSidecar,
+    AsyncServiceAutoscaling,
+    Autoscaling,
+    AWSAccessKeyAuth,
+    AWSInferentia,
+    AWSInferentiaAccelerator,
+    BasicAuthCreds,
+    BlueGreen,
+    Build,
+    Canary,
+    CoreNATSOutputConfig,
+    CPUUtilizationMetric,
+    CronMetric,
+    CUDAVersion,
+    DockerFileBuild,
+    DynamicVolumeConfig,
+    Endpoint,
+    GcpTPU,
+    GitSource,
+    GPUType,
+    HealthProbe,
+    HelmRepo,
+    HttpProbe,
+    HuggingfaceArtifactSource,
+    Image,
+    KafkaInputConfig,
+    KafkaMetricConfig,
+    KafkaOutputConfig,
+    KafkaSASLAuth,
+    LocalSource,
+    Manual,
+    NATSInputConfig,
+    NATSMetricConfig,
+    NATSOutputConfig,
+    NATSUserPasswordAuth,
+    NodepoolSelector,
+    NodeSelector,
+    NvidiaGPU,
+    NvidiaMIGGPU,
+    NvidiaTimeslicingGPU,
+    OCIRepo,
+    Param,
+    Port,
+    PythonBuild,
+    RemoteSource,
+    Resources,
+    Rolling,
+    RPSMetric,
+    Schedule,
+    SecretMount,
+    ServiceAutoscaling,
+    SQSInputConfig,
+    SQSOutputConfig,
+    SQSQueueMetricConfig,
+    StaticVolumeConfig,
+    StringDataMount,
+    TPUType,
+    TruefoundryArtifactSource,
+    VolumeBrowser,
+    VolumeMount,
+    WorkerConfig,
+)