coze-coding-utils 0.1.7__tar.gz → 0.1.8__tar.gz

{coze_coding_utils-0.1.7 → coze_coding_utils-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: coze-coding-utils
-Version: 0.1.7
+Version: 0.1.8
 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.7 → coze_coding_utils-0.1.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "coze-coding-utils"
-version = "0.1.7"
+version = "0.1.8"
 description = "Utilities for Coze coding client runtime context and helpers."
 readme = "README.md"
 requires-python = ">=3.10"

coze_coding_utils-0.1.8/src/coze_coding_utils/__init__.py

@@ -0,0 +1,3 @@
+__all__ = []
+
+__version__ = "0.1.8"

{coze_coding_utils-0.1.7 → coze_coding_utils-0.1.8}/src/coze_coding_utils/runtime_ctx/context.py

@@ -12,6 +12,7 @@ HEADER_RPC_PERSIST_REC_REC_BIZ_SCENE = "rpc-persist-rec-rec-biz-scene"
 HEADER_RPC_PERSIST_COZE_RECORD_ROOT_ID = "rpc-persist-coze-record-root-id" # root_id，串联一次完整请求，通常标识一次完整对话
 HEADER_RPC_PERSIST_REC_ROOT_ENTITY_TYPE = "rpc-persist-rec-rec-root-entity-type" # 对应最顶层实体类型，用于标识资源消耗的来源归属实体
 HEADER_RPC_PERSIST_REC_ROOT_ENTITY_ID = "rpc-persist-rec-rec-root-entity-id" #对应最顶层实体ID
+HEADER_RPC_PERSIST_REC_REC_EXT_INFO = "rpc-persist-rec-rec-ext-info" # 扩展信息，json字符串格式
 
 # Environment variable keys
 ENV_SPACE_ID = "COZE_PROJECT_SPACE_ID"
@@ -26,14 +27,16 @@ class Context:
     project_id: str
     logid: str = ""
     method: str = ""
+
     x_tt_env: Optional[str] = None
     x_use_ppe: Optional[str] = None
     x_tt_env_fe: Optional[str] = None
-
+    
     rpc_persist_rec_rec_biz_scene: Optional[str] = None
     rpc_persist_coze_record_root_id: Optional[str] = None
     rpc_persist_rec_root_entity_type: Optional[str] = None
     rpc_persist_rec_root_entity_id: Optional[str] = None
+    rpc_persist_rec_rec_ext_info: Optional[str] = None
 
 
 def new_context(method: str, headers: Optional[Mapping[str, str]] = None) -> Context:
@@ -44,23 +47,24 @@ def new_context(method: str, headers: Optional[Mapping[str, str]] = None) -> Con
         project_id=os.getenv(ENV_PROJECT_ID, ""),
         method=method,
     )
+    # logid 与 method 保持一致，不读取头部覆盖
+    ctx.logid = method
     if headers:
-        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_RPC_PERSIST_REC_REC_BIZ_SCENE in headers:
-            ctx.rpc_persist_rec_rec_biz_scene = headers[HEADER_RPC_PERSIST_REC_REC_BIZ_SCENE]
-        if HEADER_RPC_PERSIST_COZE_RECORD_ROOT_ID in headers:
-            ctx.rpc_persist_coze_record_root_id = headers[HEADER_RPC_PERSIST_COZE_RECORD_ROOT_ID]
-        if HEADER_RPC_PERSIST_REC_ROOT_ENTITY_TYPE in headers:
-            ctx.rpc_persist_rec_root_entity_type = headers[HEADER_RPC_PERSIST_REC_ROOT_ENTITY_TYPE]
-        if HEADER_RPC_PERSIST_REC_ROOT_ENTITY_ID in headers:
-            ctx.rpc_persist_rec_root_entity_id = headers[HEADER_RPC_PERSIST_REC_ROOT_ENTITY_ID]
+        norm = {k.casefold(): v for k, v in headers.items()}
+        HEADERS_TO_ATTR = {
+            HEADER_X_TT_ENV: "x_tt_env",
+            HEADER_X_USE_PPE: "x_use_ppe",
+            HEADER_X_TT_ENV_FE: "x_tt_env_fe",
+            HEADER_RPC_PERSIST_REC_REC_BIZ_SCENE: "rpc_persist_rec_rec_biz_scene",
+            HEADER_RPC_PERSIST_COZE_RECORD_ROOT_ID: "rpc_persist_coze_record_root_id",
+            HEADER_RPC_PERSIST_REC_ROOT_ENTITY_TYPE: "rpc_persist_rec_root_entity_type",
+            HEADER_RPC_PERSIST_REC_ROOT_ENTITY_ID: "rpc_persist_rec_root_entity_id",
+            HEADER_RPC_PERSIST_REC_REC_EXT_INFO: "rpc_persist_rec_rec_ext_info",
+        }
+        for hk, attr in HEADERS_TO_ATTR.items():
+            val = norm.get(hk)
+            if val:
+                setattr(ctx, attr, val)
     return ctx
 
 
@@ -69,22 +73,21 @@ 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.rpc_persist_rec_rec_biz_scene:
-        headers[HEADER_RPC_PERSIST_REC_REC_BIZ_SCENE] = ctx.rpc_persist_rec_rec_biz_scene
-    if ctx.rpc_persist_coze_record_root_id:
-        headers[HEADER_RPC_PERSIST_COZE_RECORD_ROOT_ID] = ctx.rpc_persist_coze_record_root_id
-    if ctx.rpc_persist_rec_root_entity_type:
-        headers[HEADER_RPC_PERSIST_REC_ROOT_ENTITY_TYPE] = ctx.rpc_persist_rec_root_entity_type
-    if ctx.rpc_persist_rec_root_entity_id:
-        headers[HEADER_RPC_PERSIST_REC_ROOT_ENTITY_ID] = ctx.rpc_persist_rec_root_entity_id
+    ATTR_TO_HEADER = {
+        "logid": HEADER_X_TT_LOGID,
+        "x_tt_env": HEADER_X_TT_ENV,
+        "x_use_ppe": HEADER_X_USE_PPE,
+        "x_tt_env_fe": HEADER_X_TT_ENV_FE,
+        "rpc_persist_rec_rec_biz_scene": HEADER_RPC_PERSIST_REC_REC_BIZ_SCENE,
+        "rpc_persist_coze_record_root_id": HEADER_RPC_PERSIST_COZE_RECORD_ROOT_ID,
+        "rpc_persist_rec_root_entity_type": HEADER_RPC_PERSIST_REC_ROOT_ENTITY_TYPE,
+        "rpc_persist_rec_root_entity_id": HEADER_RPC_PERSIST_REC_ROOT_ENTITY_ID,
+        "rpc_persist_rec_rec_ext_info": HEADER_RPC_PERSIST_REC_REC_EXT_INFO,
+    }
+    for attr, hk in ATTR_TO_HEADER.items():
+        v = getattr(ctx, attr)
+        if v:
+            headers[hk] = v
     return headers
 
 __all__ = [

coze_coding_utils-0.1.7/src/coze_coding_utils/__init__.py

@@ -1,3 +0,0 @@
-__all__ = []
-
-__version__ = "0.1.7"