2f01d210f8
Add full ChatStream support to the pico channel, enabling real-time token streaming to connected WebSocket clients. Changes: - Add streamer tracking in turnState (acquire/release per conversation turn) - Integrate ChatStream in pipeline_llm (use when streamer is available) - Wire streamer lifecycle through agent loop - Add Streaming field to PicoSettings config - Skip bus publish when response was already streamed - Add streaming test scaffold When a channel supports streaming and has it enabled in config, the agent loop calls ChatStream() instead of Chat(), allowing tokens to flow in real-time through the WebSocket as the LLM generates them. Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
58 lines
2.3 KiB
Go
58 lines
2.3 KiB
Go
// PicoClaw - Ultra-lightweight personal AI agent
|
|
|
|
package interfaces
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/sipeed/picoclaw/pkg/bus"
|
|
"github.com/sipeed/picoclaw/pkg/channels"
|
|
)
|
|
|
|
// MessageBus publishes inbound and outbound messages.
|
|
// It is the primary communication channel for the agent loop.
|
|
type MessageBus interface {
|
|
// PublishInbound sends an inbound message to be processed.
|
|
PublishInbound(ctx context.Context, msg bus.InboundMessage) error
|
|
|
|
// PublishOutbound sends an outbound message to the appropriate channel.
|
|
PublishOutbound(ctx context.Context, msg bus.OutboundMessage) error
|
|
|
|
// PublishOutboundMedia sends an outbound media message.
|
|
PublishOutboundMedia(ctx context.Context, msg bus.OutboundMediaMessage) error
|
|
|
|
// InboundChan returns the channel for receiving inbound messages.
|
|
InboundChan() <-chan bus.InboundMessage
|
|
|
|
// GetStreamer returns a Streamer for the given channel+chatID if the channel
|
|
// supports streaming.
|
|
GetStreamer(ctx context.Context, channel, chatID string) (bus.Streamer, bool)
|
|
}
|
|
|
|
// ChannelManager manages channel lifecycle and provides channel access.
|
|
type ChannelManager interface {
|
|
// GetChannel returns the channel with the given name.
|
|
GetChannel(name string) (channels.Channel, bool)
|
|
|
|
// GetEnabledChannels returns the list of enabled channel names.
|
|
GetEnabledChannels() []string
|
|
|
|
// InvokeTypingStop signals that typing has stopped.
|
|
InvokeTypingStop(channel, chatID string)
|
|
|
|
// SendMessage sends a text message to the specified channel and chat.
|
|
SendMessage(ctx context.Context, msg bus.OutboundMessage) error
|
|
|
|
// SendMedia sends a media message to the specified channel and chat.
|
|
SendMedia(ctx context.Context, msg bus.OutboundMediaMessage) error
|
|
|
|
// SendPlaceholder sends a placeholder message (e.g., for audio transcription).
|
|
SendPlaceholder(ctx context.Context, channel, chatID string) bool
|
|
|
|
// DismissToolFeedback clears any tracked tool feedback animation for the
|
|
// given channel/chat. Call this when a turn ends without a final response
|
|
// (e.g., ResponseHandled tools) to avoid orphaned animation goroutines.
|
|
// outboundCtx carries topic/thread info needed for channels that use
|
|
// scoped tracker keys (e.g., Telegram forum topics); may be nil.
|
|
DismissToolFeedback(ctx context.Context, channel, chatID string, outboundCtx *bus.InboundContext)
|
|
}
|