feat(agent): Agent 实体(指令/工具/guardrail/handoff + 单步 RunStep,含多模态路由/调用记录)
This commit is contained in:
1 parent
b798e129c1
commit
be67d74822
6 files changed
+843
No files matched your search
@@ -0,0 +1,129 @@
|
||||
package agent
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/model"
|
||||
)
|
||||
|
||||
// Stop reasons carried by StepResult.StopReason, consumed by the loop (M2) to
|
||||
// decide how to continue after a step.
|
||||
const (
|
||||
// StopReasonStop is a normal final answer with no tool calls.
|
||||
StopReasonStop = "stop"
|
||||
// StopReasonToolCalls means the assistant requested tool calls.
|
||||
StopReasonToolCalls = "tool_calls"
|
||||
// StopReasonGuardrailInput means the input guardrail blocked the request.
|
||||
StopReasonGuardrailInput = "guardrail_input"
|
||||
// StopReasonGuardrailOutput means the output guardrail blocked the response.
|
||||
StopReasonGuardrailOutput = "guardrail_output"
|
||||
)
|
||||
|
||||
// ToolCall is a single function call the assistant requested, decoded from a
|
||||
// step. Args holds the tool arguments as a JSON object.
|
||||
type ToolCall struct {
|
||||
Name string `json:"name"`
|
||||
Args map[string]any `json:"args"`
|
||||
}
|
||||
|
||||
// StepResult is the outcome of a single agent step.
|
||||
type StepResult struct {
|
||||
// Final is true when the step produced a final answer (or was intercepted by
|
||||
// a guardrail) and no tool execution is required.
|
||||
Final bool
|
||||
// Text is the assistant's final text, or the guardrail's block reason when
|
||||
// Final is true.
|
||||
Text string
|
||||
// ToolCalls is non-empty when Final is false, listing the tool calls to run.
|
||||
ToolCalls []ToolCall
|
||||
// StopReason classifies how the step ended (see the StopReason* consts).
|
||||
StopReason string
|
||||
}
|
||||
|
||||
// RunStep executes one step of the agent loop. It assembles a model.Request from
|
||||
// the agent's instructions prepended as a leading system message plus the
|
||||
// conversation messages, runs input guardrails, calls the model adapter, runs
|
||||
// output guardrails and decodes any tool calls. sessionID identifies the
|
||||
// conversation for downstream tracing/telemetry and is currently carried without
|
||||
// a sink; it is reserved for trace/persist (M2/M9).
|
||||
//
|
||||
// A final text answer yields Final=true with StopReason "stop"; a tool request
|
||||
// yields Final=false with the decoded ToolCalls; a guardrail denial yields
|
||||
// Final=true carrying the block reason. A nil adapter or a failed guardrail or
|
||||
// model call returns a wrapped error.
|
||||
func (a *Agent) RunStep(ctx context.Context, sessionID string, messages []model.Message) (*StepResult, error) {
|
||||
req := a.buildRequest(messages)
|
||||
|
||||
if a.guards != nil {
|
||||
verdict, err := a.guards.CheckInput(ctx, &req)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("agent: check input: %w", err)
|
||||
}
|
||||
if verdict != nil && !verdict.Allow {
|
||||
return &StepResult{Final: true, Text: verdict.Reason, StopReason: StopReasonGuardrailInput}, nil
|
||||
}
|
||||
}
|
||||
|
||||
if a.adapter == nil {
|
||||
return nil, fmt.Errorf("agent: run step: %w", ErrNoAdapter)
|
||||
}
|
||||
|
||||
start := time.Now()
|
||||
resp, err := a.adapter.Chat(ctx, req, model.Options{})
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("agent: chat: %w", err)
|
||||
}
|
||||
if a.recorder != nil {
|
||||
a.recorder(model.CallRecord{
|
||||
Model: resp.Model,
|
||||
Purpose: req.Purpose,
|
||||
Tokens: resp.TotalTokens,
|
||||
TookMs: time.Since(start).Milliseconds(),
|
||||
})
|
||||
}
|
||||
|
||||
if a.guards != nil {
|
||||
verdict, err := a.guards.CheckOutput(ctx, &resp)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("agent: check output: %w", err)
|
||||
}
|
||||
if verdict != nil && !verdict.Allow {
|
||||
return &StepResult{Final: true, Text: verdict.Reason, StopReason: StopReasonGuardrailOutput}, nil
|
||||
}
|
||||
}
|
||||
|
||||
if calls, ok := extractToolCalls(resp); ok {
|
||||
return &StepResult{Final: false, ToolCalls: calls, StopReason: StopReasonToolCalls}, nil
|
||||
}
|
||||
return &StepResult{Final: true, Text: resp.Content, StopReason: StopReasonStop}, nil
|
||||
}
|
||||
|
||||
// extractToolCalls decodes tool calls from a model.Response. The current model
|
||||
// layer does not parse provider tool_calls into a structured field, so the agent
|
||||
// uses two conventions in priority order:
|
||||
//
|
||||
// 1. The normalized assistant-message placeholder: when resp.Message carries a
|
||||
// ToolCallName it is a single call with no argument payload.
|
||||
// 2. A JSON envelope in resp.Content of the form
|
||||
// {"tool_calls":[{"name":"...","args":{...}}]}.
|
||||
//
|
||||
// It reports ok=true (with a non-empty slice) when a call is decoded, and
|
||||
// ok=false otherwise.
|
||||
func extractToolCalls(resp model.Response) ([]ToolCall, bool) {
|
||||
if resp.Message.Role == model.RoleAssistant && resp.Message.ToolCallName != "" {
|
||||
return []ToolCall{{Name: resp.Message.ToolCallName, Args: map[string]any{}}}, true
|
||||
}
|
||||
var payload struct {
|
||||
ToolCalls []ToolCall `json:"tool_calls"`
|
||||
}
|
||||
if err := json.Unmarshal([]byte(resp.Content), &payload); err != nil {
|
||||
return nil, false
|
||||
}
|
||||
if len(payload.ToolCalls) == 0 {
|
||||
return nil, false
|
||||
}
|
||||
return payload.ToolCalls, true
|
||||
}
|
||||
Reference in new issue
Block a user