Skip to content

Restruct the database #123

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
InftyAI-Agent merged 4 commits into from
Jan 29, 2026
Merged

Restruct the database #123

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
528a5c3
Restruct the dataabase
kerthcet Jan 29, 2026
9d6f45f
update seed
kerthcet Jan 29, 2026
0ec957e
fix lint
kerthcet Jan 29, 2026
e44922c
fix lint
kerthcet Jan 29, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion Makefile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -55,4 +55,4 @@ seed-cleanup:

build-dashboard:
rm -rf alphatrion/static/*
cd dashboard && npm install && npm run build
cd dashboard && npm install && npm run build
15 changes: 10 additions & 5 deletions alphatrion/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,18 +1,23 @@
from alphatrion.experiment.craft_exp import CraftExperiment
from alphatrion.experiment.experiment import (
CheckpointConfig,
Experiment,
ExperimentConfig,
MonitorMode,
)
from alphatrion.log.log import log_artifact, log_metrics, log_params
from alphatrion.metadata.sql_models import Status
from alphatrion.project.project import Project
from alphatrion.runtime.runtime import init
from alphatrion.tracing.tracing import task, workflow
from alphatrion.trial.trial import CheckpointConfig, MonitorMode, Trial, TrialConfig

__all__ = [
"init",
"log_artifact",
"log_params",
"log_metrics",
"CraftExperiment",
"Trial",
"TrialConfig",
"Project",
"Experiment",
"ExperimentConfig",
"CheckpointConfig",
"MonitorMode",
"Status",
Expand Down
10 changes: 5 additions & 5 deletions alphatrion/artifact/artifact.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,8 @@


class Artifact:
def __init__(self, project_id: str, insecure: bool = False):
self._project_id = project_id
def __init__(self, team_id: str, insecure: bool = False):
self._team_id = team_id
self._url = os.environ.get(consts.ARTIFACT_REGISTRY_URL)
self._url = self._url.replace("https://", "").replace("http://", "")
self._client = oras.client.OrasClient(
Expand Down Expand Up @@ -48,7 +48,7 @@ def push(
raise ValueError("No files to push.")

url = self._url if self._url.endswith("/") else f"{self._url}/"
path = f"{self._project_id}/{repo_name}:{version}"
path = f"{self._team_id}/{repo_name}:{version}"
target = f"{url}{path}"

try:
Expand All @@ -61,7 +61,7 @@ def push(
# TODO: should we store it in the metadb instead?
def list_versions(self, repo_name: str) -> list[str]:
url = self._url if self._url.endswith("/") else f"{self._url}/"
target = f"{url}{self._project_id}/{repo_name}"
target = f"{url}{self._team_id}/{repo_name}"
try:
tags = self._client.get_tags(target)
return tags
Expand All @@ -70,7 +70,7 @@ def list_versions(self, repo_name: str) -> list[str]:

def delete(self, repo_name: str, versions: str | list[str]):
url = self._url if self._url.endswith("/") else f"{self._url}/"
target = f"{url}{self._project_id}/{repo_name}"
target = f"{url}{self._team_id}/{repo_name}"

try:
self._client.delete_tags(target, tags=versions)
Expand Down
134 changes: 0 additions & 134 deletions alphatrion/experiment/base.py
View file
Open in desktop

This file was deleted.

69 changes: 0 additions & 69 deletions alphatrion/experiment/craft_exp.py
View file
Open in desktop

This file was deleted.

Loading
Loading