coze-coding-utils 0.1.2__py3-none-any.whl → 0.1.5__py3-none-any.whl

coze_coding_utils/runtime_ctx/context.py

@@ -1,11 +1,23 @@
 import os
 import uuid
 from dataclasses import dataclass
-from typing import Optional, Dict
+from typing import Optional, Dict, Mapping
 
+# Header keys
+HEADER_X_TT_LOGID = "x-tt-logid"
+HEADER_X_TT_ENV = "x-tt-env"
+HEADER_X_USE_PPE = "x-use-ppe"
+HEADER_X_TT_ENV_FE = "x-tt-env-fe"
+HEADER_BENEFIT_BIZ_SCENE = "benefit-biz-scene"
 
-@dataclass
+# Environment variable keys
+ENV_SPACE_ID = "COZE_PROJECT_SPACE_ID"
+ENV_PROJECT_ID = "COZE_PROJECT_ID"
+
+
+@dataclass(slots=True)
 class Context:
+    """运行时上下文，封装请求关联的标识与环境信息。"""
     run_id: str
     space_id: str
     project_id: str
@@ -17,18 +29,47 @@ class Context:
     benefit_biz_scene: Optional[str] = None
 
 
-def new_context(method: str, headers: Optional[Dict[str, str]] = None) -> Context:
+def new_context(method: str, headers: Optional[Mapping[str, str]] = None) -> Context:
+    """创建上下文对象，读取必要环境变量并可从请求头补充可选字段。"""
     ctx = Context(
         run_id=str(uuid.uuid4()),
-        space_id=os.getenv("COZE_PROJECT_SPACE_ID", ""),
-        project_id=os.getenv("COZE_PROJECT_ID", ""),
+        space_id=os.getenv(ENV_SPACE_ID, ""),
+        project_id=os.getenv(ENV_PROJECT_ID, ""),
         method=method,
     )
     if headers:
-        ctx.logid = headers.get("x-tt-logid", "")
-        ctx.x_tt_env = headers.get("x-tt-env")
-        ctx.x_use_ppe = headers.get("x-use-ppe")
-        ctx.x_tt_env_fe = headers.get("x-tt-env-fe")
-        ctx.benefit_biz_scene = headers.get("benefit-biz-scene")
+        if HEADER_X_TT_LOGID in headers:
+            ctx.logid = headers[HEADER_X_TT_LOGID]
+        if HEADER_X_TT_ENV in headers:
+            ctx.x_tt_env = headers[HEADER_X_TT_ENV]
+        if HEADER_X_USE_PPE in headers:
+            ctx.x_use_ppe = headers[HEADER_X_USE_PPE]
+        if HEADER_X_TT_ENV_FE in headers:
+            ctx.x_tt_env_fe = headers[HEADER_X_TT_ENV_FE]
+        if HEADER_BENEFIT_BIZ_SCENE in headers:
+            ctx.benefit_biz_scene = headers[HEADER_BENEFIT_BIZ_SCENE]
     return ctx
 
+
+def default_headers(ctx: Context | None) -> Dict[str, str]:
+    """从上下文生成请求头字典，仅包含已设置的字段。"""
+    if not ctx:
+        return {}
+    headers: Dict[str, str] = {}
+    if ctx.logid:
+        headers[HEADER_X_TT_LOGID] = ctx.logid
+    if ctx.x_tt_env:
+        headers[HEADER_X_TT_ENV] = ctx.x_tt_env
+    if ctx.x_use_ppe:
+        headers[HEADER_X_USE_PPE] = ctx.x_use_ppe
+    if ctx.x_tt_env_fe:
+        headers[HEADER_X_TT_ENV_FE] = ctx.x_tt_env_fe
+    if ctx.benefit_biz_scene:
+        headers[HEADER_BENEFIT_BIZ_SCENE] = ctx.benefit_biz_scene
+    return headers
+
+__all__ = [
+    "Context",
+    "new_context",
+    "default_headers",
+]

{coze_coding_utils-0.1.2.dist-info → coze_coding_utils-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: coze-coding-utils
-Version: 0.1.2
+Version: 0.1.5
 Summary: Utilities for Coze coding client runtime context and helpers.
 Project-URL: Homepage, https://code.byted.org/stone/coze-coding-client
 Author: Bytedance Stone Team

coze_coding_utils-0.1.5.dist-info/RECORD

@@ -0,0 +1,7 @@
+coze_coding_utils/__init__.py,sha256=vJ5SRiIxcBwcGt0tBF9QingGIl8ICxQuTnAMisQs34A,36
+coze_coding_utils/runtime_ctx/__init__.py,sha256=4W8VliAYUP1KY2gLJ_YDy2TmcXYVm-PY7XikQD_bFwA,2
+coze_coding_utils/runtime_ctx/context.py,sha256=oU9LlYIhQsOl07vSxOWhOSJvh5_jQwIllP_oDERLEu8,2324
+coze_coding_utils-0.1.5.dist-info/METADATA,sha256=l7jTX5A3KdMqDWH6c9SBRiehhvUrDPXLeyUtTKi4hBc,978
+coze_coding_utils-0.1.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+coze_coding_utils-0.1.5.dist-info/licenses/LICENSE,sha256=lzckZhAjHlpSJcWvppoST095IHFpBwKiB2pKcBv7vP4,1078
+coze_coding_utils-0.1.5.dist-info/RECORD,,

coze_coding_utils-0.1.2.dist-info/RECORD

@@ -1,7 +0,0 @@
-coze_coding_utils/__init__.py,sha256=vJ5SRiIxcBwcGt0tBF9QingGIl8ICxQuTnAMisQs34A,36
-coze_coding_utils/runtime_ctx/__init__.py,sha256=4W8VliAYUP1KY2gLJ_YDy2TmcXYVm-PY7XikQD_bFwA,2
-coze_coding_utils/runtime_ctx/context.py,sha256=gvQnDJ9YSuOLrSpw_a36IKbMeMC4yR3PAfkLSpr45fc,939
-coze_coding_utils-0.1.2.dist-info/METADATA,sha256=GvIBIaxO94Sf2FVuMqI6Z0y_WIUhK2bzxL254t6PqOI,978
-coze_coding_utils-0.1.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-coze_coding_utils-0.1.2.dist-info/licenses/LICENSE,sha256=lzckZhAjHlpSJcWvppoST095IHFpBwKiB2pKcBv7vP4,1078
-coze_coding_utils-0.1.2.dist-info/RECORD,,