PyPI - veadk-python - Versions diffs - 0.2.5__py3-none-any.whl → 0.2.7__py3-none-any.whl - Mend

veadk-python 0.2.5py3-none-any.whl → 0.2.7py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of veadk-python might be problematic. Click here for more details.

Files changed (94) hide show

veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/src/templates/admin/posts.html ADDED Viewed

@@ -0,0 +1,53 @@
+{% extends "base.html" %}
+{% block title %}文章管理{% endblock %}
+{% block content %}
+<div class="admin-posts">
+    <div class="admin-header">
+        <h2>文章管理</h2>
+        <a href="{{ url_for('admin_edit_post') }}" class="btn">新建文章</a>
+    </div>
+    <table class="posts-table">
+        <thead>
+            <tr>
+                <th>标题</th>
+                <th>发布时间</th>
+                <th>操作</th>
+            </tr>
+        </thead>
+        <tbody>
+            {% for post in posts.items %}
+            <tr>
+                <td><a href="{{ url_for('post_detail', post_id=post.id) }}">{{ post.title }}</a></td>
+                <td>{{ post.created_at.strftime('%Y-%m-%d %H:%M') }}</td>
+                <td>
+                    <a href="{{ url_for('admin_edit_post', post_id=post.id) }}">编辑</a>
+                    <form action="{{ url_for('admin_delete_post', post_id=post.id) }}" method="POST" style="display:inline;">
+                        <button type="submit" class="delete-btn" onclick="return confirm('确定要删除这篇文章吗？')">删除</button>
+                    </form>
+                </td>
+            </tr>
+            {% else %}
+            <tr>
+                <td colspan="3">暂无文章</td>
+            </tr>
+            {% endfor %}
+        </tbody>
+    </table>
+    <!-- 分页 -->
+    <div class="pagination">
+        {% if posts.has_prev %}
+            <a href="{{ url_for('admin_posts', page=posts.prev_num) }}">&laquo; 上一页</a>
+        {% endif %}
+        <span>第 {{ posts.page }} 页，共 {{ posts.pages }} 页</span>
+        {% if posts.has_next %}
+            <a href="{{ url_for('admin_posts', page=posts.next_num) }}">下一页 &raquo;</a>
+        {% endif %}
+    </div>
+</div>
+{% endblock %}

veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/src/templates/base.html ADDED Viewed

@@ -0,0 +1,45 @@
+<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>{% block title %}博客系统{% endblock %}</title>
+    <link rel="stylesheet" href="{{ url_for('static', filename='css/style.css') }}">
+</head>
+<body>
+    <header>
+        <div class="container">
+            <h1><a href="{{ url_for('index') }}">我的博客</a></h1>
+            <nav>
+                <a href="{{ url_for('index') }}">首页</a>
+                {% if session.admin_logged_in %}
+                    <a href="{{ url_for('admin_dashboard') }}">管理后台</a>
+                    <a href="{{ url_for('admin_logout') }}">退出</a>
+                {% else %}
+                    <a href="{{ url_for('admin_login') }}">登录</a>
+                {% endif %}
+            </nav>
+        </div>
+    </header>
+    <main class="container">
+        {% with messages = get_flashed_messages() %}
+            {% if messages %}
+                <div class="flash-messages">
+                    {% for message in messages %}
+                        <div class="flash-message">{{ message }}</div>
+                    {% endfor %}
+                </div>
+            {% endif %}
+        {% endwith %}
+        {% block content %}{% endblock %}
+    </main>
+    <footer>
+        <div class="container">
+            <p>&copy; 2023 我的博客. All rights reserved.</p>
+        </div>
+    </footer>
+</body>
+</html>

veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/src/templates/index.html ADDED Viewed

@@ -0,0 +1,29 @@
+{% extends "base.html" %}
+{% block content %}
+<div class="posts">
+    {% for post in posts.items %}
+        <article class="post-preview">
+            <h2><a href="{{ url_for('post_detail', post_id=post.id) }}">{{ post.title }}</a></h2>
+            <p class="post-meta">发布于 {{ post.created_at.strftime('%Y-%m-%d %H:%M') }}</p>
+            <p>{{ post.content[:200] }}{% if post.content|length > 200 %}...{% endif %}</p>
+            <a href="{{ url_for('post_detail', post_id=post.id) }}" class="read-more">阅读全文</a>
+        </article>
+    {% else %}
+        <p>暂无文章。</p>
+    {% endfor %}
+</div>
+<!-- 分页 -->
+<div class="pagination">
+    {% if posts.has_prev %}
+        <a href="{{ url_for('index', page=posts.prev_num) }}">&laquo; 上一页</a>
+    {% endif %}
+    <span>第 {{ posts.page }} 页，共 {{ posts.pages }} 页</span>
+    {% if posts.has_next %}
+        <a href="{{ url_for('index', page=posts.next_num) }}">下一页 &raquo;</a>
+    {% endif %}
+</div>
+{% endblock %}

veadk/integrations/ve_faas/web_template/{{cookiecutter.local_dir_name}}/src/templates/post.html ADDED Viewed

@@ -0,0 +1,14 @@
+{% extends "base.html" %}
+{% block title %}{{ post.title }} - 博客系统{% endblock %}
+{% block content %}
+<article class="post-detail">
+    <h1>{{ post.title }}</h1>
+    <p class="post-meta">发布于 {{ post.created_at.strftime('%Y-%m-%d %H:%M') }}</p>
+    <div class="post-content">
+        {{ post.content|replace('\n', '<br>')|safe }}
+    </div>
+    <a href="{{ url_for('index') }}" class="back-link">&laquo; 返回首页</a>
+</article>
+{% endblock %}

veadk/integrations/ve_prompt_pilot/ve_prompt_pilot.py CHANGED Viewed

@@ -65,12 +65,15 @@ class VePromptPilot:
                 api_key=self.api_key,
             ):  # stream chunks of optimized prompt
                 # Process each chunk as it arrives
-                optimized_prompt += chunk.data.content
+                optimized_prompt += chunk.data.content if chunk.data else ""
                 # print(chunk.data.content, end="", flush=True)
                 if chunk.event == "usage":
-                    usage = chunk.data.usage
+                    usage = chunk.data.usage if chunk.data else 0
             optimized_prompt = optimized_prompt.replace("\\n", "\n")
             print(f"Optimized prompt for agent {agent.name}:\n{optimized_prompt}")
-            logger.info(f"Token usage: {usage['total_tokens']}")
+            if usage:
+                logger.info(f"Token usage: {usage['total_tokens']}")
+            else:
+                logger.warning("No usage data.")
         return optimized_prompt

veadk/integrations/ve_tls/__init__.py ADDED Viewed

@@ -0,0 +1,13 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.

veadk/integrations/ve_tls/utils.py ADDED Viewed

@@ -0,0 +1,117 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+import hashlib
+import json
+from typing import Any, Literal
+import httpx
+from volcengine.ApiInfo import ApiInfo
+from volcengine.auth.SignerV4 import SignerV4
+from volcengine.tls.const import (
+    APPLICATION_JSON,
+    CONTENT_MD5,
+    CONTENT_TYPE,
+    DATA,
+    WEB_TRACKS,
+)
+from volcengine.tls.TLSService import TLSService
+from veadk.utils.logger import get_logger
+logger = get_logger(__name__)
+HEADER_API_VERSION = "x-tls-apiversion"
+API_VERSION_V_0_3_0 = "0.3.0"
+API_INFO = {
+    "CreateProject": ApiInfo("POST", "/CreateProject", {}, {}, {}),
+    "DescribeProjects": ApiInfo("GET", "/DescribeProjects", {}, {}, {}),
+    "CreateTraceInstance": ApiInfo("POST", "/CreateTraceInstance", {}, {}, {}),
+    "DescribeTraceInstances": ApiInfo("GET", "/DescribeTraceInstances", {}, {}, {}),
+    "DescribeTraceInstance": ApiInfo("GET", "/DescribeTraceInstance", {}, {}, {}),
+}
+def __prepare_request(
+    client: TLSService,
+    api: str,
+    params: dict | None = None,
+    body: Any | None = None,
+    request_headers: dict | None = None,
+):
+    if params is None:
+        params = {}
+    if body is None:
+        body = {}
+    request = client.prepare_request(API_INFO[api], params)
+    if request_headers is None:
+        request_headers = {CONTENT_TYPE: APPLICATION_JSON}
+    request.headers.update(request_headers)
+    if "json" in request.headers[CONTENT_TYPE] and api != WEB_TRACKS:
+        request.body = json.dumps(body)
+    else:
+        request.body = body[DATA]
+    if len(request.body) != 0:
+        if isinstance(request.body, str):
+            request.headers[CONTENT_MD5] = hashlib.md5(
+                request.body.encode("utf-8")
+            ).hexdigest()
+        else:
+            request.headers[CONTENT_MD5] = hashlib.md5(request.body).hexdigest()
+    SignerV4.sign(request, client.service_info.credentials)
+    return request
+def ve_tls_request(
+    client: TLSService,
+    api: str,
+    params: dict | None = None,
+    body: dict | None = None,
+    request_headers: dict | None = None,
+    method: Literal["POST", "GET", "PUT", "DELETE"] = "POST",
+):
+    """Customize a standard HTTP request to the Volcengine TLS API"""
+    if request_headers is None:
+        request_headers = {HEADER_API_VERSION: API_VERSION_V_0_3_0}
+    elif HEADER_API_VERSION not in request_headers:
+        request_headers[HEADER_API_VERSION] = API_VERSION_V_0_3_0
+    if CONTENT_TYPE not in request_headers:
+        request_headers[CONTENT_TYPE] = APPLICATION_JSON
+    request = __prepare_request(client, api, params, body, request_headers)
+    url = request.build()
+    with httpx.Client() as session:
+        while True:
+            try:
+                response = session.request(
+                    method=method,
+                    url=url,
+                    headers=request.headers,
+                    data=request.body,  # type: ignore
+                    timeout=60,
+                )
+                return response.json()
+            except Exception as e:
+                logger.error(
+                    f"Error occurred while making {method} request to {url}: {e}. Response: {response}"
+                )

veadk/integrations/ve_tls/ve_tls.py ADDED Viewed

@@ -0,0 +1,208 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+import os
+from veadk.consts import DEFAULT_TLS_LOG_PROJECT_NAME, DEFAULT_TLS_TRACING_INSTANCE_NAME
+from veadk.utils.logger import get_logger
+logger = get_logger(__name__)
+class VeTLS:
+    def __init__(
+        self,
+        access_key: str | None = None,
+        secret_key: str | None = None,
+        region: str = "cn-beijing",
+    ):
+        try:
+            from volcengine.tls.TLSService import TLSService
+            from veadk.integrations.ve_tls.utils import ve_tls_request
+        except ImportError:
+            raise ImportError(
+                "Please install volcengine SDK before init VeTLS: pip install volcengine"
+            )
+        self._ve_tls_request = ve_tls_request
+        self.access_key = (
+            access_key if access_key else os.getenv("VOLCENGINE_ACCESS_KEY", "")
+        )
+        self.secret_key = (
+            secret_key if secret_key else os.getenv("VOLCENGINE_SECRET_KEY", "")
+        )
+        self.region = region
+        self._client = TLSService(
+            endpoint=f"https://tls-{self.region}.volces.com",
+            access_key_id=self.access_key,
+            access_key_secret=self.secret_key,
+            region=self.region,
+        )
+    def get_project_id_by_name(self, project_name: str) -> str:
+        """Get the ID of a log project by its name.
+        Args:
+            project_name (str): The name of the log project.
+        Returns:
+            str: The ID of the log project, or None if not found.
+        """
+        logger.info(f"Getting ID for log project '{project_name}' in TLS...")
+        request_body = {
+            "ProjectName": project_name,
+            "IsFullName": True,
+        }
+        try:
+            res = None
+            res = self._ve_tls_request(
+                client=self._client,
+                api="DescribeProjects",
+                body=request_body,
+                method="GET",
+            )
+            projects = res["Projects"]
+            for project in projects:
+                if project["ProjectName"] == project_name:
+                    return project["ProjectId"]
+            return "<no_project_id_found>"
+        except KeyError:
+            raise ValueError(f"Failed to get log project ID: {res}")
+    def create_log_project(self, project_name: str) -> str:
+        """Create a log project in TLS.
+        Args:
+            project_name (str): The name of the log project to create.
+        Returns:
+            str: The ID of the created log project.
+        """
+        logger.info(f"Creating log project '{project_name}' in TLS...")
+        request_body = {
+            "ProjectName": project_name,
+            "Region": self.region,
+            "Description": "Created by Volcengine Agent Development Kit (VeADK)",
+            "Tags": [{"Key": "provider", "Value": "VeADK"}],
+        }
+        try:
+            res = self._ve_tls_request(
+                client=self._client, api="CreateProject", body=request_body
+            )
+            if res["ErrorCode"] == "ProjectAlreadyExist":
+                logger.debug(
+                    f"Log project '{project_name}' already exists. Check its ID."
+                )
+                return self.get_project_id_by_name(project_name)
+            return res["ProjectId"]
+        except KeyError:
+            raise ValueError(f"Failed to create log project: {res}")
+    def get_trace_instance_by_name(self, log_project_id: str, trace_instance_name: str):
+        logger.info(f"Getting trace instance '{trace_instance_name}' in TLS...")
+        request_body = {
+            "PageSize": 100,
+            "ProjectId": log_project_id,
+            "TraceInstanceName": trace_instance_name,
+        }
+        try:
+            res = self._ve_tls_request(
+                client=self._client,
+                api="DescribeTraceInstances",
+                body=request_body,
+                method="GET",
+            )
+            for instance in res["TraceInstances"]:
+                if instance["TraceInstanceName"] == trace_instance_name:
+                    return instance
+        except KeyError:
+            raise ValueError(f"Failed to create log project: {res}")
+    def create_tracing_instance(self, log_project_id: str, trace_instance_name: str):
+        """Create a tracing instance in TLS.
+        Args:
+            instance_name (str): The name of the tracing instance to create.
+        Returns:
+            dict: The tracing instance.
+        """
+        logger.info(f"Creating tracing instance '{trace_instance_name}' in TLS...")
+        request_body = {
+            "ProjectId": log_project_id,
+            "TraceInstanceName": trace_instance_name,
+            "Description": "Created by Volcengine Agent Development Kit (VeADK)",
+        }
+        try:
+            res = None
+            res = self._ve_tls_request(
+                client=self._client,
+                api="CreateTraceInstance",
+                body=request_body,
+            )
+            if res["ErrorCode"] == "TopicAlreadyExist":
+                logger.debug(
+                    f"Log project '{trace_instance_name}' already exists. Check its ID."
+                )
+                return self.get_trace_instance_by_name(
+                    log_project_id, trace_instance_name
+                )
+            # after creation, get the trace instance details
+            res = self._ve_tls_request(
+                client=self._client,
+                api="DescribeTraceInstance",
+                body={"TraceInstanceID": res["TraceInstanceID"]},
+                method="GET",
+            )
+            return res
+        except KeyError:
+            raise ValueError(f"Failed to create tracing instance: {res}")
+    def get_trace_topic_id(self):
+        """Get the trace topic ID under VeADK default names.
+        This method is a tool function just designed for `veadk/config.py`.
+        Returns:
+            str: The trace topic ID.
+        """
+        logger.info("Getting trace topic ID for tracing instance in TLS...")
+        log_project_id = self.create_log_project(DEFAULT_TLS_LOG_PROJECT_NAME)
+        instance = self.create_tracing_instance(
+            log_project_id, DEFAULT_TLS_TRACING_INSTANCE_NAME
+        )
+        if not instance:
+            raise ValueError("None instance")
+        logger.info(f"Fetched trace topic id: {instance['TraceTopicId']}")
+        return instance["TraceTopicId"]

veadk-python 0.2.5__py3-none-any.whl → 0.2.7__py3-none-any.whl

Potentially problematic release.

veadk-python 0.2.5py3-none-any.whl → 0.2.7py3-none-any.whl