dataflow.internal_websockets.handlers

Attributes

connection_manager
conversation_manager
credits_manager
logger
router
settings

Classes

AgentRegistry

Functions

format_chunk_for_client(chunk)	Format a streaming chunk for client consumption.
stream_agent_response(websocket, thread_id)	Stream agent responses via WebSocket.

Module Contents

class dataflow.internal_websockets.handlers.AgentRegistry

static get_agent(agent_id: str)

Async:

Parameters:

agent_id (str)

Get Agent.

Parameters:

agent_id (str) – [TODO: Add description]

dataflow.internal_websockets.handlers.format_chunk_for_client(chunk: Any) → dict[str, Any]

Format a streaming chunk for client consumption.

Parameters:

chunk (Any)

Return type:

dict[str, Any]

async dataflow.internal_websockets.handlers.stream_agent_response(websocket: fastapi.WebSocket, thread_id: str)

Stream agent responses via WebSocket.

Parameters:

• websocket (fastapi.WebSocket)

• thread_id (str)

dataflow.internal_websockets.handlers.connection_manager

dataflow.internal_websockets.handlers.conversation_manager

dataflow.internal_websockets.handlers.credits_manager

dataflow.internal_websockets.handlers.logger

dataflow.internal_websockets.handlers.router

dataflow.internal_websockets.handlers.settings