refactor: reduce amount of data in session.yaml

This commit is contained in:
2025-04-09 11:52:02 -06:00
parent 5b1afa05da
commit 7421500eb0
4 changed files with 0 additions and 40 deletions

View File

@@ -85,9 +85,6 @@ def list_sessions() -> None:
table.add_column("Driver")
table.add_column("Status")
table.add_column("Ports")
table.add_column("Project")
table.add_column("Project Name")
table.add_column("MCPs")
for session in sessions:
ports_str = ", ".join(
@@ -110,18 +107,12 @@ def list_sessions() -> None:
else str(session.status)
)
# Format MCPs as a comma-separated list
mcps_str = ", ".join(session.mcps) if session.mcps else ""
table.add_row(
session.id,
session.name,
session.driver,
f"[{status_color}]{status_name}[/{status_color}]",
ports_str,
session.project or "",
session.project_name or "",
mcps_str,
)
console.print(table)

View File

@@ -112,11 +112,6 @@ class ContainerManager:
driver=labels.get("mc.driver", "unknown"),
status=status,
container_id=container_id,
created_at=container.attrs["Created"],
project=labels.get("mc.project"),
project_name=labels.get("mc.project_name"),
model=labels.get("mc.model"),
provider=labels.get("mc.provider"),
)
# Get port mappings
@@ -621,18 +616,7 @@ class ContainerManager:
driver=driver_name,
status=SessionStatus.RUNNING,
container_id=container.id,
environment=env_vars,
project=project,
project_name=project_name,
created_at=container.attrs["Created"],
ports=ports,
mcps=mcp_names,
run_command=run_command,
uid=uid,
gid=gid,
model=model,
provider=provider,
ssh=ssh,
)
# Save session to the session manager

View File

@@ -100,18 +100,7 @@ class Session(BaseModel):
driver: str
status: SessionStatus
container_id: Optional[str] = None
environment: Dict[str, str] = Field(default_factory=dict)
project: Optional[str] = None
project_name: Optional[str] = None
created_at: str
ports: Dict[int, int] = Field(default_factory=dict)
mcps: List[str] = Field(default_factory=list)
run_command: Optional[str] = None
uid: Optional[int] = None
gid: Optional[int] = None
model: Optional[str] = None
provider: Optional[str] = None
ssh: bool = False
class Config(BaseModel):

View File

@@ -88,16 +88,12 @@ def mock_session_manager():
@pytest.fixture
def mock_container_manager():
"""Mock the ContainerManager class with proper initialization."""
timestamp = "2023-01-01T00:00:00Z" # Use fixed timestamp for reproducibility
mock_session = Session(
id="test-session-id",
name="test-session",
driver="goose",
status=SessionStatus.RUNNING,
ports={"8080": "8080"},
project=None,
created_at=timestamp,
mcps=[],
)
with patch("mcontainer.cli.container_manager") as mock_manager: