大森
e18dde7c15
- Add bubbletea/lipgloss/glamour dependencies for TUI - Create internal/tui package with EventWriter, styles, and bubbletea Model - Support streaming output display in conversation window - Add right panel with statistics and active agent status - Implement multi-agent collaboration with sub-agents - Add AgentCallTool for delegating tasks to sub-agents - Support parallel tool execution with goroutines - Auto-discover sub-agents from ~/.orca/prompts/ directory - Fix orchestrator routing based on msg.To field - Add non-blocking event writer with timeout to prevent blocking
88 lines
2.5 KiB
Go
88 lines
2.5 KiB
Go
// Package actor 为 Orca 框架实现 Actor 模型。
|
|
package actor
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/orca/orca/pkg/bus"
|
|
)
|
|
|
|
// Worker 是一个处理任务并进行工具调用的智能体。
|
|
//
|
|
// Worker 是 Actor 系统中的执行单元。它们接收来自编排器的
|
|
// 任务请求，处理这些任务（可能需要进行工具调用），并返回结果。
|
|
type Worker struct {
|
|
*BaseAgent
|
|
}
|
|
|
|
// NewWorker 使用给定的 ID 创建一个新的 Worker 智能体。
|
|
// 智能体在创建时会自动启动。
|
|
func NewWorker(id string) *Worker {
|
|
w := &Worker{
|
|
BaseAgent: NewBaseAgent(id, "worker"),
|
|
}
|
|
w.SetHandler(w.handleMessage)
|
|
if err := w.Start(); err != nil {
|
|
panic(fmt.Sprintf("worker: failed to start: %v", err))
|
|
}
|
|
return w
|
|
}
|
|
|
|
// handleMessage 将传入的消息路由到适当的处理程序。
|
|
func (w *Worker) handleMessage(ctx context.Context, msg bus.Message) (bus.Message, error) {
|
|
switch msg.Type {
|
|
case bus.MsgTypeTaskRequest:
|
|
return w.handleTask(ctx, msg)
|
|
case bus.MsgTypeToolCall:
|
|
return w.handleToolCall(ctx, msg)
|
|
case bus.MsgTypeSystem:
|
|
return w.handleSystem(ctx, msg)
|
|
default:
|
|
return bus.Message{}, fmt.Errorf("worker %s: unsupported message type %s", w.ID(), msg.Type)
|
|
}
|
|
}
|
|
|
|
// handleTask 处理任务请求并返回任务响应。
|
|
func (w *Worker) handleTask(ctx context.Context, msg bus.Message) (bus.Message, error) {
|
|
// Process the task - in a real implementation this would involve
|
|
// the LLM, tool calls, etc.
|
|
return bus.Message{
|
|
ID: msg.ID + "-response",
|
|
Type: bus.MsgTypeTaskResponse,
|
|
From: w.ID(),
|
|
To: msg.From,
|
|
Content: msg.Content,
|
|
Metadata: map[string]string{
|
|
"processed_by": w.ID(),
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
// handleToolCall 处理工具调用请求，转换到 WaitingForTool 状态，并返回结果。
|
|
func (w *Worker) handleToolCall(ctx context.Context, msg bus.Message) (bus.Message, error) {
|
|
w.setStatus(StatusWaitingForTool)
|
|
defer w.setStatus(StatusProcessing)
|
|
|
|
// In a real implementation, this would invoke the actual tool.
|
|
// For now, acknowledge the tool call.
|
|
return bus.Message{
|
|
ID: msg.ID + "-result",
|
|
Type: bus.MsgTypeToolResult,
|
|
From: w.ID(),
|
|
To: msg.From,
|
|
Content: msg.Content,
|
|
}, nil
|
|
}
|
|
|
|
// handleSystem 处理内部系统消息。
|
|
func (w *Worker) handleSystem(ctx context.Context, msg bus.Message) (bus.Message, error) {
|
|
return bus.Message{
|
|
ID: msg.ID + "-ack",
|
|
Type: bus.MsgTypeSystem,
|
|
From: w.ID(),
|
|
To: msg.From,
|
|
Content: "worker acknowledged",
|
|
}, nil
|
|
}
|