v2.7.3 (#564)

Co-authored-by: Bastien <bastien.baret@gmail.com>
Co-authored-by: Laure Hugo <201583486+laure0303@users.noreply.github.com>
Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com>
Co-authored-by: Paul Cacheux <paul.cacheux@mistral.ai>
Co-authored-by: Val <102326092+vdeva@users.noreply.github.com>
Co-authored-by: Mistral Vibe <vibe@mistral.ai>

vibe/core/session/resume_sessions.py

@@ -0,0 +1,93 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Literal
+
+from vibe.core.config import VibeConfig
+from vibe.core.logger import logger
+from vibe.core.nuage.client import WorkflowsClient
+from vibe.core.nuage.workflow import WorkflowExecutionStatus
+from vibe.core.session.session_loader import SessionLoader
+
+ResumeSessionSource = Literal["local", "remote"]
+
+SHORT_SESSION_ID_LEN = 8
+
+
+def short_session_id(session_id: str, source: ResumeSessionSource = "local") -> str:
+    if source == "remote":
+        return session_id[-SHORT_SESSION_ID_LEN:]
+    return session_id[:SHORT_SESSION_ID_LEN]
+
+
+_ACTIVE_STATUSES = {
+    WorkflowExecutionStatus.RUNNING,
+    WorkflowExecutionStatus.RETRYING_AFTER_ERROR,
+}
+
+
+@dataclass(frozen=True)
+class ResumeSessionInfo:
+    session_id: str
+    source: ResumeSessionSource
+    cwd: str
+    title: str | None
+    end_time: str | None
+    status: str | None = None
+
+    @property
+    def option_id(self) -> str:
+        return f"{self.source}:{self.session_id}"
+
+
+def list_local_resume_sessions(
+    config: VibeConfig, cwd: str | None
+) -> list[ResumeSessionInfo]:
+    return [
+        ResumeSessionInfo(
+            session_id=session["session_id"],
+            source="local",
+            cwd=session["cwd"],
+            title=session.get("title"),
+            end_time=session.get("end_time"),
+        )
+        for session in SessionLoader.list_sessions(config.session_logging, cwd=cwd)
+    ]
+
+
+async def list_remote_resume_sessions(config: VibeConfig) -> list[ResumeSessionInfo]:
+    if not config.nuage_enabled or not config.nuage_api_key:
+        logger.debug("Remote resume listing skipped: missing Nuage configuration")
+        return []
+
+    async with WorkflowsClient(
+        base_url=config.nuage_base_url,
+        api_key=config.nuage_api_key,
+        timeout=config.api_timeout,
+    ) as client:
+        response = await client.get_workflow_runs(
+            workflow_identifier=config.nuage_workflow_id, page_size=50
+        )
+
+    sessions: list[ResumeSessionInfo] = []
+    for execution in response.executions:
+        if execution.status not in _ACTIVE_STATUSES:
+            continue
+
+        sessions.append(
+            ResumeSessionInfo(
+                session_id=execution.execution_id,
+                source="remote",
+                cwd="",
+                title="Vibe Nuage",
+                end_time=(
+                    execution.end_time.isoformat()
+                    if execution.end_time
+                    else execution.start_time.isoformat()
+                ),
+                status=execution.status,
+            )
+        )
+
+    logger.debug("Remote resume listing filtered sessions: %d", len(sessions))
+    return sessions