giagentcore

type AfterToolCallContext struct {
	AssistantMessage llm.Message
	ToolCall         AgentToolCall
	Args             map[string]any
	Result           AgentToolResult
	IsError          bool
	Context          AgentContext
}

type AfterToolCallResult struct {
	Content      []llm.ContentPart
	HasContent   bool
	Details      any
	HasDetails   bool
	IsError      bool
	HasIsError   bool
	Terminate    bool
	HasTerminate bool
}

type Agent struct {
	// contains filtered or unexported fields
}

type AgentContext struct {
	SystemPrompt string
	Messages     []llm.Message
	Tools        []AgentTool
}

type AgentEvent struct {
	Type                  string
	Messages              []llm.Message
	Message               llm.Message
	ToolResults           []llm.Message
	AssistantMessageEvent llm.AssistantMessageEvent
	ToolCallID            string
	ToolName              string
	Args                  map[string]any
	PartialResult         any
	Result                any
	IsError               bool
}

type AgentEventSink func(event AgentEvent) error

type AgentLoopConfig struct {
	llm.SimpleStreamOptions
	Model llm.Model

	ConvertToLLM     func(messages []llm.Message) ([]llm.Message, error)
	TransformContext func(ctx context.Context, messages []llm.Message) ([]llm.Message, error)
	GetAPIKey        func(provider string) string

	ShouldStopAfterTurn func(ShouldStopAfterTurnContext) (bool, error)
	PrepareNextTurn     func(PrepareNextTurnContext) (AgentLoopTurnUpdate, bool, error)
	GetSteeringMessages func() ([]llm.Message, error)
	GetFollowUpMessages func() ([]llm.Message, error)

	ToolExecution  string
	BeforeToolCall func(context.Context, BeforeToolCallContext) (BeforeToolCallResult, error)
	AfterToolCall  func(context.Context, AfterToolCallContext) (AfterToolCallResult, error)
}

type AgentLoopTurnUpdate struct {
	Context       *AgentContext
	Model         *llm.Model
	ThinkingLevel *string
}

type AgentMessage = llm.Message

type AgentOption func(*AgentOptions)

type AgentOptions struct {
	InitialState     AgentState
	ConvertToLLM     func(messages []llm.Message) ([]llm.Message, error)
	TransformContext func(ctx context.Context, messages []llm.Message) ([]llm.Message, error)
	StreamFn         StreamFn
	GetAPIKey        func(provider string) string
	BeforeToolCall   func(context.Context, BeforeToolCallContext) (BeforeToolCallResult, error)
	AfterToolCall    func(context.Context, AfterToolCallContext) (AfterToolCallResult, error)
	PrepareNextTurn  func(context.Context) (AgentLoopTurnUpdate, bool, error)
	SteeringMode     string
	FollowUpMode     string
	SessionID        string
	ThinkingBudgets  map[string]int
	Transport        string
	MaxRetryDelayMs  int
	ToolExecution    string
}

type AgentState struct {
	SystemPrompt     string
	Model            llm.Model
	ThinkingLevel    string
	Tools            []AgentTool
	Messages         []llm.Message
	IsStreaming      bool
	StreamingMessage *llm.Message
	PendingToolCalls map[string]bool
	ErrorMessage     string
}

type AgentTool struct {
	Name        string
	Label       string
	Description string
	Parameters  llm.Schema

	PrepareArguments func(args any) (map[string]any, error)
	Execute          func(ctx context.Context, toolCallID string, params map[string]any, onUpdate AgentToolUpdateCallback) (AgentToolResult, error)
	ExecutionMode    string
}

type AgentToolCall = llm.ContentPart

type AgentToolResult struct {
	Content   []llm.ContentPart
	Details   any
	Terminate bool
}

type AgentToolUpdateCallback func(partialResult AgentToolResult)

type BeforeToolCallContext struct {
	AssistantMessage llm.Message
	ToolCall         AgentToolCall
	Args             map[string]any
	Context          AgentContext
}

type BeforeToolCallResult struct {
	Block  bool
	Reason string
}

type PrepareNextTurnContext = ShouldStopAfterTurnContext

type ProxyAssistantMessageEvent struct {
	Type             string    `json:"type"`
	ContentIndex     int       `json:"contentIndex,omitempty"`
	Delta            string    `json:"delta,omitempty"`
	ContentSignature string    `json:"contentSignature,omitempty"`
	ID               string    `json:"id,omitempty"`
	ToolName         string    `json:"toolName,omitempty"`
	Reason           string    `json:"reason,omitempty"`
	ErrorMessage     string    `json:"errorMessage,omitempty"`
	Usage            llm.Usage `json:"usage,omitempty"`
}

type ProxyStreamOptions struct {
	llm.SimpleStreamOptions
	AuthToken  string
	ProxyURL   string
	HTTPClient *http.Client
}

type ShouldStopAfterTurnContext struct {
	Message     llm.Message
	ToolResults []llm.Message
	Context     AgentContext
	NewMessages []llm.Message
}

type StreamFn func(model llm.Model, llmContext llm.Context, options llm.SimpleStreamOptions) (*llm.AssistantMessageEventStream, error)