Skip to content

Commit 3632280

Browse files
test: cover context-to-action bridge payload mapping
Agent-Logs-Url: https://github.com/MervinPraison/PraisonAI/sessions/5012c814-9f16-4594-84a6-a9b9cec978e5 Co-authored-by: MervinPraison <454862+MervinPraison@users.noreply.github.com>
1 parent b26fe19 commit 3632280

1 file changed

Lines changed: 12 additions & 15 deletions

File tree

src/praisonai/praisonai/observability/langfuse.py

Lines changed: 12 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -67,18 +67,15 @@ def _convert_context_to_action(self, ctx_event: ContextEvent) -> Optional[Action
6767
if not action_type:
6868
return None
6969

70-
tool_result_summary = (
71-
ctx_event.data.get("tool_result_summary")
72-
if isinstance(ctx_event.data, dict)
73-
else None
74-
)
75-
if tool_result_summary is None and isinstance(ctx_event.data, dict):
76-
tool_result_summary = ctx_event.data.get("result")
70+
event_data = ctx_event.data if isinstance(ctx_event.data, dict) else {}
71+
tool_result_summary = event_data.get("tool_result_summary")
72+
if tool_result_summary is None:
73+
tool_result_summary = event_data.get("result")
7774

78-
status = ctx_event.data.get("status") if isinstance(ctx_event.data, dict) else None
79-
error_message = ctx_event.data.get("error_message") if isinstance(ctx_event.data, dict) else None
80-
if error_message is None and isinstance(ctx_event.data, dict):
81-
error_message = ctx_event.data.get("error")
75+
status = event_data.get("status")
76+
error_message = event_data.get("error_message")
77+
if error_message is None:
78+
error_message = event_data.get("error")
8279
if status is None and action_type == ActionEventType.TOOL_END:
8380
status = "error" if error_message else "completed"
8481

@@ -88,13 +85,13 @@ def _convert_context_to_action(self, ctx_event: ContextEvent) -> Optional[Action
8885
timestamp=ctx_event.timestamp,
8986
agent_id=ctx_event.session_id,
9087
agent_name=ctx_event.agent_name or "unknown",
91-
tool_name=ctx_event.data.get("tool_name"),
92-
tool_args=ctx_event.data.get("tool_args"),
88+
tool_name=event_data.get("tool_name"),
89+
tool_args=event_data.get("tool_args"),
9390
tool_result_summary=tool_result_summary,
94-
duration_ms=ctx_event.data.get("duration_ms", 0.0),
91+
duration_ms=event_data.get("duration_ms", 0.0),
9592
status=status,
9693
error_message=error_message,
97-
metadata=ctx_event.data,
94+
metadata=event_data,
9895
)
9996

10097

0 commit comments

Comments
 (0)