opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-09-08 02:43:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: replace bare dict with dict[str, Any] in ops_trace_manager (#35070)

Browse Source
This commit is contained in:
wdeveloper16 2026-04-13 16:05:29 +02:00 committed by GitHub
parent 06b63d65d1
commit 253e8a3f98
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
api/core/ops/ops_trace_manager.py
View File

@ -324,7 +324,7 @@ class OpsTraceManager:
@classmethod @classmethod
def encrypt_tracing_config( def encrypt_tracing_config(
cls, tenant_id: str, tracing_provider: str, tracing_config: dict, current_trace_config=None cls, tenant_id: str, tracing_provider: str, tracing_config: dict[str, Any], current_trace_config=None
): ):
""" """
Encrypt tracing config. Encrypt tracing config.
@ -363,7 +363,7 @@ class OpsTraceManager:
return encrypted_config.model_dump() return encrypted_config.model_dump()
@classmethod @classmethod
def decrypt_tracing_config(cls, tenant_id: str, tracing_provider: str, tracing_config: dict): def decrypt_tracing_config(cls, tenant_id: str, tracing_provider: str, tracing_config: dict[str, Any]):
""" """
Decrypt tracing config Decrypt tracing config
:param tenant_id: tenant id :param tenant_id: tenant id
@ -408,7 +408,7 @@ class OpsTraceManager:
return dict(decrypted_config) return dict(decrypted_config)
@classmethod @classmethod
def obfuscated_decrypt_token(cls, tracing_provider: str, decrypt_tracing_config: dict): def obfuscated_decrypt_token(cls, tracing_provider: str, decrypt_tracing_config: dict[str, Any]):
""" """
Decrypt tracing config Decrypt tracing config
:param tracing_provider: tracing provider :param tracing_provider: tracing provider
@ -581,7 +581,7 @@ class OpsTraceManager:
return app_trace_config return app_trace_config
@staticmethod @staticmethod
def check_trace_config_is_effective(tracing_config: dict, tracing_provider: str): def check_trace_config_is_effective(tracing_config: dict[str, Any], tracing_provider: str):
""" """
Check trace config is effective Check trace config is effective
:param tracing_config: tracing config :param tracing_config: tracing config
@ -596,7 +596,7 @@ class OpsTraceManager:
return trace_instance(config).api_check() return trace_instance(config).api_check()
@staticmethod @staticmethod
def get_trace_config_project_key(tracing_config: dict, tracing_provider: str): def get_trace_config_project_key(tracing_config: dict[str, Any], tracing_provider: str):
""" """
get trace config is project key get trace config is project key
:param tracing_config: tracing config :param tracing_config: tracing config
@ -611,7 +611,7 @@ class OpsTraceManager:
return trace_instance(config).get_project_key() return trace_instance(config).get_project_key()
@staticmethod @staticmethod
def get_trace_config_project_url(tracing_config: dict, tracing_provider: str): def get_trace_config_project_url(tracing_config: dict[str, Any], tracing_provider: str):
""" """
get trace config is project key get trace config is project key
:param tracing_config: tracing config :param tracing_config: tracing config
@ -1322,8 +1322,8 @@ class TraceTask:
error=error, error=error,
) )
def node_execution_trace(self, **kwargs) -> WorkflowNodeTraceInfo | dict: def node_execution_trace(self, **kwargs) -> WorkflowNodeTraceInfo | dict[str, Any]:
node_data: dict = kwargs.get("node_execution_data", {}) node_data: dict[str, Any] = kwargs.get("node_execution_data", {})
if not node_data: if not node_data:
return {} return {}
@ -1431,7 +1431,7 @@ class TraceTask:
return node_trace return node_trace
return DraftNodeExecutionTrace(**node_trace.model_dump()) return DraftNodeExecutionTrace(**node_trace.model_dump())
def _extract_streaming_metrics(self, message_data) -> dict: def _extract_streaming_metrics(self, message_data) -> dict[str, Any]:
if not message_data.message_metadata: if not message_data.message_metadata:
return {} return {}