Azure Container Apps session execute new api-version

src/containerapp/HISTORY.rst

@@ -7,6 +7,7 @@ upcoming
 * 'az containerapp create': Fix Role assignment error when the default Azure Container Registry could not be found
 * Upgrade api-version to 2024-10-02-preview
 * 'az containerapp create/update': `--yaml` support property pollingInterval and cooldownPeriod
+* 'az containerapp session code-interpreter': `--path` support upload-file/list-files/show-file-content/show-file-metadata/delete-file for code interpreter sessions
 
 1.0.0b4
 ++++++

src/containerapp/azext_containerapp/_clients.py

@@ -27,7 +27,6 @@
 logger = get_logger(__name__)
 
 PREVIEW_API_VERSION = "2024-10-02-preview"
-SESSION_API_VERSION = "2024-02-02-preview"
 POLLING_TIMEOUT = 1500  # how many seconds before exiting
 POLLING_SECONDS = 2  # how many seconds between requests
 POLLING_TIMEOUT_FOR_MANAGED_CERTIFICATE = 1500  # how many seconds before exiting
@@ -1018,7 +1017,7 @@ def list(cls, cmd, resource_group_name, environment_name):
 
 
 class SessionPoolPreviewClient():
-    api_version = SESSION_API_VERSION
+    api_version = PREVIEW_API_VERSION
 
     @classmethod
     def create(cls, cmd, resource_group_name, name, session_pool_envelope, no_wait=False):
@@ -1157,11 +1156,11 @@ def list_by_subscription(cls, cmd, formatter=lambda x: x):
 
 
 class SessionCodeInterpreterPreviewClient():
-    api_version = SESSION_API_VERSION
+    api_version = PREVIEW_API_VERSION
 
     @classmethod
     def execute(cls, cmd, identifier, code_interpreter_envelope, session_pool_endpoint, no_wait=False):
-        url_fmt = "{}/code/execute?identifier={}&api-version={}"
+        url_fmt = "{}/executions?identifier={}&api-version={}"
         request_url = url_fmt.format(
             session_pool_endpoint,
             identifier,
@@ -1180,10 +1179,11 @@ def execute(cls, cmd, identifier, code_interpreter_envelope, session_pool_endpoi
         return r.json()
 
     @classmethod
-    def upload(cls, cmd, identifier, filepath, session_pool_endpoint, no_wait=False):
-        url_fmt = "{}/files/upload?identifier={}&api-version={}"
+    def upload(cls, cmd, identifier, filepath, path, session_pool_endpoint, no_wait=False):
+        url_fmt = "{}/files?{}identifier={}&api-version={}"
         request_url = url_fmt.format(
             session_pool_endpoint,
+            "path=" + path + "&" if path is not None else "",
             identifier,
             cls.api_version)
 
@@ -1220,11 +1220,12 @@ def upload(cls, cmd, identifier, filepath, session_pool_endpoint, no_wait=False)
         return r.json()
 
     @classmethod
-    def show_file_content(cls, cmd, identifier, filename, session_pool_endpoint):
-        url_fmt = "{}/files/content/{}?identifier={}&api-version={}"
+    def show_file_content(cls, cmd, identifier, filename, path, session_pool_endpoint):
+        url_fmt = "{}/files/{}/content?{}identifier={}&api-version={}"
         request_url = url_fmt.format(
             session_pool_endpoint,
             filename,
+            "path=" + path + "&" if path is not None else "",
             identifier,
             cls.api_version)
 
@@ -1235,11 +1236,12 @@ def show_file_content(cls, cmd, identifier, filename, session_pool_endpoint):
         return json.dumps(r.content.decode())
 
     @classmethod
-    def show_file_metadata(cls, cmd, identifier, filename, session_pool_endpoint):
-        url_fmt = "{}/files/{}?identifier={}&api-version={}"
+    def show_file_metadata(cls, cmd, identifier, filename, path, session_pool_endpoint):
+        url_fmt = "{}/files/{}?{}identifier={}&api-version={}"
         request_url = url_fmt.format(
             session_pool_endpoint,
             filename,
+            "path=" + path + "&" if path is not None else "",
             identifier,
             cls.api_version)
         logger.warning(request_url)
@@ -1248,11 +1250,12 @@ def show_file_metadata(cls, cmd, identifier, filename, session_pool_endpoint):
         return r.json()
 
     @classmethod
-    def delete_file(cls, cmd, identifier, filename, session_pool_endpoint, no_wait=False):
-        url_fmt = "{}/files/{}?identifier={}&api-version={}"
+    def delete_file(cls, cmd, identifier, filename, path, session_pool_endpoint, no_wait=False):
+        url_fmt = "{}/files/{}?{}identifier={}&api-version={}"
         request_url = url_fmt.format(
             session_pool_endpoint,
             filename,
+            "path=" + path + "&" if path is not None else "",
             identifier,
             cls.api_version)
 

src/containerapp/azext_containerapp/_models.py

@@ -308,13 +308,10 @@
 }
 
 SessionCodeInterpreterExecution = {
-    "properties": {
-        "identifier": None,
-        "codeInputType": None,
-        "executionType": None,
-        "code": None,
-        "timeoutInSeconds": None
-    }
+    "codeInputType": None,
+    "executionType": None,
+    "code": None,
+    "timeoutInSeconds": None
 }
 
 DaprComponentResiliency = {

src/containerapp/azext_containerapp/_params.py

@@ -435,7 +435,7 @@ def load_arguments(self, _):
     with self.argument_context('containerapp session code-interpreter', arg_group='file') as c:
         c.argument('filename', help="The file to delete or show from the session")
         c.argument('filepath', help="The local path to the file to upload to the session")
-        c.argument('path', help="The path to list files from the session")
+        c.argument('path', help="The path of the file from the session")
 
     with self.argument_context('containerapp session code-interpreter', arg_group='execute') as c:
         c.argument('code', help="The code to execute in the code interpreter session")

src/containerapp/azext_containerapp/containerapp_session_code_interpreter_decorator.py

@@ -83,16 +83,14 @@ def validate_arguments(self):
                                      "sessionPools")
 
     def construct_payload(self):
-        self.session_code_interpreter_def["properties"]["identifier"] = self.get_argument_identifier()
-
-        self.session_code_interpreter_def["properties"]["codeInputType"] = "inline"
-        self.session_code_interpreter_def["properties"]["executionType"] = "synchronous"
+        self.session_code_interpreter_def["codeInputType"] = "inline"
+        self.session_code_interpreter_def["executionType"] = "synchronous"
 
         if self.get_argument_timeout_in_seconds() is not None:
-            self.session_code_interpreter_def["properties"]["timeoutInSeconds"] = self.get_argument_timeout_in_seconds()
+            self.session_code_interpreter_def["timeoutInSeconds"] = self.get_argument_timeout_in_seconds()
         else:
-            self.session_code_interpreter_def["properties"]["timeoutInSeconds"] = 60
-        self.session_code_interpreter_def["properties"]["code"] = self.get_argument_code()
+            self.session_code_interpreter_def["timeoutInSeconds"] = 60
+        self.session_code_interpreter_def["code"] = self.get_argument_code()
 
     def execute(self):
         try:
@@ -111,6 +109,7 @@ def upload(self):
                 cmd=self.cmd,
                 identifier=self.get_argument_identifier(),
                 filepath=self.get_argument_filepath(),
+                path=self.get_argument_path(),
                 session_pool_endpoint=self.get_sessionpool_endpoint(),
                 no_wait=self.get_argument_no_wait())
         except Exception as e:
@@ -122,6 +121,7 @@ def show_file_content(self):
                 cmd=self.cmd,
                 identifier=self.get_argument_identifier(),
                 filename=self.get_argument_filename(),
+                path=self.get_argument_path(),
                 session_pool_endpoint=self.get_sessionpool_endpoint())
         except Exception as e:
             handle_raw_exception(e)
@@ -132,6 +132,7 @@ def show_file_metadata(self):
                 cmd=self.cmd,
                 identifier=self.get_argument_identifier(),
                 filename=self.get_argument_filename(),
+                path=self.get_argument_path(),
                 session_pool_endpoint=self.get_sessionpool_endpoint())
         except Exception as e:
             handle_raw_exception(e)
@@ -152,6 +153,7 @@ def delete_file(self):
                 cmd=self.cmd,
                 identifier=self.get_argument_identifier(),
                 filename=self.get_argument_filename(),
+                path=self.get_argument_path(),
                 session_pool_endpoint=self.get_sessionpool_endpoint(),
                 no_wait=self.get_argument_no_wait())
         except Exception as e:

src/containerapp/azext_containerapp/custom.py

@@ -2989,6 +2989,7 @@ def upload_session_code_interpreter(cmd,
                                     resource_group_name,
                                     identifier,
                                     filepath,
+                                    path=None,
                                     session_pool_location=None):
     raw_parameters = locals()
     session_code_interpreter_decorator = SessionCodeInterpreterCommandsPreviewDecorator(
@@ -3009,6 +3010,7 @@ def show_file_content_session_code_interpreter(cmd,
                                                resource_group_name,
                                                identifier,
                                                filename,
+                                               path=None,
                                                session_pool_location=None):
     raw_parameters = locals()
     session_code_interpreter_decorator = SessionCodeInterpreterCommandsPreviewDecorator(
@@ -3029,6 +3031,7 @@ def show_file_metadata_session_code_interpreter(cmd,
                                                 resource_group_name,
                                                 identifier,
                                                 filename,
+                                                path=None,
                                                 session_pool_location=None):
     raw_parameters = locals()
     session_code_interpreter_decorator = SessionCodeInterpreterCommandsPreviewDecorator(
@@ -3069,6 +3072,7 @@ def delete_file_session_code_interpreter(cmd,
                                          resource_group_name,
                                          identifier,
                                          filename,
+                                         path=None,
                                          session_pool_location=None):
     raw_parameters = locals()
     session_code_interpreter_decorator = SessionCodeInterpreterCommandsPreviewDecorator(