recce-nightly 1.20.0.20250918__py3-none-any.whl → 1.20.0.20250921__py3-none-any.whl

recce/state/cloud.py

@@ -75,15 +75,15 @@ class CloudStateLoader(RecceStateLoader):
                     'Please provide a share URL in the command argument with option "--share-url <share-url>"'
                 )
                 return False
-        elif self.catalog == "snapshot":
+        elif self.catalog == "session":
             if self.cloud_options.get("api_token") is None:
                 self.error_message = RECCE_API_TOKEN_MISSING.error_message
                 self.hint_message = RECCE_API_TOKEN_MISSING.hint_message
                 return False
-            if self.cloud_options.get("snapshot_id") is None:
-                self.error_message = "No snapshot ID is provided for the snapshot catalog."
+            if self.cloud_options.get("session_id") is None:
+                self.error_message = "No session ID is provided for the session catalog."
                 self.hint_message = (
-                    'Please provide a snapshot ID in the command argument with option "--snapshot-id <snapshot-id>"'
+                    'Please provide a session ID in the command argument with option "--session-id <session-id>"'
                 )
                 return False
         return True
@@ -106,8 +106,8 @@ class CloudStateLoader(RecceStateLoader):
             return self._load_state_from_github()
         elif self.catalog == "preview":
             return self._load_state_from_preview()
-        elif self.catalog == "snapshot":
-            return self._load_state_from_snapshot(), None
+        elif self.catalog == "session":
+            return self._load_state_from_session(), None
         else:
             raise RecceException(f"Unsupported catalog type: {self.catalog}")
 
@@ -196,37 +196,37 @@ class CloudStateLoader(RecceStateLoader):
 
             return RecceState.from_file(tmp.name, file_type=file_type)
 
-    def _load_state_from_snapshot(self) -> RecceState:
+    def _load_state_from_session(self) -> RecceState:
         """
-        Load state from snapshot by:
-        1. Get snapshot info
-        2. Download artifacts for both base and current snapshots
+        Load state from session by:
+        1. Get session info
+        2. Download artifacts for both base and current sessions
         3. Download recce_state if available, otherwise create empty state with artifacts
         """
-        if self.snapshot_id is None:
-            raise RecceException("Cannot load the snapshot state from Recce Cloud. No snapshot ID is provided.")
+        if self.session_id is None:
+            raise RecceException("Cannot load the session state from Recce Cloud. No session ID is provided.")
 
-        # 1. Get snapshot information
-        logger.debug(f"Getting snapshot {self.snapshot_id}")
-        snapshot = self.recce_cloud.get_snapshot(self.snapshot_id)
+        # 1. Get session information
+        logger.debug(f"Getting session {self.session_id}")
+        session = self.recce_cloud.get_session(self.session_id)
 
-        org_id = snapshot.get("org_id")
-        project_id = snapshot.get("project_id")
+        org_id = session.get("org_id")
+        project_id = session.get("project_id")
 
         if not org_id or not project_id:
-            raise RecceException(f"Snapshot {self.snapshot_id} does not belong to a valid organization or project.")
+            raise RecceException(f"Session {self.session_id} does not belong to a valid organization or project.")
 
-        # 2. Download manifests and catalogs for both snapshots
-        logger.debug(f"Downloading current snapshot artifacts for {self.snapshot_id}")
-        current_artifacts = self._download_snapshot_artifacts(self.recce_cloud, org_id, project_id, self.snapshot_id)
+        # 2. Download manifests and catalogs for both session
+        logger.debug(f"Downloading current session artifacts for {self.session_id}")
+        current_artifacts = self._download_session_artifacts(self.recce_cloud, org_id, project_id, self.session_id)
 
-        logger.debug(f"Downloading base snapshot artifacts for project {project_id}")
-        base_artifacts = self._download_base_snapshot_artifacts(self.recce_cloud, org_id, project_id)
+        logger.debug(f"Downloading base session artifacts for project {project_id}")
+        base_artifacts = self._download_base_session_artifacts(self.recce_cloud, org_id, project_id)
 
         # 3. Try to download existing recce_state, otherwise create new state
         try:
-            logger.debug(f"Downloading recce_state for snapshot {self.snapshot_id}")
-            state = self._download_snapshot_recce_state(self.recce_cloud, org_id, project_id, self.snapshot_id)
+            logger.debug(f"Downloading recce_state for session {self.session_id}")
+            state = self._download_session_recce_state(self.recce_cloud, org_id, project_id, self.session_id)
         except Exception as e:
             logger.debug(f"No existing recce_state found, creating new state: {e}")
             state = RecceState()
@@ -237,12 +237,12 @@ class CloudStateLoader(RecceStateLoader):
 
         return state
 
-    def _download_snapshot_artifacts(self, recce_cloud, org_id: str, project_id: str, snapshot_id: str) -> dict:
-        """Download manifest and catalog for a snapshot, return JSON data directly."""
+    def _download_session_artifacts(self, recce_cloud, org_id: str, project_id: str, session_id: str) -> dict:
+        """Download manifest and catalog for a session, return JSON data directly."""
         import requests
 
         # Get download URLs
-        presigned_urls = recce_cloud.get_download_urls_by_snapshot_id(org_id, project_id, snapshot_id)
+        presigned_urls = recce_cloud.get_download_urls_by_session_id(org_id, project_id, session_id)
 
         artifacts = {}
 
@@ -251,40 +251,40 @@ class CloudStateLoader(RecceStateLoader):
         if response.status_code == 200:
             artifacts["manifest"] = response.json()
         else:
-            raise RecceException(f"Failed to download manifest for snapshot {snapshot_id}")
+            raise RecceException(f"Failed to download manifest for session {session_id}")
 
         # Download catalog
         response = requests.get(presigned_urls["catalog_url"])
         if response.status_code == 200:
             artifacts["catalog"] = response.json()
         else:
-            raise RecceException(f"Failed to download catalog for snapshot {snapshot_id}")
+            raise RecceException(f"Failed to download catalog for session {session_id}")
 
         return artifacts
 
-    def _download_snapshot_recce_state(self, recce_cloud, org_id: str, project_id: str, snapshot_id: str) -> RecceState:
-        """Download recce_state for a snapshot."""
+    def _download_session_recce_state(self, recce_cloud, org_id: str, project_id: str, session_id: str) -> RecceState:
+        """Download recce_state for a session."""
         # Get download URLs (now includes recce_state_url)
-        presigned_urls = recce_cloud.get_download_urls_by_snapshot_id(org_id, project_id, snapshot_id)
+        presigned_urls = recce_cloud.get_download_urls_by_session_id(org_id, project_id, session_id)
         recce_state_url = presigned_urls.get("recce_state_url")
 
         if not recce_state_url:
-            raise RecceException(f"No recce_state_url found for snapshot {snapshot_id}")
+            raise RecceException(f"No recce_state_url found for session {session_id}")
 
         # Reuse the existing download method
         state = self._download_state_from_url(recce_state_url, SupportedFileTypes.FILE)
 
         if state is None:
-            raise RecceException(f"Failed to download recce_state for snapshot {snapshot_id}")
+            raise RecceException(f"Failed to download recce_state for session {session_id}")
 
         return state
 
-    def _download_base_snapshot_artifacts(self, recce_cloud, org_id: str, project_id: str) -> dict:
-        """Download manifest and catalog for the base snapshot, return JSON data directly."""
+    def _download_base_session_artifacts(self, recce_cloud, org_id: str, project_id: str) -> dict:
+        """Download manifest and catalog for the base session, return JSON data directly."""
         import requests
 
-        # Get download URLs for base snapshot
-        presigned_urls = recce_cloud.get_base_snapshot_download_urls(org_id, project_id)
+        # Get download URLs for base session
+        presigned_urls = recce_cloud.get_base_session_download_urls(org_id, project_id)
 
         artifacts = {}
 
@@ -293,14 +293,14 @@ class CloudStateLoader(RecceStateLoader):
         if response.status_code == 200:
             artifacts["manifest"] = response.json()
         else:
-            raise RecceException(f"Failed to download base snapshot manifest for project {project_id}")
+            raise RecceException(f"Failed to download base session manifest for project {project_id}")
 
         # Download catalog
         response = requests.get(presigned_urls["catalog_url"])
         if response.status_code == 200:
             artifacts["catalog"] = response.json()
         else:
-            raise RecceException(f"Failed to download base snapshot catalog for project {project_id}")
+            raise RecceException(f"Failed to download base session catalog for project {project_id}")
 
         return artifacts
 
@@ -318,8 +318,8 @@ class CloudStateLoader(RecceStateLoader):
             return self._export_state_to_github()
         elif self.catalog == "preview":
             return self._export_state_to_preview()
-        elif self.catalog == "snapshot":
-            return self._export_state_to_snapshot()
+        elif self.catalog == "session":
+            return self._export_state_to_session()
         else:
             raise RecceException(f"Unsupported catalog type: {self.catalog}")
 
@@ -374,25 +374,25 @@ class CloudStateLoader(RecceStateLoader):
             logger.warning(message)
         return message, None
 
-    def _export_state_to_snapshot(self) -> Tuple[Union[str, None], None]:
-        """Export state to snapshot (upload recce_state with empty artifacts)."""
-        if self.snapshot_id is None:
-            raise RecceException("Cannot export state to snapshot. No snapshot ID is provided.")
+    def _export_state_to_session(self) -> Tuple[Union[str, None], None]:
+        """Export state to session (upload recce_state with empty artifacts)."""
+        if self.session_id is None:
+            raise RecceException("Cannot export state to session. No session ID is provided.")
 
-        # Get snapshot information
-        snapshot = self.recce_cloud.get_snapshot(self.snapshot_id)
-        org_id = snapshot.get("org_id")
-        project_id = snapshot.get("project_id")
+        # Get session information
+        session = self.recce_cloud.get_session(self.session_id)
+        org_id = session.get("org_id")
+        project_id = session.get("project_id")
 
         if not org_id or not project_id:
-            raise RecceException(f"Snapshot {self.snapshot_id} does not belong to a valid organization or project.")
+            raise RecceException(f"Session {self.session_id} does not belong to a valid organization or project.")
 
         # Get upload URLs (now includes recce_state_url)
-        presigned_urls = self.recce_cloud.get_upload_urls_by_snapshot_id(org_id, project_id, self.snapshot_id)
+        presigned_urls = self.recce_cloud.get_upload_urls_by_session_id(org_id, project_id, self.session_id)
         recce_state_url = presigned_urls.get("recce_state_url")
 
         if not recce_state_url:
-            raise RecceException(f"No recce_state_url found for snapshot {self.snapshot_id}")
+            raise RecceException(f"No recce_state_url found for session {self.session_id}")
 
         # Create a copy of the state with empty artifacts for upload
         upload_state = RecceState()

recce/state/state_loader.py

@@ -37,7 +37,7 @@ class RecceStateLoader(ABC):
         self.pr_info = None
         self.catalog: Literal["github", "preview", "snapshot"] = "github"
         self.share_id = None
-        self.snapshot_id = None
+        self.session_id = None
 
         if self.cloud_mode:
             if self.cloud_options.get("github_token"):
@@ -48,9 +48,9 @@ class RecceStateLoader(ABC):
                 if self.pr_info.id is None:
                     raise RecceException("Cannot get the pull request information from GitHub.")
             elif self.cloud_options.get("api_token"):
-                if self.cloud_options.get("snapshot_id"):
-                    self.catalog = "snapshot"
-                    self.snapshot_id = self.cloud_options.get("snapshot_id")
+                if self.cloud_options.get("session_id"):
+                    self.catalog = "session"
+                    self.session_id = self.cloud_options.get("session_id")
                 else:
                     self.catalog = "preview"
                     self.share_id = self.cloud_options.get("share_id")

recce/util/recce_cloud.py

@@ -244,48 +244,46 @@ class RecceCloud:
             logger.warning(f"Failed to set Onboarding State in Recce Cloud. Reason: {str(e)}")
         return
 
-    def get_snapshot(self, snapshot_id: str):
-        api_url = f"{self.base_url_v2}/snapshots/{snapshot_id}"
+    def get_session(self, session_id: str):
+        api_url = f"{self.base_url_v2}/sessions/{session_id}"
         response = self._request("GET", api_url)
         if response.status_code == 403:
             return {"status": "error", "message": response.json().get("detail")}
         if response.status_code != 200:
             raise RecceCloudException(
-                message="Failed to get snapshot from Recce Cloud.",
+                message="Failed to get session from Recce Cloud.",
                 reason=response.text,
                 status_code=response.status_code,
             )
         data = response.json()
         if data["success"] is not True:
             raise RecceCloudException(
-                message="Failed to get snapshot from Recce Cloud.",
+                message="Failed to get session from Recce Cloud.",
                 reason=data.get("message", "Unknown error"),
                 status_code=response.status_code,
             )
-        return data["snapshot"]
+        return data["session"]
 
-    def update_snapshot(self, org_id: str, project_id: str, snapshot_id: str, adapter_type: str):
-        api_url = f"{self.base_url_v2}/organizations/{org_id}/projects/{project_id}/snapshots/{snapshot_id}"
+    def update_session(self, org_id: str, project_id: str, session_id: str, adapter_type: str):
+        api_url = f"{self.base_url_v2}/organizations/{org_id}/projects/{project_id}/sessions/{session_id}"
         data = {"adapter_type": adapter_type}
         response = self._request("PATCH", api_url, json=data)
         if response.status_code == 403:
             return {"status": "error", "message": response.json().get("detail")}
         if response.status_code != 200:
             raise RecceCloudException(
-                message="Failed to update snapshot in Recce Cloud.",
+                message="Failed to update session in Recce Cloud.",
                 reason=response.text,
                 status_code=response.status_code,
             )
         return response.json()
 
-    def get_download_urls_by_snapshot_id(self, org_id: str, project_id: str, snapshot_id: str) -> dict[str, str]:
-        api_url = (
-            f"{self.base_url_v2}/organizations/{org_id}/projects/{project_id}/snapshots/{snapshot_id}/download-url"
-        )
+    def get_download_urls_by_session_id(self, org_id: str, project_id: str, session_id: str) -> dict[str, str]:
+        api_url = f"{self.base_url_v2}/organizations/{org_id}/projects/{project_id}/sessions/{session_id}/download-url"
         response = self._request("GET", api_url)
         if response.status_code != 200:
             raise RecceCloudException(
-                message="Failed to download snapshot from Recce Cloud.",
+                message="Failed to download session from Recce Cloud.",
                 reason=response.text,
                 status_code=response.status_code,
             )
@@ -302,13 +300,13 @@ class RecceCloud:
             presigned_urls[key] = self._replace_localhost_with_docker_internal(url)
         return presigned_urls
 
-    def get_base_snapshot_download_urls(self, org_id: str, project_id: str) -> dict[str, str]:
-        """Get download URLs for the base snapshot of a project."""
-        api_url = f"{self.base_url_v2}/organizations/{org_id}/projects/{project_id}/base-snapshot/download-url"
+    def get_base_session_download_urls(self, org_id: str, project_id: str) -> dict[str, str]:
+        """Get download URLs for the base session of a project."""
+        api_url = f"{self.base_url_v2}/organizations/{org_id}/projects/{project_id}/base-session/download-url"
         response = self._request("GET", api_url)
         if response.status_code != 200:
             raise RecceCloudException(
-                message="Failed to download base snapshot from Recce Cloud.",
+                message="Failed to download base session from Recce Cloud.",
                 reason=response.text,
                 status_code=response.status_code,
             )
@@ -325,12 +323,12 @@ class RecceCloud:
             presigned_urls[key] = self._replace_localhost_with_docker_internal(url)
         return presigned_urls
 
-    def get_upload_urls_by_snapshot_id(self, org_id: str, project_id: str, snapshot_id: str) -> dict[str, str]:
-        api_url = f"{self.base_url_v2}/organizations/{org_id}/projects/{project_id}/snapshots/{snapshot_id}/upload-url"
+    def get_upload_urls_by_session_id(self, org_id: str, project_id: str, session_id: str) -> dict[str, str]:
+        api_url = f"{self.base_url_v2}/organizations/{org_id}/projects/{project_id}/sessions/{session_id}/upload-url"
         response = self._request("GET", api_url)
         if response.status_code != 200:
             raise RecceCloudException(
-                message="Failed to get upload URLs for snapshot from Recce Cloud.",
+                message="Failed to get upload URLs for session from Recce Cloud.",
                 reason=response.text,
                 status_code=response.status_code,
             )

{recce_nightly-1.20.0.20250918.dist-info → recce_nightly-1.20.0.20250921.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: recce-nightly
-Version: 1.20.0.20250918
+Version: 1.20.0.20250921
 Summary: Environment diff tool for dbt
 Home-page: https://github.com/InfuseAI/recce
 Author: InfuseAI Dev Team

{recce_nightly-1.20.0.20250918.dist-info → recce_nightly-1.20.0.20250921.dist-info}/RECORD

@@ -1,7 +1,7 @@
-recce/VERSION,sha256=ROmUJbh-DrHJsAgTturORju_cuPBNgPXukHuvYXiw04,16
+recce/VERSION,sha256=927IbuVVxqKUqOGffZnfB1fLn7HHSPlGOrpZ2KnDc9w,16
 recce/__init__.py,sha256=yNb0QT-yoStex0VZALNJvUwtPLommoVCStcow31guqo,2392
-recce/artifact.py,sha256=Y4qC3ymckCFSBRQ_amvhp1kwE6QrXHFzno4b_mMAfZs,9252
-recce/cli.py,sha256=ncHiK7cSkHiVXZ8FiXPAYqEc1zMwxPK-ykue56K5Ots,45170
+recce/artifact.py,sha256=pwI377OgHIbb8HI9k-2h1K7byO2_VXfO63JXs1Ws-24,9229
+recce/cli.py,sha256=Ro3vvno_korHXZwT_putOJJSNLtuvf7fJWfhCbHAfe8,46109
 recce/config.py,sha256=A4CbKcIdwQ4A9Q2ba4riHUshVQw1D-qDelMOdlt2khU,4661
 recce/connect_to_cloud.py,sha256=_SkX2pdyXa9FNyaHvItyYVPF2nZxy2JnCyd_o_psh2Y,4750
 recce/core.py,sha256=MtBWxemvCQDdUITwkU2JyuQYqcjlA0a76M8Kg53ECxQ,10754
@@ -11,7 +11,7 @@ recce/git.py,sha256=8Eg-6NzL-KjA3rT-ibbAyaCwGlzV0JqH3yGikrJNMDA,2344
 recce/github.py,sha256=PEpM6ZRiinsPbXSWj4aJCKbZrN1jUXzpzAfJq_CGah4,7420
 recce/pull_request.py,sha256=aW0B1NE2LUKTam1S4TQ7smXB9KLE1DV8GnyBqNXA6j8,3832
 recce/run.py,sha256=PNafwUUJdIG8b1o0y1QuvjACpA3E-5a3keH45CrWHN0,14044
-recce/server.py,sha256=TYkPHaz_XamNhKpL0ldkVJApFq7c6gG89wtn0F1sP34,22733
+recce/server.py,sha256=2xy9AkHJx2c_PNtgmg4QwbbbVLfBP68ZGsskEzHtbRQ,22730
 recce/summary.py,sha256=Mbxvxr9KazR5o9icqhhjiGHsoAiWxQU4PdN7HytBJ1c,19154
 recce/adapter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 recce/adapter/base.py,sha256=T_JNeLHgiHSaegw-DbrvHOaYjMyZcjj2Qtg5cWh_fco,3548
@@ -23,13 +23,13 @@ recce/apis/check_api.py,sha256=KMCXSMl1qqzx2jQgRqCrD4j_cY3EHBbM3H2-t-6saAU,6227
 recce/apis/check_func.py,sha256=gktbCcyk3WGvWRJJ-wDnwv7NrIny2nTHWLl1-kdiVRo,4183
 recce/apis/run_api.py,sha256=eOaxOxXDkH59uqGCd4blld7edavUx7JU_DCd2WAYrL8,3416
 recce/apis/run_func.py,sha256=6wC8TDU-h7TLr2VZH7HNsWaUVlQ9HBN5N_dwqfi4lMY,7440
-recce/data/404.html,sha256=sjEskTB5D1VSh8FA1rPxcVfmO1TCtvuwHZsWUm6oiig,59087
+recce/data/404.html,sha256=AkLP52yWPUFEj7Bjv8H1-ohLuofIUB7mXlrnACMTgr4,59087
 recce/data/auth_callback.html,sha256=H-XfdlAFiw5VU2RpKBVQbwh1AIqJrPHrFA0S09nNJZA,94779
 recce/data/favicon.ico,sha256=B2mBumUOnzvUrXrqNkrc5QfdDXjzEXRcWkWur0fJ6sM,2565
-recce/data/index.html,sha256=smcduJBo1hmRT6tDCpAxQxyctexmqCrfexiUJuEw5tA,77450
-recce/data/index.txt,sha256=5KPOUDFO6rISVNcj4TYC7Bv7y5DTxDu4ewSNEz0_Rm8,6409
-recce/data/_next/static/ZUbzMSlc3CArT7vsmEueJ/_buildManifest.js,sha256=avQsL1oPzYwcdZvar5cABpd9QZyaNp5U46Y9ET0fq24,544
-recce/data/_next/static/ZUbzMSlc3CArT7vsmEueJ/_ssgManifest.js,sha256=Z49s4suAsf5y_GfnQSvm4qtq2ggxEbZPfEDTXjy6XgA,80
+recce/data/index.html,sha256=njKsbzX3bAMWX-vSgKzglqI8WYtwh8_Onbkv4_CXHWM,77450
+recce/data/index.txt,sha256=TkkxXfVjA-exikwv1Ly33RO_PDGQNJ8jOsYfb2AzuT4,6409
+recce/data/_next/static/7VDijMrfP7uONDsXVHgCK/_buildManifest.js,sha256=avQsL1oPzYwcdZvar5cABpd9QZyaNp5U46Y9ET0fq24,544
+recce/data/_next/static/7VDijMrfP7uONDsXVHgCK/_ssgManifest.js,sha256=Z49s4suAsf5y_GfnQSvm4qtq2ggxEbZPfEDTXjy6XgA,80
 recce/data/_next/static/chunks/068b80ea-5225768a06b94b6f.js,sha256=9-CgOMA67KPg0s3LBhBMLS7nl5icw5BMDhs8Zk-TjO4,4628
 recce/data/_next/static/chunks/0ddaf06c-47655700b71b89f7.js,sha256=ZAiDuPV8ad-3MSwYva4LCOLRFOyaSENn2vBB6ZSU3So,696
 recce/data/_next/static/chunks/12f8fac4-10d6e05f7330d12b.js,sha256=79b2y12TyKWM49nc8dsZRwojyWcC7Yz-P7-f3wBfy9s,152792
@@ -62,7 +62,7 @@ recce/data/_next/static/chunks/main-bc07477fd7b7c32d.js,sha256=Vn4NGVdiaYAdp8ryF
 recce/data/_next/static/chunks/polyfills-42372ed130431b0a.js,sha256=CXPB1kyIrcjjyVBBDLWLKI9yEY1ZZbeASUON648vloM,112594
 recce/data/_next/static/chunks/webpack-364bf2670cd39ec1.js,sha256=A9nbzHfxYRScJi-DoXyjmjrhk0ZNGZvVESrMt3_2J_E,3580
 recce/data/_next/static/chunks/app/layout-24244460abfd34dd.js,sha256=2uyuWecojLN2_lZd56j_Cdz0msiHe8CRhNBz4aGccfc,1411
-recce/data/_next/static/chunks/app/page-8b1eb61c94f06c1b.js,sha256=2Q7RLMBc5e3VUL_RPCGzobcG5X3iiIn_-Uwy3wY_4xA,168955
+recce/data/_next/static/chunks/app/page-9f7291a41c6274cd.js,sha256=C7qnj2oWooIbiJS5iyLixGYindaui2r2t5gkuwCD-30,168954
 recce/data/_next/static/chunks/app/_not-found/page-101c1a92b86f368b.js,sha256=s26MwrqBIbf4Vq5yzoKhKoJwQ2Y6ognSlvxf4B4UWFE,2678
 recce/data/_next/static/chunks/pages/_app-8096e5691e76ad38.js,sha256=tJlgoS3QWhdYC0L12-5KpE3LHgv_K3hYb4YRYMPz6xE,238
 recce/data/_next/static/chunks/pages/_error-2ce4373ab74aab6b.js,sha256=FXe38dEBW7IgNfpjJ4dg3H5SAEYgKdT5lKVYxbofIBE,221
@@ -95,11 +95,11 @@ recce/models/check.py,sha256=jjR1SGyczjrqyK-0Zas6ikLIGSgVp54lvfcQA19AniI,1588
 recce/models/run.py,sha256=QK2gvOWvko9YYhd2NLs3BPt5l4MSCZGwpzTAiqx9zJw,1161
 recce/models/types.py,sha256=9ReOyIv3rUD3_cIQfB9Rplb0L2YQuZr4kS9Zai30nB8,5234
 recce/state/__init__.py,sha256=V1FRPTQJUz-uwI3Cn8wDa5Z9bueVs86MR_1Ti4RGfPc,650
-recce/state/cloud.py,sha256=SLPBQljUS7vKjKxG0UwY4bfey1t5ZI0BCDx4DBIjH8A,25031
+recce/state/cloud.py,sha256=QHdOJjjd4wVTk3m6yhah9FppeZa6DUOBr3SJwECFpOU,24950
 recce/state/const.py,sha256=Me3uVQHi_uZysl3tfpmghPInuKF236X6LfMKVvJlCgA,827
 recce/state/local.py,sha256=bZIkl7cAfyYaGgYEr3uD_JLrtwlHBnu8_o1Qz2djQzw,1920
 recce/state/state.py,sha256=L1cEAl6S9rxTvSIc8ICNwOYoCYK5tR3M0kBo4NTcXK4,3617
-recce/state/state_loader.py,sha256=4lJYiOdlAg8h7rVBiBiffnkSyiQSmg9NejIJv4fFHTM,5984
+recce/state/state_loader.py,sha256=_4vzZ3XiIMdC7G2m2WUhGi_IIoJLzsE7KwTVUeebMc0,5979
 recce/tasks/__init__.py,sha256=b553AtDHjYROgmMePv_Hv_X3fjh4iEn11gzzpUJz6_o,610
 recce/tasks/core.py,sha256=JFYa1CfgOiRPQ7KVTwMuxJjhMB-pvCwB-xezVt-h3RU,4080
 recce/tasks/dataframe.py,sha256=03UBWwt0DFTXlaEOtnV5i_mxdRKD7UbRayewEL2Ub48,3650
@@ -122,12 +122,12 @@ recce/util/logger.py,sha256=6UgLFkRiur9jJfu2ZRdo4LUvMw4f75V-l-1HT1-sgKo,747
 recce/util/onboarding_state.py,sha256=kwFirKlfXdl5WFkR_nmilqGKFyELNcSPMqYq-by35fk,1991
 recce/util/perf_tracking.py,sha256=FjhrdbbXIgybxS_oPpFsJ9VUDR93d7bUs8VNNqpXNxw,2483
 recce/util/pydantic_model.py,sha256=KumKuyCjbTzEMsKLE4-b-eZfp0gLhYDdmVtw1-hxiJw,587
-recce/util/recce_cloud.py,sha256=YSzdC1GYcoy-BbPgnouocp9qqt1LZ-HDWKcn1ulmoT4,15011
+recce/util/recce_cloud.py,sha256=45-HmG3yW51VLssmwQQRZjoDDM9LwEq2UHTMMykX16Y,14961
 recce/util/singleton.py,sha256=1cU99I0f9tjuMQLMJyLsK1oK3fZJMsO5-TbRHAMXqds,627
 recce/yaml/__init__.py,sha256=EgXYlFeJZchatUClRDXbIC5Oqb2_nBvB2NqItYVihio,1292
-recce_nightly-1.20.0.20250918.dist-info/licenses/LICENSE,sha256=CQjjMy9aYPhfe8xG_bcpIfKtNkdxLZ5IOb8oPygtUhY,11343
+recce_nightly-1.20.0.20250921.dist-info/licenses/LICENSE,sha256=CQjjMy9aYPhfe8xG_bcpIfKtNkdxLZ5IOb8oPygtUhY,11343
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/test_cli.py,sha256=4N_F_lg00fGBuK3I9EyaF2Doi7PpKCrjqbuJ5YI1KqE,5467
+tests/test_cli.py,sha256=CFtP4eJrtcHwpzoEgJAyveQcqxAmDQuWXPrLc_Iny7Q,6626
 tests/test_config.py,sha256=ODDFe_XF6gphmSmmc422dGLBaCCmG-IjDzTkD5SJsJE,1557
 tests/test_connect_to_cloud.py,sha256=b2fgV8L1iQBdEwh6RumMsIIyYg7GtMOMRz1dvE3WRPg,3059
 tests/test_core.py,sha256=WgCFm8Au3YQI-V5UbZ6LA8irMNHRc7NWutIq2Mny0LE,6242
@@ -153,8 +153,8 @@ tests/tasks/test_row_count.py,sha256=21PaP2aq-x8-pqwzWHRT1sixhQ8g3CQNRWOZTTmbK0s
 tests/tasks/test_schema.py,sha256=7ds4Vx8ixaiIWDR49Lvjem4xlPkRP1cXazDRY3roUak,3121
 tests/tasks/test_top_k.py,sha256=YR_GS__DJsbDlQVaEEdJvNQ3fh1VmV5Nb3G7lb0r6YM,1779
 tests/tasks/test_valuediff.py,sha256=_xQJGgxsXoy2NYk_Z6Hsw2FlVh6zk2nN_iUueyRN1e8,2046
-recce_nightly-1.20.0.20250918.dist-info/METADATA,sha256=BGaD2GcXnlhBXtvF3Sh5vRGsIhcd0-nb2nE4A3WAFh4,9366
-recce_nightly-1.20.0.20250918.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-recce_nightly-1.20.0.20250918.dist-info/entry_points.txt,sha256=oqoY_IiwIqXbgrIsPnlqUqao2eiIeP2dprowkOlmeyg,40
-recce_nightly-1.20.0.20250918.dist-info/top_level.txt,sha256=6PKGVpf75idP0C6KEaldDzzZUauIxNu1ZDstau1pI4I,12
-recce_nightly-1.20.0.20250918.dist-info/RECORD,,
+recce_nightly-1.20.0.20250921.dist-info/METADATA,sha256=C9jUp7DvpC5WQU9voaMdyd0uwduEeSjN9hMQfxGeDHU,9366
+recce_nightly-1.20.0.20250921.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+recce_nightly-1.20.0.20250921.dist-info/entry_points.txt,sha256=oqoY_IiwIqXbgrIsPnlqUqao2eiIeP2dprowkOlmeyg,40
+recce_nightly-1.20.0.20250921.dist-info/top_level.txt,sha256=6PKGVpf75idP0C6KEaldDzzZUauIxNu1ZDstau1pI4I,12
+recce_nightly-1.20.0.20250921.dist-info/RECORD,,

tests/test_cli.py

@@ -5,6 +5,8 @@ from click.testing import CliRunner
 
 from recce.cli import run as cli_command_run
 from recce.cli import server as cli_command_server
+from recce.cli import snapshot as cli_command_snapshot
+from recce.cli import upload_session as cli_command_upload_session
 from recce.core import RecceContext
 from recce.state import CloudStateLoader
 
@@ -131,3 +133,28 @@ class TestCommandRun(TestCase):
 
         self.runner.invoke(cli_command_run, [])
         mock_cli_run.assert_called_once()
+
+
+class TestCommandUploadSession(TestCase):
+    def setUp(self):
+        self.runner = CliRunner()
+        pass
+
+    @patch("recce.cli.prepare_api_token", return_value="unittest_token")
+    @patch("recce.cli.upload_artifacts_to_session", return_value=0)
+    def test_cmd_upload_session(self, mock_upload_artifacts_to_session, mock_prepare_api_token):
+        self.runner.invoke(
+            cli_command_upload_session,
+            ["--session-id", "unittest_session", "--api-token", mock_prepare_api_token.return_value],
+        )
+        mock_upload_artifacts_to_session.assert_called_once_with(
+            "target", session_id="unittest_session", token="unittest_token", debug=False
+        )
+
+        self.runner.invoke(
+            cli_command_snapshot,
+            ["--snapshot-id", "unittest_session", "--api-token", mock_prepare_api_token.return_value],
+        )
+        mock_upload_artifacts_to_session.assert_called_once_with(
+            "target", session_id="unittest_session", token="unittest_token", debug=False
+        )