Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 8 additions & 4 deletions integrations/langgraph/python/ag_ui_langgraph/agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -235,7 +235,7 @@ async def _handle_stream_events(self, input: RunAgentInput) -> AsyncGenerator[st
CustomEvent(
type=EventType.CUSTOM,
name=LangGraphEventTypes.OnInterrupt.value,
value=json.dumps(interrupt.value, default=json_safe_stringify) if not isinstance(interrupt.value, str) else interrupt.value,
value=dump_json_safe(interrupt.value),
raw_event=interrupt,
)
)
Expand Down Expand Up @@ -311,7 +311,7 @@ async def prepare_stream(self, input: RunAgentInput, agent_state: State, config:
CustomEvent(
type=EventType.CUSTOM,
name=LangGraphEventTypes.OnInterrupt.value,
value=json.dumps(interrupt.value) if not isinstance(interrupt.value, str) else interrupt.value,
value=dump_json_safe(interrupt.value),
raw_event=interrupt,
)
)
Expand Down Expand Up @@ -741,7 +741,7 @@ async def _handle_single_event(self, event: Any, state: State) -> AsyncGenerator
ToolCallArgsEvent(
type=EventType.TOOL_CALL_ARGS,
tool_call_id=tool_call_output.tool_call_id,
delta=json.dumps(event["data"]["input"]),
delta=dump_json_safe(event["data"]["input"]),
raw_event=event
)
)
Expand All @@ -758,7 +758,7 @@ async def _handle_single_event(self, event: Any, state: State) -> AsyncGenerator
type=EventType.TOOL_CALL_RESULT,
tool_call_id=tool_call_output.tool_call_id,
message_id=str(uuid.uuid4()),
content=tool_call_output.content,
content=dump_json_safe(tool_call_output.content),
role="tool"
)
)
Expand Down Expand Up @@ -917,3 +917,7 @@ def get_stream_kwargs(
kwargs.update(fork)

return kwargs


def dump_json_safe(value):
return json.dumps(value, default=json_safe_stringify) if not isinstance(value, str) else value
2 changes: 1 addition & 1 deletion integrations/langgraph/python/pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[tool.poetry]
name = "ag-ui-langgraph"
version = "0.0.18a0"
version = "0.0.19"
description = "Implementation of the AG-UI protocol for LangGraph."
authors = ["Ran Shem Tov <[email protected]>"]
readme = "README.md"
Expand Down
Loading