Quellcode durchsuchen

Langfuse view button (#7684)

Charlie.Wei vor 8 Monaten
Ursprung
Commit
e7afee1176
2 geänderte Dateien mit 5 neuen und 7 gelöschten Zeilen
  1. 0 1
      api/core/ops/entities/config_entity.py
  2. 5 6
      api/services/ops_service.py

+ 0 - 1
api/core/ops/entities/config_entity.py

@@ -21,7 +21,6 @@ class LangfuseConfig(BaseTracingConfig):
     """
     public_key: str
     secret_key: str
-    project_key: str
     host: str = 'https://api.langfuse.com'
 
     @field_validator("host")

+ 5 - 6
api/services/ops_service.py

@@ -26,16 +26,15 @@ class OpsService:
         decrypt_tracing_config = OpsTraceManager.decrypt_tracing_config(
             tenant_id, tracing_provider, trace_config_data.tracing_config
         )
+        new_decrypt_tracing_config = OpsTraceManager.obfuscated_decrypt_token(tracing_provider, decrypt_tracing_config)
+
         if tracing_provider == "langfuse" and (
             "project_key" not in decrypt_tracing_config or not decrypt_tracing_config.get("project_key")
         ):
             project_key = OpsTraceManager.get_trace_config_project_key(decrypt_tracing_config, tracing_provider)
-            decrypt_tracing_config["project_key"] = project_key
-
-        decrypt_tracing_config = OpsTraceManager.obfuscated_decrypt_token(tracing_provider, decrypt_tracing_config)
-
-        trace_config_data.tracing_config = decrypt_tracing_config
+            new_decrypt_tracing_config.update({"project_key": project_key})
 
+        trace_config_data.tracing_config = new_decrypt_tracing_config
         return trace_config_data.to_dict()
 
     @classmethod
@@ -79,7 +78,7 @@ class OpsService:
         # get tenant id
         tenant_id = db.session.query(App).filter(App.id == app_id).first().tenant_id
         tracing_config = OpsTraceManager.encrypt_tracing_config(tenant_id, tracing_provider, tracing_config)
-        if tracing_provider == "langfuse":
+        if tracing_provider == "langfuse" and project_key:
             tracing_config["project_key"] = project_key
         trace_config_data = TraceAppConfig(
             app_id=app_id,