Revert "feat: add langfuse llm node input and output" (langgenius#16947)

Author: crazywoola

api/core/ops/langfuse_trace/langfuse_trace.py

@@ -29,7 +29,7 @@
 )
 from core.ops.utils import filter_none_values
 from extensions.ext_database import db
-from models.model import EndUser, Message
+from models.model import EndUser
 from models.workflow import WorkflowNodeExecution
 
 logger = logging.getLogger(__name__)
@@ -213,32 +213,9 @@ def workflow_trace(self, trace_info: WorkflowTraceInfo):
 
             if process_data and process_data.get("model_mode") == "chat":
                 total_token = metadata.get("total_tokens", 0)
-
-                # through workflow_run_id get message data
-                message_data = (
-                    db.session.query(
-                        Message.answer_tokens,  # input
-                        Message.message_tokens,  # output
-                    )
-                    .filter(Message.workflow_run_id == trace_info.workflow_run_id)
-                    .first()
-                )
-
-                if message_data:
-                    # chatflow data
-                    input_tokens = message_data.message_tokens
-                    output_tokens = message_data.answer_tokens
-                else:
-                    # workflow data
-                    input_tokens = json.loads(node_execution.outputs).get("usage", {}).get("prompt_tokens", 0)
-                    output_tokens = json.loads(node_execution.outputs).get("usage", {}).get("completion_tokens", 0)
-
                 # add generation
                 generation_usage = GenerationUsage(
                     total=total_token,
-                    input=input_tokens,
-                    output=output_tokens,
-                    unit=UnitEnum.TOKENS,
                 )
 
                 node_generation_data = LangfuseGeneration(

api/tasks/ops_trace_task.py

@@ -49,6 +49,6 @@ def process_trace_tasks(file_info):
     except Exception:
         failed_key = f"{OPS_TRACE_FAILED_KEY}_{app_id}"
         redis_client.incr(failed_key)
-        logging.exception(f"Processing trace tasks failed, app_id: {app_id}")
+        logging.info(f"Processing trace tasks failed, app_id: {app_id}")
     finally:
         storage.delete(file_path)