veadk-python 0.5.28__tar.gz → 0.5.30__tar.gz

{veadk_python-0.5.28 → veadk_python-0.5.30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: veadk-python
-Version: 0.5.28
+Version: 0.5.30
 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
@@ -212,7 +212,7 @@ Requires-Dist: pydantic-settings==2.10.1
 Requires-Dist: a2a-sdk>=0.3.7
 Requires-Dist: deprecated==1.2.18
 Requires-Dist: google-adk>=1.19.0
-Requires-Dist: litellm>=1.74.3
+Requires-Dist: litellm<=1.82.6,>=1.74.3
 Requires-Dist: loguru==0.7.3
 Requires-Dist: opentelemetry-exporter-otlp==1.37.0
 Requires-Dist: opentelemetry-instrumentation-logging>=0.56b0

{veadk_python-0.5.28 → veadk_python-0.5.30}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "veadk-python"
-version = "0.5.28"
+version = "0.5.30"
 description = "Volcengine agent development kit, integrations with Volcengine cloud services."
 readme = "README.md"
 requires-python = ">=3.10"
@@ -16,7 +16,7 @@ dependencies = [
     "a2a-sdk>=0.3.7", # For Google Agent2Agent protocol
     "deprecated==1.2.18",
     "google-adk>=1.19.0", # For basic agent architecture
-    "litellm>=1.74.3", # For model inference
+    "litellm>=1.74.3,<=1.82.6", # For model inference
     "loguru==0.7.3", # For better logging
     "opentelemetry-exporter-otlp==1.37.0",
     "opentelemetry-instrumentation-logging>=0.56b0",

{veadk_python-0.5.28 → veadk_python-0.5.30}/veadk/agent.py

@@ -87,6 +87,8 @@ class Agent(LlmAgent):
         example_store (Optional[BaseExampleProvider]): Example store for providing example Q/A.
         enable_shadowchar (bool): Whether to enable shadow character for the agent.
         enable_dynamic_load_skills (bool): Whether to enable dynamic loading of skills.
+        enable_responses_cache (bool): Whether Ark Responses API should reuse
+            `previous_response_id` and caching for multi-turn continuation.
     """
 
     model_config = ConfigDict(arbitrary_types_allowed=True, extra="allow")
@@ -118,6 +120,7 @@ class Agent(LlmAgent):
     tracers: list[BaseTracer] = []
 
     enable_responses: bool = False
+    enable_responses_cache: bool = True
 
     context_cache_config: Optional[ContextCacheConfig] = None
 
@@ -194,6 +197,7 @@ class Agent(LlmAgent):
                     model=f"{self.model_provider}/{self.model_name}",
                     api_key=self.model_api_key,
                     api_base=self.model_api_base,
+                    enable_responses_cache=self.enable_responses_cache,
                     **self.model_extra_config,
                 )
             else:

{veadk_python-0.5.28 → veadk_python-0.5.30}/veadk/models/ark_llm.py

@@ -481,10 +481,17 @@ def _remove_caching(request_data: dict) -> None:
     request_data.pop("caching", None)
 
 
-def request_reorganization_by_ark(request_data: Dict) -> Dict:
+def request_reorganization_by_ark(
+    request_data: Dict, enable_responses_cache: bool = True
+) -> Dict:
     # 1. model provider
     request_data = get_model_without_provider(request_data)
 
+    if not enable_responses_cache:
+        request_data.pop("previous_response_id", None)
+        _remove_caching(request_data)
+        request_data.pop("store", None)
+
     # 2. filtered input
     request_data["input"] = filtered_inputs(
         request_data.get("input"),
@@ -672,7 +679,9 @@ class ArkLlmClient:
     ) -> Union[ArkTypeResponse, AsyncStream[ResponseStreamEvent]]:
         # 1. Get request params
         api_base = kwargs.pop("api_base", DEFAULT_VIDEO_MODEL_API_BASE)
-        api_key = kwargs.pop("api_key", settings.model.api_key)
+        api_key = kwargs.pop("api_key", None)
+        if api_key is None:
+            api_key = settings.model.api_key
 
         # 2. Call openai responses
         client = AsyncArk(
@@ -689,6 +698,7 @@ class ArkLlm(Gemini):
     llm_client: ArkLlmClient = Field(default_factory=ArkLlmClient)
     _additional_args: Dict[str, Any] = None
     use_interactions_api: bool = True
+    enable_responses_cache: bool = True
 
     def __init__(self, **kwargs):
         # adk version check
@@ -699,12 +709,14 @@ class ArkLlm(Gemini):
                 "`pip install -U 'google-adk>=1.21.0'`"
             )
         super().__init__(**kwargs)
+        self.enable_responses_cache = kwargs.get("enable_responses_cache", True)
         drop_params = kwargs.pop("drop_params", None)
         self._additional_args = dict(kwargs)
         self._additional_args.pop("llm_client", None)
         self._additional_args.pop("messages", None)
         self._additional_args.pop("tools", None)
         self._additional_args.pop("stream", None)
+        self._additional_args.pop("enable_responses_cache", None)
         if drop_params is not None:
             self._additional_args["drop_params"] = drop_params
 
@@ -733,7 +745,7 @@ class ArkLlm(Gemini):
         # ------------------------------------------------------ #
         # get previous_response_id
         previous_response_id = None
-        if llm_request.previous_interaction_id:
+        if self.enable_responses_cache and llm_request.previous_interaction_id:
             previous_response_id = llm_request.previous_interaction_id
         responses_args = {
             "model": self.model,
@@ -786,7 +798,9 @@ class ArkLlm(Gemini):
     async def generate_content_via_responses(
         self, responses_args: dict, stream: bool = False
     ):
-        responses_args = request_reorganization_by_ark(responses_args)
+        responses_args = request_reorganization_by_ark(
+            responses_args, enable_responses_cache=self.enable_responses_cache
+        )
         if stream:
             responses_args["stream"] = True
             async for part in await self.llm_client.aresponses(**responses_args):

{veadk_python-0.5.28 → veadk_python-0.5.30}/veadk/skills/utils.py

@@ -174,6 +174,7 @@ def load_skills_from_cloud(skill_space_ids: str) -> list[Skill]:
                 "InnerTags": {"source": "sandbox"},
             }
             logger.debug(f"ListSkillsBySpaceId request body: {request_body}")
+            scheme = os.getenv("AGENTKIT_TOP_SCHEME", "https").lower()
 
             response = ve_request(
                 request_body=request_body,
@@ -185,6 +186,7 @@ def load_skills_from_cloud(skill_space_ids: str) -> list[Skill]:
                 region=region,
                 host=host,
                 header={"X-Security-Token": session_token},
+                scheme=scheme,
             )
 
             if isinstance(response, str):

{veadk_python-0.5.28 → veadk_python-0.5.30}/veadk/tools/skills_tools/download_skills_tool.py

@@ -28,6 +28,86 @@ from veadk.utils.logger import get_logger
 logger = get_logger(__name__)
 
 
+def _download_skill_via_vestack(
+    tos_path: str,
+    skill_name: str,
+    access_key: str,
+    secret_key: str,
+    session_token: str,
+    service: str,
+    region: str,
+    host: str,
+    scheme: str,
+    zip_path: Path,
+) -> bool:
+    import json
+    import requests
+    from veadk.utils.volcengine_sign import ve_request
+
+    try:
+        path_parts = tos_path.split("/")
+        if len(path_parts) >= 3:
+            skill_id = path_parts[1]
+            skill_version = path_parts[2]
+        else:
+            logger.error(f"Invalid TosPath format for skill '{skill_name}': {tos_path}")
+            return False
+    except Exception as e:
+        logger.error(f"Failed to parse TosPath for skill '{skill_name}': {e}")
+        return False
+
+    # Call GenTempTosObjectDownloadUrl API
+    temp_url_request_body = {
+        "SkillId": skill_id,
+        "SkillVersion": skill_version,
+    }
+
+    temp_url_res = ve_request(
+        request_body=temp_url_request_body,
+        action="GenTempTosObjectDownloadUrl",
+        ak=access_key,
+        sk=secret_key,
+        service=service,
+        version="2025-10-30",
+        region=region,
+        host=host,
+        header={"X-Security-Token": session_token},
+        scheme=scheme,
+    )
+
+    if isinstance(temp_url_res, str):
+        temp_url_res = json.loads(temp_url_res)
+
+    if (
+        "ResponseMetadata" in temp_url_res
+        and "Error" in temp_url_res["ResponseMetadata"]
+    ):
+        error_details = temp_url_res["ResponseMetadata"]["Error"]
+        logger.error(
+            f"Failed to get temporary download URL for '{skill_name}': {error_details}"
+        )
+        return False
+    else:
+        signed_url = temp_url_res.get("Result", {}).get("SignedUrl")
+        if not signed_url:
+            logger.error(
+                f"Failed to get SignedUrl from GenTempTosObjectDownloadUrl response: {temp_url_res}"
+            )
+            return False
+        else:
+            try:
+                response = requests.get(signed_url)
+                response.raise_for_status()
+                with open(zip_path, "wb") as f:
+                    f.write(response.content)
+                return True
+            except Exception as e:
+                logger.warning(
+                    f"Failed to download skill '{skill_name}' from minio: {e}"
+                )
+                return False
+
+
 def download_skills_tool(
     download_path: str, skill_names: Optional[list[str]] = None
 ) -> str:
@@ -92,6 +172,7 @@ def download_skills_tool(
 
         all_downloaded_skills = []
 
+        scheme = os.getenv("AGENTKIT_TOP_SCHEME", "https").lower()
         # Iterate through each skill space
         for skill_space_id in skill_space_ids_list:
             try:
@@ -112,6 +193,7 @@ def download_skills_tool(
                     region=region,
                     host=host,
                     header={"X-Security-Token": session_token},
+                    scheme=scheme,
                 )
 
                 if isinstance(response, str):
@@ -161,11 +243,26 @@ def download_skills_tool(
 
                     # Download zip file
                     zip_path = download_dir / f"{skill_name}.zip"
-                    success = tos_client.download(
-                        bucket_name=tos_bucket,
-                        object_key=tos_path,
-                        save_path=str(zip_path),
-                    )
+
+                    if cloud_provider == "vestack":
+                        success = _download_skill_via_vestack(
+                            tos_path=tos_path,
+                            skill_name=skill_name,
+                            access_key=access_key,
+                            secret_key=secret_key,
+                            session_token=session_token,
+                            service=service,
+                            region=region,
+                            host=host,
+                            scheme=scheme,
+                            zip_path=zip_path,
+                        )
+                    else:
+                        success = tos_client.download(
+                            bucket_name=tos_bucket,
+                            object_key=tos_path,
+                            save_path=str(zip_path),
+                        )
 
                     if not success:
                         logger.warning(f"Failed to download skill '{skill_name}'")

{veadk_python-0.5.28 → veadk_python-0.5.30}/veadk/tools/skills_tools/register_skills_tool.py

@@ -111,46 +111,99 @@ def register_skills_tool(
             secret_key = cred.secret_access_key
             session_token = cred.session_token
 
-        res = ve_request(
-            request_body={},
-            action="GetCallerIdentity",
-            ak=access_key,
-            sk=secret_key,
-            service="sts",
-            version="2018-01-01",
-            region=region,
-            host="sts.volcengineapi.com"
-            if cloud_provider != "byteplus"
-            else "open.byteplusapi.com",
-            header={"X-Security-Token": session_token},
-        )
-        try:
-            account_id = res["Result"]["AccountId"]
-        except KeyError as e:
-            logger.error(
-                f"Error occurred while getting account id: {e}, response is {res}"
+        account_id = ""
+        if cloud_provider != "vestack":
+            res = ve_request(
+                request_body={},
+                action="GetCallerIdentity",
+                ak=access_key,
+                sk=secret_key,
+                service="sts",
+                version="2018-01-01",
+                region=region,
+                host="sts.volcengineapi.com"
+                if cloud_provider != "byteplus"
+                else "open.byteplusapi.com",
+                header={"X-Security-Token": session_token},
             )
-            return f"Error: Failed to get account id when registering skill '{skill_name}'."
+            try:
+                account_id = res["Result"]["AccountId"]
+            except KeyError as e:
+                logger.error(
+                    f"Error occurred while getting account id: {e}, response is {res}"
+                )
+                return f"Error: Failed to get account id when registering skill '{skill_name}'."
 
         tos_bucket = f"agentkit-platform-{region}-{account_id}-skill"
+        scheme = os.getenv("AGENTKIT_TOP_SCHEME", "https").lower()
+        if cloud_provider == "vestack":
+            import requests
+
+            # Call GenTempTosObjectUrl API
+            temp_url_request_body = {
+                "SkillName": skill_name,
+            }
+
+            temp_url_res = ve_request(
+                request_body=temp_url_request_body,
+                action="GenTempTosObjectUrl",
+                ak=access_key,
+                sk=secret_key,
+                service=agentkit_tool_service,
+                version="2025-10-30",
+                region=region,
+                host=agentkit_skill_host,
+                header={"X-Security-Token": session_token},
+                scheme=scheme,
+            )
 
-        tos_client = VeTOS(
-            ak=access_key,
-            sk=secret_key,
-            session_token=session_token,
-            bucket_name=tos_bucket,
-            region=region,
-        )
+            if isinstance(temp_url_res, str):
+                temp_url_res = json.loads(temp_url_res)
+
+            if (
+                "ResponseMetadata" in temp_url_res
+                and "Error" in temp_url_res["ResponseMetadata"]
+            ):
+                error_details = temp_url_res["ResponseMetadata"]["Error"]
+                logger.error(
+                    f"Failed to get temporary upload URL for '{skill_name}': {error_details}"
+                )
+                return f"Failed to get temporary upload URL for '{skill_name}': {error_details}"
+
+            signed_url = temp_url_res.get("Result", {}).get("SignedUrl")
+            tos_url = temp_url_res.get("Result", {}).get("TosUrl")
+
+            if not signed_url or not tos_url:
+                logger.error(
+                    f"Failed to get SignedUrl or TosUrl from GenTempTosObjectUrl response: {temp_url_res}"
+                )
+                return f"Failed to get temporary upload URL for '{skill_name}'."
+
+            try:
+                with open(zip_file_path, "rb") as f:
+                    response = requests.put(signed_url, data=f)
+                    response.raise_for_status()
+            except Exception as e:
+                logger.error(f"Failed to upload skill '{skill_name}' to minio: {e}")
+                return f"Failed to upload skill '{skill_name}' to minio: {e}"
+        else:
+            tos_client = VeTOS(
+                ak=access_key,
+                sk=secret_key,
+                session_token=session_token,
+                bucket_name=tos_bucket,
+                region=region,
+            )
 
-        object_key = (
-            f"uploads/{datetime.now().strftime('%Y%m%d_%H%M%S')}/{skill_name}.zip"
-        )
-        tos_client.upload_file(
-            file_path=zip_file_path, bucket_name=tos_bucket, object_key=object_key
-        )
-        tos_url = tos_client.build_tos_url(
-            bucket_name=tos_bucket, object_key=object_key
-        )
+            object_key = (
+                f"uploads/{datetime.now().strftime('%Y%m%d_%H%M%S')}/{skill_name}.zip"
+            )
+            tos_client.upload_file(
+                file_path=zip_file_path, bucket_name=tos_bucket, object_key=object_key
+            )
+            tos_url = tos_client.build_tos_url(
+                bucket_name=tos_bucket, object_key=object_key
+            )
 
         skill_space_ids = os.getenv("SKILL_SPACE_ID", "")
         skill_space_ids_list = [
@@ -173,6 +226,7 @@ def register_skills_tool(
             region=region,
             host=agentkit_skill_host,
             header={"X-Security-Token": session_token},
+            scheme=scheme,
         )
 
         if isinstance(response, str):

{veadk_python-0.5.28 → veadk_python-0.5.30}/veadk/tools/skills_tools/skills_tool.py

@@ -117,6 +117,12 @@ class SkillsTool(BaseTool):
             # 1. Download skill from TOS if not found locally
             user_skill_dir = skill_dir / skill_name
             if not user_skill_dir.exists() or not user_skill_dir.is_dir():
+                cloud_provider = (os.getenv("CLOUD_PROVIDER") or "").lower()
+                if cloud_provider == "vestack":
+                    error_msg = f"Error: Skill '{skill_name}' not found locally or in skill space. Downloading from TOS_SKILLS_DIR is not supported in vestack environment."
+                    logger.error(error_msg)
+                    return error_msg
+
                 # Try to download from TOS
                 logger.info(
                     f"Skill '{skill_name}' not found locally or in skill space, attempting to download from TOS..."
@@ -240,22 +246,35 @@ class SkillsTool(BaseTool):
 
                     tos_bucket, tos_path = skill.bucket_name, skill.path
 
-                    # Initialize VeTOS client
-                    tos_client = VeTOS(
-                        ak=access_key,
-                        sk=secret_key,
-                        session_token=session_token,
-                        bucket_name=tos_bucket,
-                        region=region,
-                    )
-
                     save_path = skill_dir / f"{skill_name}.zip"
 
-                    success = tos_client.download(
-                        bucket_name=tos_bucket,
-                        object_key=tos_path,
-                        save_path=save_path,
-                    )
+                    cloud_provider = (os.getenv("CLOUD_PROVIDER") or "").lower()
+                    if cloud_provider == "vestack":
+                        success = self._download_skill_via_vestack(
+                            skill=skill,
+                            tos_path=tos_path,
+                            cloud_provider=cloud_provider,
+                            access_key=access_key,
+                            secret_key=secret_key,
+                            session_token=session_token,
+                            skill_name=skill_name,
+                            save_path=save_path,
+                        )
+                    else:
+                        # Initialize VeTOS client
+                        tos_client = VeTOS(
+                            ak=access_key,
+                            sk=secret_key,
+                            session_token=session_token,
+                            bucket_name=tos_bucket,
+                            region=region,
+                        )
+
+                        success = tos_client.download(
+                            bucket_name=tos_bucket,
+                            object_key=tos_path,
+                            save_path=save_path,
+                        )
 
                     if not success:
                         return (
@@ -344,6 +363,94 @@ class SkillsTool(BaseTool):
             logger.error(f"Failed to invoke skill {skill_name}: {e}")
             return f"Error invoking skill '{skill_name}': {e}"
 
+    def _download_skill_via_vestack(
+        self,
+        skill: Any,
+        tos_path: str,
+        cloud_provider: str,
+        access_key: str,
+        secret_key: str,
+        session_token: str,
+        skill_name: str,
+        save_path: Any,
+    ) -> bool:
+        """Download a skill using the vestack environment GenTempTosObjectDownloadUrl API."""
+        import json
+        import requests
+        from veadk.utils.volcengine_sign import ve_request
+
+        # Extract skill_id and skill_version from TosPath
+        # skills/s-yeh6iwdnggwobasystug/v1/web-search.zip
+        skill_id = skill.id
+        skill_version = ""
+        try:
+            path_parts = tos_path.split("/")
+            if len(path_parts) >= 3:
+                skill_id = path_parts[1]
+                skill_version = path_parts[2]
+        except Exception:
+            pass
+
+        # Call GenTempTosObjectDownloadUrl API
+        temp_url_request_body = {
+            "SkillId": skill_id,
+            "SkillVersion": skill_version,
+        }
+
+        agentkit_tool_service = os.getenv("AGENTKIT_TOOL_SERVICE_CODE", "agentkit")
+        region = os.getenv("AGENTKIT_TOOL_REGION", "cn-beijing")
+        default_sld = "byteplusapi" if cloud_provider == "byteplus" else "volcengineapi"
+        agentkit_skill_host = os.getenv(
+            "AGENTKIT_SKILL_HOST",
+            agentkit_tool_service + "." + region + f".{default_sld}.com",
+        )
+        scheme = os.getenv("AGENTKIT_TOP_SCHEME", "https").lower()
+
+        temp_url_res = ve_request(
+            request_body=temp_url_request_body,
+            action="GenTempTosObjectDownloadUrl",
+            ak=access_key,
+            sk=secret_key,
+            service=agentkit_tool_service,
+            version="2025-10-30",
+            region=region,
+            host=agentkit_skill_host,
+            header={"X-Security-Token": session_token},
+            scheme=scheme,
+        )
+
+        if isinstance(temp_url_res, str):
+            temp_url_res = json.loads(temp_url_res)
+
+        if (
+            "ResponseMetadata" in temp_url_res
+            and "Error" in temp_url_res["ResponseMetadata"]
+        ):
+            error_details = temp_url_res["ResponseMetadata"]["Error"]
+            logger.error(
+                f"Failed to get temporary download URL for '{skill_name}': {error_details}"
+            )
+            return False
+        else:
+            signed_url = temp_url_res.get("Result", {}).get("SignedUrl")
+            if not signed_url:
+                logger.error(
+                    f"Failed to get SignedUrl from GenTempTosObjectDownloadUrl response: {temp_url_res}"
+                )
+                return False
+            else:
+                try:
+                    response = requests.get(signed_url)
+                    response.raise_for_status()
+                    with open(save_path, "wb") as f:
+                        f.write(response.content)
+                    return True
+                except Exception as e:
+                    logger.error(
+                        f"Failed to download skill '{skill_name}' from minio: {e}"
+                    )
+                    return False
+
     def _format_skill_content(self, skill_name: str, content: str, skill_dir) -> str:
         """Format skill content for display to the agent."""
         header = (

{veadk_python-0.5.28 → veadk_python-0.5.30}/veadk/version.py

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

{veadk_python-0.5.28 → veadk_python-0.5.30}/veadk_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: veadk-python
-Version: 0.5.28
+Version: 0.5.30
 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
@@ -212,7 +212,7 @@ Requires-Dist: pydantic-settings==2.10.1
 Requires-Dist: a2a-sdk>=0.3.7
 Requires-Dist: deprecated==1.2.18
 Requires-Dist: google-adk>=1.19.0
-Requires-Dist: litellm>=1.74.3
+Requires-Dist: litellm<=1.82.6,>=1.74.3
 Requires-Dist: loguru==0.7.3
 Requires-Dist: opentelemetry-exporter-otlp==1.37.0
 Requires-Dist: opentelemetry-instrumentation-logging>=0.56b0

{veadk_python-0.5.28 → veadk_python-0.5.30}/veadk_python.egg-info/requires.txt

@@ -2,7 +2,7 @@ pydantic-settings==2.10.1
 a2a-sdk>=0.3.7
 deprecated==1.2.18
 google-adk>=1.19.0
-litellm>=1.74.3
+litellm<=1.82.6,>=1.74.3
 loguru==0.7.3
 opentelemetry-exporter-otlp==1.37.0
 opentelemetry-instrumentation-logging>=0.56b0