veadk-python 0.2.26__tar.gz → 0.2.27__tar.gz

{veadk_python-0.2.26/veadk_python.egg-info → veadk_python-0.2.27}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: veadk-python
-Version: 0.2.26
+Version: 0.2.27
 Summary: Volcengine agent development kit, integrations with Volcengine cloud services.
 Author-email: Yaozheng Fang <fangyozheng@gmail.com>, Guodong Li <cu.eric.lee@gmail.com>, Zhi Han <sliverydayday@gmail.com>, Meng Wang <mengwangwm@gmail.com>
 License:                                  Apache License

{veadk_python-0.2.26 → veadk_python-0.2.27}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "veadk-python"
-version = "0.2.26"
+version = "0.2.27"
 description = "Volcengine agent development kit, integrations with Volcengine cloud services."
 readme = "README.md"
 requires-python = ">=3.10"

{veadk_python-0.2.26 → veadk_python-0.2.27}/veadk/auth/veauth/ark_veauth.py

@@ -36,7 +36,7 @@ def get_ark_token(region: str = "cn-beijing") -> str:
         session_token = cred.session_token
 
     res = ve_request(
-        request_body={"ProjectName": "default", "Filter": {}},
+        request_body={"ProjectName": "default", "Filter": {"AllowAll": True}},
         header={"X-Security-Token": session_token},
         action="ListApiKeys",
         ak=access_key,
@@ -48,6 +48,10 @@ def get_ark_token(region: str = "cn-beijing") -> str:
     )
     try:
         first_api_key_id = res["Result"]["Items"][0]["Id"]
+        logger.warning("By default, VeADK fetches the first API Key in the list.")
+        logger.info(
+            f"Try to fetch ARK API Key with id={first_api_key_id}, name={res['Result']['Items'][0]['Name']}"
+        )
     except KeyError:
         raise ValueError(f"Failed to get ARK api key list: {res}")
 
@@ -65,7 +69,7 @@ def get_ark_token(region: str = "cn-beijing") -> str:
     )
     try:
         api_key = res["Result"]["ApiKey"]
-        logger.info("Successfully fetching ARK API Key.")
+        logger.info(f"Successfully fetched ARK API Key (starts with {api_key[:8]}).")
         return api_key
     except KeyError:
         raise ValueError(f"Failed to get ARK api key: {res}")

{veadk_python-0.2.26 → veadk_python-0.2.27}/veadk/cli/cli_deploy.py

@@ -67,6 +67,7 @@ TEMP_PATH = "/tmp"
     help="Expected Volcengine Identity client name",
 )
 @click.option("--path", default=".", help="Local project path")
+@click.option("--iam-role", default=None, help="iam role for the vefaas function")
 def deploy(
     volcengine_access_key: str,
     volcengine_secret_key: str,
@@ -80,6 +81,7 @@ def deploy(
     user_pool_name: str,
     client_name: str,
     path: str,
+    iam_role: str,
 ) -> None:
     """Deploy a user project to Volcengine FaaS application.
 
@@ -131,6 +133,8 @@ def deploy(
     from veadk.config import getenv
     from veadk.utils.logger import get_logger
     from veadk.utils.misc import formatted_timestamp, load_module_from_file
+    import os
+    from veadk.config import veadk_environments
 
     logger = get_logger(__name__)
 
@@ -138,6 +142,11 @@ def deploy(
         volcengine_access_key = getenv("VOLCENGINE_ACCESS_KEY")
     if not volcengine_secret_key:
         volcengine_secret_key = getenv("VOLCENGINE_SECRET_KEY")
+    if not iam_role:
+        iam_role = getenv("IAM_ROLE", None, allow_false_values=True)
+    else:
+        os.environ["IAM_ROLE"] = iam_role
+        veadk_environments["IAM_ROLE"] = iam_role
 
     user_proj_abs_path = Path(path).resolve()
     template_dir_path = Path(vefaas.__file__).parent / "template"

{veadk_python-0.2.26 → veadk_python-0.2.27}/veadk/integrations/ve_faas/ve_faas.py

@@ -32,7 +32,7 @@ from veadk.integrations.ve_faas.ve_faas_utils import (
     zip_and_encode_folder,
 )
 from veadk.utils.logger import get_logger
-from veadk.utils.misc import formatted_timestamp
+from veadk.utils.misc import formatted_timestamp, getenv
 from veadk.utils.volcengine_sign import ve_request
 
 logger = get_logger(__name__)
@@ -114,10 +114,11 @@ class VeFaaS:
                 name=function_name,
                 description="Created by VeADK (Volcengine Agent Development Kit)",
                 tags=[TagForCreateFunctionInput(key="provider", value="veadk")],
-                runtime="native-python3.10/v1",
+                runtime="native-python3.12/v1",
                 request_timeout=1800,
                 envs=envs,
                 memory_mb=2048,
+                role=getenv("IAM_ROLE", None, allow_false_values=True),
             )
         )
 

{veadk_python-0.2.26 → veadk_python-0.2.27}/veadk/knowledgebase/backends/vikingdb_knowledge_backend.py

@@ -137,7 +137,7 @@ class VikingDBKnowledgeBackend(BaseKnowledgebaseBackend):
                 "it must start with an English letter, contain only letters, numbers, and underscores, and have a length of 1-128."
             )
 
-    def _get_tos_client(self) -> VeTOS:
+    def _get_tos_client(self, tos_bucket_name: str) -> VeTOS:
         volcengine_access_key = self.volcengine_access_key
         volcengine_secret_key = self.volcengine_secret_key
         session_token = self.session_token
@@ -153,7 +153,7 @@ class VikingDBKnowledgeBackend(BaseKnowledgebaseBackend):
             sk=volcengine_secret_key,
             session_token=session_token,
             region=self.tos_config.region,
-            bucket_name=self.tos_config.bucket,
+            bucket_name=tos_bucket_name or self.tos_config.bucket,
         )
 
     @override
@@ -461,7 +461,7 @@ class VikingDBKnowledgeBackend(BaseKnowledgebaseBackend):
         metadata: dict | None = None,
     ) -> str:
         # Here, we set the metadata via the TOS object, ref: https://www.volcengine.com/docs/84313/1254624
-        self._tos_client = self._get_tos_client()
+        self._tos_client = self._get_tos_client(tos_bucket_name)
 
         self._tos_client.bucket_name = tos_bucket_name
         coro = self._tos_client.upload(

{veadk_python-0.2.26 → veadk_python-0.2.27}/veadk/version.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-VERSION = "0.2.26"
+VERSION = "0.2.27"

{veadk_python-0.2.26 → veadk_python-0.2.27/veadk_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: veadk-python
-Version: 0.2.26
+Version: 0.2.27
 Summary: Volcengine agent development kit, integrations with Volcengine cloud services.
 Author-email: Yaozheng Fang <fangyozheng@gmail.com>, Guodong Li <cu.eric.lee@gmail.com>, Zhi Han <sliverydayday@gmail.com>, Meng Wang <mengwangwm@gmail.com>
 License:                                  Apache License