tetra-rp 0.6.0__py3-none-any.whl → 0.24.0__py3-none-any.whl

tetra_rp-0.24.0.dist-info/RECORD

@@ -0,0 +1,99 @@
+tetra_rp/__init__.py,sha256=SMc8Yhv2uiN8bj6N0crQkgIxpzNyGb4m6JsHX7Wx8Z0,3561
+tetra_rp/client.py,sha256=cK0T2GF4P9cHdPdKbwoaXEWMCvP3n2La9aPjRe7afac,7386
+tetra_rp/config.py,sha256=9FYzouOam0PVoBIITpwHu1N5dGd1ueLHjWMaZYmdquI,760
+tetra_rp/execute_class.py,sha256=a-k6nfOdksuw--BDZ5Au4asJ2eHwbClg9B3YmkPITvg,11480
+tetra_rp/logger.py,sha256=gk5-PWp3k_GQ5DxndsRkBCX0jarp_3lgZ1oiTFuThQg,1125
+tetra_rp/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tetra_rp/cli/main.py,sha256=farGvK-zRvqRyT-HUMZPMRo-8DOQBHe-tsn24sGpVHs,2623
+tetra_rp/cli/commands/__init__.py,sha256=gQ6tnU0Rvm0-ESWFUBU-KDl5dpNOpUTG509hXOQQjwY,27
+tetra_rp/cli/commands/apps.py,sha256=qO6UCpRgv_wqQmLo0AmMNCzctP4OpQQ4I9zUUOpBeRM,4765
+tetra_rp/cli/commands/build.py,sha256=VxUOnHRZ84GxFYUaje0QyjPRbNpHJhNQUZLwNkqxbQk,38431
+tetra_rp/cli/commands/deploy.py,sha256=Ehx7XDVPO1WGOFGu7eNsmxrz9Mz1DPJh93Og3pFP9RA,19278
+tetra_rp/cli/commands/init.py,sha256=VK-noKJqdl-ua49-mEx6Vo72aLspiTxX16QmHVfNsJo,4652
+tetra_rp/cli/commands/resource.py,sha256=FWFJh0-O95tc9Byj6vESppuHFC98mmGMOOMc_rohT6s,3183
+tetra_rp/cli/commands/run.py,sha256=z1-YD4zAjplsFreTSaLaiPMqBSPXayDbpQ7XvzZn77k,9199
+tetra_rp/cli/commands/test_mothership.py,sha256=9fbzc5AmIMYg-161-NlJdiZWcg-oDr0ocIqmhuOXNmk,15817
+tetra_rp/cli/commands/undeploy.py,sha256=0NWvdkS6Hb1KKx649TITJQKVOQSFX9g0Q0NKIbPZgJk,17646
+tetra_rp/cli/commands/build_utils/__init__.py,sha256=3-JRo6x364rQNvBVtodjazltU5kRfT5q9uygZG4RvOE,52
+tetra_rp/cli/commands/build_utils/handler_generator.py,sha256=-55rk0QYGbZjlI9t-r5PvsfgMfH0rj5fTVLwOanSISM,6051
+tetra_rp/cli/commands/build_utils/lb_handler_generator.py,sha256=3ZrjX3HQ6HkLR5BLBSimtHXVXdYeGm_bCRoy0lSvyw0,10972
+tetra_rp/cli/commands/build_utils/manifest.py,sha256=7eYQYqwa7W5KuKklh8HjbXBaFEVft1cM411yo5esZiE,17928
+tetra_rp/cli/commands/build_utils/mothership_handler_generator.py,sha256=pNIGxuxmGUQ_UWrfY5-0NudFFuB5WZ3Rf1XxiYpNBOM,2179
+tetra_rp/cli/commands/build_utils/scanner.py,sha256=4l5avf-yefFJ-ujuZGLB5iJR8xD_GubWVZShCBt4UoQ,24508
+tetra_rp/cli/utils/__init__.py,sha256=Ytcdqe_Kzy37LSnsBj1Drkrx0Uo7TWExjQZ2mwW_nMg,27
+tetra_rp/cli/utils/app.py,sha256=QvgXcU4Rh03df1DMe9Xbo3nCAWc-pOm6UofGsT7pMUg,361
+tetra_rp/cli/utils/conda.py,sha256=Lj_ddXxysAFboR-Hd6wTM-mrV6TBEnQzKhEgxjUQdPo,3514
+tetra_rp/cli/utils/deployment.py,sha256=eB6Fxs6jHM8we9v1jVSOaQZeb4Jzr_3ngKtB-z4qaSA,18476
+tetra_rp/cli/utils/ignore.py,sha256=_xKYhvtvc4AK9jHDzRdA9F6xxGhmySEPgqPCsN7cHdg,3858
+tetra_rp/cli/utils/skeleton.py,sha256=wPEmnE_-wovPIciqdSFT52V6FkU3xOUrNunOosfoAks,5143
+tetra_rp/cli/utils/skeleton_template/.env.example,sha256=NlyPYFfuhVMjPym_cueY1g_AzLfk4UZN3w6sjLoD0RE,93
+tetra_rp/cli/utils/skeleton_template/.flashignore,sha256=tEYF62QjHhxS0AUinT0lREqQ7AykQ3lRn6MWKeo7fow,350
+tetra_rp/cli/utils/skeleton_template/.gitignore,sha256=gzGwh94QgCJTWM3ZJ-cqkgQjnl6XlGwUiZIRxQAugdA,334
+tetra_rp/cli/utils/skeleton_template/README.md,sha256=4GJxkw2jwW4oqJwDX5Bs6VP9njrAqRZXmEkWXnJNaIQ,6691
+tetra_rp/cli/utils/skeleton_template/main.py,sha256=KdmqQ_xCC5CVt6BupgDGAhJicD4Guac7-6hqVq4TC5E,960
+tetra_rp/cli/utils/skeleton_template/mothership.py,sha256=NoBVl_gj5G-LYid_6ys6a-nzX8nraeQorYRhfYwizAg,1422
+tetra_rp/cli/utils/skeleton_template/pyproject.toml,sha256=BK-i8NIETSvGrKRyR-rTHPtPw1SjAt7ls9cYKvP44Rg,1121
+tetra_rp/cli/utils/skeleton_template/requirements.txt,sha256=bBay7JTDwJXsTYvVjrwno9hnF-j0q272lk65f2AcPjU,9
+tetra_rp/cli/utils/skeleton_template/workers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tetra_rp/cli/utils/skeleton_template/workers/cpu/__init__.py,sha256=IIxdRazQUPPzZOLzCW5-CBs_nA8mum3pReWjpSSI8GI,419
+tetra_rp/cli/utils/skeleton_template/workers/cpu/endpoint.py,sha256=hknYcoQtbTPqTkizXBItJTWlTWOZFuUXfpm7ffhNnpc,948
+tetra_rp/cli/utils/skeleton_template/workers/gpu/__init__.py,sha256=EHavaP1AGbUvY44u0tlykZfe2-dlVKrk4HWtmMw5oeo,419
+tetra_rp/cli/utils/skeleton_template/workers/gpu/endpoint.py,sha256=e9m8how8M6XzYxECJ-DI4PMdGoM62rhyUu_r8vB6RQI,1782
+tetra_rp/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tetra_rp/core/deployment.py,sha256=t1aNBz9KoumBagqak4ad6BMhU4A4X8PSJ_yQyhebejk,7637
+tetra_rp/core/discovery.py,sha256=qFMym-rP_i3MY3JiBjaShlJDI3UTxU1Zl0X03RgshdU,15083
+tetra_rp/core/exceptions.py,sha256=YmsVzrAHx5NdF2styV1g1xqjrDO5ULh9vvciU37iYfo,1572
+tetra_rp/core/validation.py,sha256=r83rezzdSoJSUBIbyjCuSxf2-tBBw0PtBkUFY-djnkI,1145
+tetra_rp/core/api/__init__.py,sha256=oldrEKMwxYoBPLvPfVlaFS3wfUtTTxCN6-HzlpTh6vE,124
+tetra_rp/core/api/runpod.py,sha256=IwV8aGGcTd6qbXTXOeFMQ4fhxPsXp9wACwShPv9Aspo,28744
+tetra_rp/core/resources/__init__.py,sha256=0wcpxF8R-_xaycecAcfm4qln9sFEMyB2rszPCbv-0Ns,1289
+tetra_rp/core/resources/app.py,sha256=P4GY3H5TCRaO5fdUtpEW-FvMr8WMD0I4L0IkJdNjMm4,26968
+tetra_rp/core/resources/base.py,sha256=U_WUm-Z64DFvnzCdvJaptE7VWZDDUH7HehYLztlVSSo,6679
+tetra_rp/core/resources/cloud.py,sha256=XJOWPfzYlDVJGHxgffcfpEaOKrWhGdi7AzTlaGuYj0o,70
+tetra_rp/core/resources/constants.py,sha256=hKUhdEwjG-uPPfnObdVYlis2rEtanTiwewLSJqDL1pU,829
+tetra_rp/core/resources/cpu.py,sha256=Kk7ZDHi7Pv8AgxPujA3FB-ujRQCSubgKDlZfhdQDqKg,3935
+tetra_rp/core/resources/environment.py,sha256=FC9kJCa8YLSar75AKUKqJYnNLrUdjZj8ZTOrspBrS00,1267
+tetra_rp/core/resources/gpu.py,sha256=XvjoldCmREQuurKIXAKKW_OwdZYc5M-3LwZfPLeIMMw,7217
+tetra_rp/core/resources/live_serverless.py,sha256=vfXm-NH7vwhkUqAz5sSb9SG-8cujUPSS-wOHNrcSgwM,5463
+tetra_rp/core/resources/load_balancer_sls_resource.py,sha256=zoJcTicC27szDNctVBI7QxOwvVm22YGVsKD2qv12Q3s,14829
+tetra_rp/core/resources/network_volume.py,sha256=SJUXnYj6RXw05d78leOBTUufGk7cSAObhrZGDjDSRwo,6610
+tetra_rp/core/resources/resource_manager.py,sha256=Ya3L0awgCnCRr1syPhryDPP8BwNN4f8LINll4hh_g1Q,20550
+tetra_rp/core/resources/serverless.py,sha256=uAXTViiHG3fy0ERl1cRRz4YNupEirgB5-dp9kmWxxFk,30288
+tetra_rp/core/resources/serverless_cpu.py,sha256=I5t0IqPI5oFG1VdYRJj-iyI86otrXByZGtytnla30G4,7517
+tetra_rp/core/resources/template.py,sha256=BPZxcI-2UIXIrxrZvogpJLVawBYIzKgb-unn5_-Mw7I,1098
+tetra_rp/core/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tetra_rp/core/utils/backoff.py,sha256=1pfa0smFNpib8nztcIgBbtrVvQeECKh-aNOfL2TztgU,1324
+tetra_rp/core/utils/constants.py,sha256=Dm4XiO5zTzfdqOSeYVfAjaf2LyHnIEVmbOi_s_k1J_E,375
+tetra_rp/core/utils/file_lock.py,sha256=bxtAexD2rbqMhdr94VbmKdNp0gfKRgxDXx1n7LX4Eso,8269
+tetra_rp/core/utils/http.py,sha256=BjiFup--TTRBILmn1XT3SnIj-Gwr6IeDFXMSAQneuAE,2157
+tetra_rp/core/utils/lru_cache.py,sha256=drwKg-DfLbeBRGTzuxKqNKMQq0EuZV15LMTZIOyZuVk,2618
+tetra_rp/core/utils/singleton.py,sha256=yYarzyuk8uNgh0SygXGuEo9Mvue3y8MmQPD3QtwVIr0,1452
+tetra_rp/protos/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tetra_rp/protos/remote_execution.py,sha256=4UBl3Peyaf53O7LqvAw9rM226CsTJxf6q3AhQiKP9Pg,5438
+tetra_rp/runtime/__init__.py,sha256=jpj1kVQ-SQ4TeOgCFgaUFhZaxVjB3t0mZkEGw6YkEl0,56
+tetra_rp/runtime/circuit_breaker.py,sha256=kN4kPiZ4R611wZb7nMaMDPAsr1fUJsi98FKMz9YkUOE,9570
+tetra_rp/runtime/config.py,sha256=5x9ZMBqtuifux55GbbSUMEz6Q-IV3FjAQlI3s6veJR4,303
+tetra_rp/runtime/exceptions.py,sha256=X6owuVMOofxS_giGsvd0eCIQZMPAwEocAXgxyPrRypM,1005
+tetra_rp/runtime/generic_handler.py,sha256=2Y5CldR1CVMlODlUsOPMdV2ZvTQk8kOw29gGT5BopfM,6167
+tetra_rp/runtime/lb_handler.py,sha256=JAJEQDy1OZLSfwz-TlJgYrZiovLtLxiwX-NiEp7wGdE,7175
+tetra_rp/runtime/load_balancer.py,sha256=_8hOLvq2Ilvotnb6eO-0rBSQ6H1-zEGJA1ptVxcdDoQ,5274
+tetra_rp/runtime/manifest_fetcher.py,sha256=AjBNZX2jtE7Cfc7fqX-1M2ghFEzac4OqY4mrvbh-Kxs,6361
+tetra_rp/runtime/metrics.py,sha256=fssrZ74of4FqatuwMV6a6eYmgNcIDfk_wCUHYnLfhY0,9229
+tetra_rp/runtime/models.py,sha256=KZXLIPqDmbR07mYhOZHxGE6HjyQhHYqepnEHUPpNU_Y,2135
+tetra_rp/runtime/mothership_provisioner.py,sha256=zbC9p6LRAQfsOtAygcSc4gQ9kdccQscWqRxXr6X_5-A,18327
+tetra_rp/runtime/production_wrapper.py,sha256=yD98TQZmhxbgPNocnjAtj9LA66EepbApBdTr_bdC5eU,8112
+tetra_rp/runtime/reliability_config.py,sha256=gfUi6Rc5Vf0TSPQM6FKQtA1jvxxz6qth7zQvE8WEyXM,4733
+tetra_rp/runtime/retry_manager.py,sha256=6vfvsSwvlzM5Tx-MA1Di9SzMtWQDV8IlPCb0J0Z2fR0,4252
+tetra_rp/runtime/serialization.py,sha256=2TORWMkWcLd1UECLHm4OQM9p2Fswmqr8Jgd8LmSWtj4,3241
+tetra_rp/runtime/service_registry.py,sha256=kH1fEz2sAUzEFhCHNs2YKfS-Bl5HQkGgBP-FV0x6zac,12302
+tetra_rp/runtime/state_manager_client.py,sha256=6a-z8w1mSLcNePLSe9ujbWxffHDH0BR8llPWfPyLMGY,8935
+tetra_rp/stubs/__init__.py,sha256=ozKsHs8q0T7o2qhQEquub9hqomh1Htys53mMraaRu2E,72
+tetra_rp/stubs/live_serverless.py,sha256=ymHgXrq42O2RB-pG1JrsnFx6xW26TtdAVVUy3RH_nZ8,4622
+tetra_rp/stubs/load_balancer_sls.py,sha256=-Q_6sR2THBeIAJecWWJ_tDMU6e5m1rJ2VxUoL6_GwFI,13423
+tetra_rp/stubs/registry.py,sha256=fwBhFCf93ybYt3rK1YRYqdL6fxO8M6kdUOjrRC-nwcs,5855
+tetra_rp/stubs/serverless.py,sha256=BM_a5Ml5VADBYu2WRNmo9qnicP8NnXDGl5ywifulbD0,947
+tetra_rp-0.24.0.dist-info/METADATA,sha256=EpQ1U1-PNmZIRtaY3Ytl6WCy9KhefOTKPUpuVZvcxtw,43147
+tetra_rp-0.24.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+tetra_rp-0.24.0.dist-info/entry_points.txt,sha256=FJi3ytBLKCcyz4tXJhOM3XNKzTxpnl3AauiPH2lHvWY,48
+tetra_rp-0.24.0.dist-info/top_level.txt,sha256=bBay7JTDwJXsTYvVjrwno9hnF-j0q272lk65f2AcPjU,9
+tetra_rp-0.24.0.dist-info/RECORD,,

{tetra_rp-0.6.0.dist-info → tetra_rp-0.24.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

tetra_rp-0.24.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+flash = tetra_rp.cli.main:app

tetra_rp/core/pool/cluster_manager.py

@@ -1,177 +0,0 @@
-import time
-from worker import Worker
-from job import Job
-
-from dataclass import WorkerStatus, JobStatus
-
-import logging
-import inspect
-
-
-def setup_logging(level=logging.INFO, fmt=None):
-    if fmt is None:
-        fmt = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
-    logging.basicConfig(level=level, format=fmt)
-
-
-def get_logger(name=None):
-    """
-    Returns a logger. If no name is provided, it infers the caller's module name.
-    """
-    if name is None:
-        # Get the caller's module name.
-        frame = inspect.stack()[1]
-        module = inspect.getmodule(frame[0])
-        name = module.__name__ if module else "__main__"
-    return logging.getLogger(name)
-
-
-logger = get_logger(__name__)
-
-
-class ClusterManager:
-    """
-    Manages workers and Jobs currently in Memory:
-    - Runpod for provisioning
-    - Real remote execution
-    - Data base for the
-    """
-
-    def __init__(self):
-        self.workers = {}  # Worker ID -> Worker
-        self.jobs = {}  # Job ID -> Job
-
-    # ----------------- Worker Management -----------------
-    # ------------------------------------------------------
-    def add_worker(self, resource_config: dict):
-        """
-        Add a new worker to the cluster
-        """
-        # here will go the logic to create a worker and add it to the cluster: RUNPOD LOGIC will be added here.
-        worker = Worker(resource_config)
-        self.workers[worker.worker_id] = worker
-
-        logger.info(f"Added worker {worker.worker_id} to the cluster")
-        return worker.worker_id
-
-    def remove_worker(self, worker_id):
-        """
-        Remove a worker from the cluster
-        """
-        worker = self.workers.get(worker_id)
-        if not worker:
-            logger.error(f"Worker {worker_id} not found")
-            return False
-        if worker.status == WorkerStatus.RUNNING:
-            logger.error(f"Worker {worker_id} is still running")
-            return False
-        del self.workers[worker_id]
-        logger.info(f"Removed worker {worker_id} from the cluster")
-        return True
-
-    def list_workers(self):
-        """
-        List all workers in the cluster
-        """
-        return list(self.workers.values())
-
-    # ----------------- Job Management -----------------
-    # ---------------------------------------------------
-
-    def submit_job(self, resource_config: dict):
-        """
-        Submit a new job to the cluster (Queueud). Then attempt to scheduel it.
-        """
-        job = Job(resource_config)
-        self.jobs[job.job_id] = job
-        logger.info(f"Submitted job {job.job_id} to the cluster")
-        # attempt to schedule the job
-        self.schedule_job(job)
-        return job.job_id
-
-    def schedule_job(self, job: Job):
-        """
-        find a suitable worker for the job. It none, Job remains queued.
-        If we want to a auto provision we can actually add a logic here to add a worker if none is available.
-        """
-        if job.status != JobStatus.QUEUED:
-            logger.error(f"Job {job.job_id} is not pending")
-            return False
-
-        # Find worker candidate
-        candidate = self.find_idle_worker(job.resource_config)
-        if candidate:
-            self.assign_job_to_worker(job, candidate)
-        else:
-            logger.info(f"No worker available for job {job.job_id}")
-            # we cn either provision new worker from here and then scehediule the job from here.
-
-    def find_idle_worker(self, resource_config: dict):
-        """
-        Find an idle worker that can run the job
-        """
-        for w in self.workers.values():
-            if w.status == WorkerStatus.IDLE:
-                # check the resource config
-                if w.resource_config == resource_config:
-                    continue
-                return w
-        return None
-
-    def assign_job_to_worker(self, job: Job, worker: Worker):
-        """
-        Mark the job as running and the worker as Running and 'execute' the job.
-        In a real system, we would send a remote command to the worker (eg: gRPC) to execute the job.
-        """
-        job.worker_id = worker.worker_id
-        job.status = JobStatus.RUNNING
-        worker.status = WorkerStatus.RUNNING
-        worker.current_job_id = job.job_id
-        logger.info(f"Assigned job {job.job_id} to worker {worker.worker_id}")
-        self._execute_job(job, worker)
-
-    def _execute_job(self, job: Job, worker: Worker):
-        """
-        Simulate the remote execution. right now, we jsut sleep for 1s.
-        In production, what we we can do is:
-        - Open a gRPC connection to the worker
-        - pass the job details
-        - wait for the compeltion call back
-        """
-        try:
-            logger.info(f"Executing job {job.job_id} on worker {worker.worker_id}")
-            time.sleep(
-                1
-            )  # Here we can add the actual execution logic, currently it mimics the execution.
-
-            # mark the job as completed
-            job.status = JobStatus.COMPLETED
-            job.result = "Job completed successfully"
-            logger.info(f"[Cluster Manager] Job {job.job_id} completed successfully")
-        except Exception as e:
-            job.status = JobStatus.FAILED
-            job.result = f"Job failed: {str(e)}"
-            logger.error(f"[Cluster Manager] Job {job.job_id} failed: {str(e)}")
-        finally:
-            worker.status = WorkerStatus.IDLE
-            worker.current_job_id = None
-
-    def get_job_status(self, job_id):
-        """
-        Get the job details
-        """
-        job = self.jobs.get(job_id)
-        if not job:
-            logger.error(f"Job {job_id} not found")
-            return None
-        return job
-
-    # this function has retry logic but it's currently fuzzy, we might have to change it.
-
-    def retry_queued_jobs(self):
-        """
-        Retry all queued jobs
-        """
-        for job in self.jobs.values():
-            if job.status == JobStatus.QUEUED:
-                self.schedule_job(job)

tetra_rp/core/pool/dataclass.py

@@ -1,18 +0,0 @@
-from enum import Enum
-
-
-class WorkerStatus(Enum):
-    """Enum representing the status of a worker"""
-
-    IDLE = "idle"
-    RUNNING = "running"
-    OFFLINE = "offline"
-
-
-class JobStatus(Enum):
-    """Enum representing the status of a job"""
-
-    QUEUED = "queued"
-    RUNNING = "running"
-    COMPLETED = "completed"
-    FAILED = "failed"

tetra_rp/core/pool/ex.py

@@ -1,38 +0,0 @@
-from cluster_manager import ClusterManager
-
-
-if __name__ == "__main__":
-    cm = ClusterManager()
-
-    # 1) Submit a job with no existing workers (use resource_config dict)
-    job_id = cm.submit_job(
-        resource_config={"gpu": "H100", "memory": 16, "network_volume": 50}
-    )
-    print(
-        "Job status:", cm.get_job_status(job_id)
-    )  # should be QUEUED, no suitable worker
-
-    # 2) Add a worker that doesn't match the GPU
-    w1 = cm.add_worker(
-        resource_config={"gpu": "H100", "memory": 16, "network_volume": 50}
-    )
-    # Re-try scheduling
-    cm.retry_queued_jobs()
-    print("Job status (still queued):", cm.get_job_status(job_id))
-
-    # 3) Add a matching worker
-    w2 = cm.add_worker(
-        resource_config={"gpu": "H100", "memory": 16, "network_volume": 50}
-    )
-    # Re-try scheduling
-    cm.retry_queued_jobs()
-    print("Job status (should complete):", cm.get_job_status(job_id))
-
-    # 4) Submit another job that requires less resources
-    job_id2 = cm.submit_job(resource_config={"memory": 8, "network_volume": 10})
-    # Should be assigned to w1 if it's idle
-    print("Job2 final status:", cm.get_job_status(job_id2))
-
-    # 5) Show final state of workers
-    for worker in cm.list_workers():
-        print("Worker:", worker)

tetra_rp/core/pool/job.py

@@ -1,22 +0,0 @@
-import uuid
-from dataclass import JobStatus
-
-
-class Job:
-    """Represents a 'job' in the system
-
-    In a real system, this might contain the function to run,
-    arguments, and reference to data or code.
-    """
-
-    def __init__(self, resource_config: dict):
-        self.job_id = str(uuid.uuid4())[:8]
-        self.resource_config = resource_config
-        self.status = JobStatus.QUEUED
-
-        self.worker_id = None
-        self.result = None
-        self.error = None
-
-    def __repr__(self):
-        return f"Job(job_id={self.job_id}, status={self.status})"

tetra_rp/core/pool/worker.py

@@ -1,19 +0,0 @@
-import uuid
-from dataclass import WorkerStatus
-
-
-class Worker:
-    """Represents a single worker in the pool
-
-    For Now we store ressources in memory
-    """
-
-    def __init__(self, resource_config: dict):
-        self.worker_id = str(uuid.uuid4())[:8]
-        self.resource_config = resource_config
-        self.status = WorkerStatus.IDLE
-
-        self.current_job_id = None
-
-    def __repr__(self):
-        return f"Worker(worker_id={self.worker_id}, status={self.status})"

tetra_rp/core/resources/utils.py

@@ -1,50 +0,0 @@
-from typing import Callable, Any, List, Union
-from pydantic import BaseModel
-from .gpu import GpuType, GpuTypeDetail
-from .serverless import ServerlessEndpoint
-
-
-"""
-Define the mapping for the methods and their return types
-Only include methods from runpod.*
-"""
-RUNPOD_TYPED_OPERATIONS = {
-    "get_gpus": List[GpuType],
-    "get_gpu": GpuTypeDetail,
-    "get_endpoints": List[ServerlessEndpoint],
-}
-
-
-def inquire(method: Callable, *args, **kwargs) -> Union[List[Any], Any]:
-    """
-    This function dynamically determines the return type of the provided method
-    based on a predefined mapping (`definitions`) and validates the result using
-    Pydantic models if applicable.
-
-    Refer to `RUNPOD_TYPED_OPERATIONS` for the mapping.
-
-    Example:
-    ----------
-    >>> import runpod
-    >>> inquire(runpod.get_gpus)
-    [
-        GpuType(id='NVIDIA A100 80GB', displayName='A100 80GB', memoryInGb=80),
-        GpuType(id='NVIDIA A100 40GB', displayName='A100 40GB', memoryInGb=40),
-        GpuType(id='NVIDIA A10', displayName='A10', memoryInGb=24)
-    ]
-    """
-    method_name = method.__name__
-    return_type = RUNPOD_TYPED_OPERATIONS.get(method_name)
-
-    raw_result = method(*args, **kwargs)
-
-    if hasattr(return_type, "__origin__") and return_type.__origin__ is list:
-        # List case
-        model_type = return_type.__args__[0]
-        if issubclass(model_type, BaseModel):
-            return [model_type.model_validate(item) for item in raw_result]
-    elif isinstance(return_type, type) and issubclass(return_type, BaseModel):
-        # Single object case
-        return return_type.model_validate(raw_result)
-    else:
-        raise ValueError(f"Unsupported return type for method '{method_name}'")

tetra_rp/core/utils/json.py

@@ -1,33 +0,0 @@
-from enum import Enum
-from typing import Any
-from pydantic import BaseModel
-
-
-def normalize_for_json(obj: Any) -> Any:
-    """
-    Recursively normalizes an object for JSON serialization.
-
-    This function handles various data types and ensures that objects
-    are converted into JSON-serializable formats. It supports the following:
-    - `BaseModel` instances: Converts them to dictionaries using `model_dump()`.
-    - Dictionaries: Recursively normalizes their values.
-    - Lists: Recursively normalizes their elements.
-    - Tuples: Recursively normalizes their elements and returns a tuple.
-    - Other types: Returns the object as is.
-
-    Args:
-        obj (Any): The object to normalize.
-
-    Returns:
-        Any: A JSON-serializable representation of the input object.
-    """
-    if isinstance(obj, BaseModel):
-        return normalize_for_json(obj.model_dump())
-    elif isinstance(obj, Enum):
-        return obj.value
-    elif isinstance(obj, dict):
-        return {k: normalize_for_json(v) for k, v in obj.items()}
-    elif isinstance(obj, (list, tuple)):
-        return type(obj)(normalize_for_json(i) for i in obj)
-    else:
-        return obj

tetra_rp-0.6.0.dist-info/RECORD

@@ -1,39 +0,0 @@
-tetra_rp/__init__.py,sha256=-1S5sYIKtnUV8V1HlSIbX1yZwiUrsO8J5b3ZEIR_phU,687
-tetra_rp/client.py,sha256=sJk3kneFJo40OAtbxucbaJ6NxMs-kD2jkpXYwG0vJhM,2681
-tetra_rp/logger.py,sha256=gk5-PWp3k_GQ5DxndsRkBCX0jarp_3lgZ1oiTFuThQg,1125
-tetra_rp/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tetra_rp/core/api/__init__.py,sha256=oldrEKMwxYoBPLvPfVlaFS3wfUtTTxCN6-HzlpTh6vE,124
-tetra_rp/core/api/runpod.py,sha256=n1pfvzDceU_yO9iOn4B9Kst6YL2KD0m-migm3bvnfJM,10374
-tetra_rp/core/pool/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tetra_rp/core/pool/cluster_manager.py,sha256=KJxEp_044HjnbOhfIdiXZbks_bFDYE1KgKeR5W9VvbY,6007
-tetra_rp/core/pool/dataclass.py,sha256=YngS328_NTewY8Etitj4k7MmdM5GWqqE_OMbytrVNlw,338
-tetra_rp/core/pool/ex.py,sha256=AZOrn9t_X5ycMl-tDg7-jcIURj_9kVmzn9_da8h1TFI,1273
-tetra_rp/core/pool/job.py,sha256=4bisW_ZwiQ2-qD5l0y9SbHcO4EQvSKimmBBU1fpI_YE,567
-tetra_rp/core/pool/worker.py,sha256=N4cOnf8MiDcPFH2XSMmSnnWMACZYUNnKWVhOx2aSxvM,478
-tetra_rp/core/resources/__init__.py,sha256=UhIwo1Y6-tw5qsULamR296sQiztuz-oWrSTreqfmFSw,814
-tetra_rp/core/resources/base.py,sha256=UJeDiFN45aO1n5SBcxn56ohLhj-AWHoj0KO7mF4yJ_o,1440
-tetra_rp/core/resources/cloud.py,sha256=XJOWPfzYlDVJGHxgffcfpEaOKrWhGdi7AzTlaGuYj0o,70
-tetra_rp/core/resources/constants.py,sha256=F1gPqFaXcCmfrbUSO9PQtUBv984TxFc3pySgVy-kXk8,158
-tetra_rp/core/resources/cpu.py,sha256=YIE-tKolSU3JJzpPB7ey-PbRdqKWsJZ_Ad4h2OYaaiA,1231
-tetra_rp/core/resources/environment.py,sha256=FC9kJCa8YLSar75AKUKqJYnNLrUdjZj8ZTOrspBrS00,1267
-tetra_rp/core/resources/gpu.py,sha256=2jIIMr8PNnlIAP8ZTKO8Imx-rdxXp2rbdSHJeVfjawk,1858
-tetra_rp/core/resources/live_serverless.py,sha256=6r4I4TEx9AmZ0-OJvE86qrY0S7BEx9t_P2zwHVdtbew,1074
-tetra_rp/core/resources/network_volume.py,sha256=mzYA7Bf4JTFiDyH0sLrUbPjq-26MWgbbRWAMbX2h6iA,2881
-tetra_rp/core/resources/resource_manager.py,sha256=kUVZDblfUzaG78S8FwOzu4rN6QSegUgQNK3fJ_X7l0w,2834
-tetra_rp/core/resources/serverless.py,sha256=lZMITI1cwOBVNX3dbFyR01AeRZ0eatNC4CRE7vwUrqs,14726
-tetra_rp/core/resources/template.py,sha256=UkflJXZFWIbQkLuUt4oRLAjn-yIpw9_mT2X1cAH69CU,3141
-tetra_rp/core/resources/utils.py,sha256=mgXfgz_NuHN_IC7TzMNdH9II-LMjxcDCG7syDTcPiGs,1721
-tetra_rp/core/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tetra_rp/core/utils/backoff.py,sha256=1pfa0smFNpib8nztcIgBbtrVvQeECKh-aNOfL2TztgU,1324
-tetra_rp/core/utils/json.py,sha256=q0r7aEdfh8kKVeHGeh9fBDfuhHYNopSreislAMB6HhM,1163
-tetra_rp/core/utils/singleton.py,sha256=JRli0HhBfq4P9mBUOg1TZUUwMvIenRqWdymX3qFMm2k,210
-tetra_rp/protos/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tetra_rp/protos/remote_execution.py,sha256=I4641-Dlzj4O7AuZbVei8O9aV2VNrRTcE8r7Fm0e-V8,1901
-tetra_rp/stubs/__init__.py,sha256=ozKsHs8q0T7o2qhQEquub9hqomh1Htys53mMraaRu2E,72
-tetra_rp/stubs/live_serverless.py,sha256=o1NH5XEwUD-27NXJsEGO0IwnuDp8iXwUiw5nZtaZZOI,4199
-tetra_rp/stubs/registry.py,sha256=V4m3CeXl8j1pguHuuflxqpWeBgVDQ93YkhxJbElyP7Q,2599
-tetra_rp/stubs/serverless.py,sha256=BM_a5Ml5VADBYu2WRNmo9qnicP8NnXDGl5ywifulbD0,947
-tetra_rp-0.6.0.dist-info/METADATA,sha256=CPOdAGUy5vZOSLazBQubYChiQkuwDldW3G0K9KsyBlw,28055
-tetra_rp-0.6.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-tetra_rp-0.6.0.dist-info/top_level.txt,sha256=bBay7JTDwJXsTYvVjrwno9hnF-j0q272lk65f2AcPjU,9
-tetra_rp-0.6.0.dist-info/RECORD,,