fix(sandbox): sandbox management api support passing config

Signed-off-by: Sodawyx <sodawyx@126.com>

Author: Sodawyx

agentrun/sandbox/api/__sandbox_data_async_template.py

@@ -86,22 +86,22 @@ async def create_sandbox_async(
             data["ossMountConfig"] = oss_mount_config
         if polar_fs_config is not None:
             data["polarFsConfig"] = polar_fs_config
-        return await self.post_async("/", data=data)
+        return await self.post_async("/", data=data, config=config)
 
     async def delete_sandbox_async(
         self, sandbox_id: str, config: Optional[Config] = None
     ):
         self.__refresh_access_token(sandbox_id=sandbox_id, config=config)
-        return await self.delete_async("/")
+        return await self.delete_async("/", config=config)
 
     async def stop_sandbox_async(
         self, sandbox_id: str, config: Optional[Config] = None
     ):
         self.__refresh_access_token(sandbox_id=sandbox_id, config=config)
-        return await self.post_async("/stop")
+        return await self.post_async("/stop", config=config)
 
     async def get_sandbox_async(
         self, sandbox_id: str, config: Optional[Config] = None
     ):
         self.__refresh_access_token(sandbox_id=sandbox_id, config=config)
-        return await self.get_async("/")
+        return await self.get_async("/", config=config)

agentrun/sandbox/api/sandbox_data.py

@@ -99,7 +99,7 @@ async def create_sandbox_async(
             data["ossMountConfig"] = oss_mount_config
         if polar_fs_config is not None:
             data["polarFsConfig"] = polar_fs_config
-        return await self.post_async("/", data=data)
+        return await self.post_async("/", data=data, config=config)
 
     def create_sandbox(
         self,
@@ -121,34 +121,34 @@ def create_sandbox(
             data["ossMountConfig"] = oss_mount_config
         if polar_fs_config is not None:
             data["polarFsConfig"] = polar_fs_config
-        return self.post("/", data=data)
+        return self.post("/", data=data, config=config)
 
     async def delete_sandbox_async(
         self, sandbox_id: str, config: Optional[Config] = None
     ):
         self.__refresh_access_token(sandbox_id=sandbox_id, config=config)
-        return await self.delete_async("/")
+        return await self.delete_async("/", config=config)
 
     def delete_sandbox(self, sandbox_id: str, config: Optional[Config] = None):
         self.__refresh_access_token(sandbox_id=sandbox_id, config=config)
-        return self.delete("/")
+        return self.delete("/", config=config)
 
     async def stop_sandbox_async(
         self, sandbox_id: str, config: Optional[Config] = None
     ):
         self.__refresh_access_token(sandbox_id=sandbox_id, config=config)
-        return await self.post_async("/stop")
+        return await self.post_async("/stop", config=config)
 
     def stop_sandbox(self, sandbox_id: str, config: Optional[Config] = None):
         self.__refresh_access_token(sandbox_id=sandbox_id, config=config)
-        return self.post("/stop")
+        return self.post("/stop", config=config)
 
     async def get_sandbox_async(
         self, sandbox_id: str, config: Optional[Config] = None
     ):
         self.__refresh_access_token(sandbox_id=sandbox_id, config=config)
-        return await self.get_async("/")
+        return await self.get_async("/", config=config)
 
     def get_sandbox(self, sandbox_id: str, config: Optional[Config] = None):
         self.__refresh_access_token(sandbox_id=sandbox_id, config=config)
-        return self.get("/")
+        return self.get("/", config=config)

agentrun/sandbox/client.py

@@ -637,7 +637,7 @@ async def stop_sandbox_async(
         """
         try:
             result = await self.__sandbox_data_api.stop_sandbox_async(
-                sandbox_id
+                sandbox_id, config=config
             )
 
             # 判断返回结果是否成功
@@ -676,7 +676,9 @@ def stop_sandbox(
             ServerError: 服务器错误
         """
         try:
-            result = self.__sandbox_data_api.stop_sandbox(sandbox_id)
+            result = self.__sandbox_data_api.stop_sandbox(
+                sandbox_id, config=config
+            )
 
             # 判断返回结果是否成功
             if result.get("code") != "SUCCESS":
@@ -715,7 +717,7 @@ async def delete_sandbox_async(
         """
         try:
             result = await self.__sandbox_data_api.delete_sandbox_async(
-                sandbox_id
+                sandbox_id, config=config
             )
 
             # 判断返回结果是否成功
@@ -754,7 +756,9 @@ def delete_sandbox(
             ServerError: 服务器错误
         """
         try:
-            result = self.__sandbox_data_api.delete_sandbox(sandbox_id)
+            result = self.__sandbox_data_api.delete_sandbox(
+                sandbox_id, config=config
+            )
 
             # 判断返回结果是否成功
             if result.get("code") != "SUCCESS":
@@ -794,7 +798,9 @@ async def get_sandbox_async(
             ServerError: 服务器错误
         """
         try:
-            result = await self.__sandbox_data_api.get_sandbox_async(sandbox_id)
+            result = await self.__sandbox_data_api.get_sandbox_async(
+                sandbox_id, config=config
+            )
 
             # 判断返回结果是否成功
             if result.get("code") != "SUCCESS":
@@ -834,7 +840,9 @@ def get_sandbox(
             ServerError: 服务器错误
         """
         try:
-            result = self.__sandbox_data_api.get_sandbox(sandbox_id)
+            result = self.__sandbox_data_api.get_sandbox(
+                sandbox_id, config=config
+            )
 
             # 判断返回结果是否成功
             if result.get("code") != "SUCCESS":