gateway

type AgentProcessor interface {
	Process(ctx context.Context, sessionID, content string) (string, error)
}

type AuthMessage struct {
	Token    string `json:"token,omitempty"`
	DeviceID string `json:"device_id,omitempty"`
}

type ChatMessage struct {
	SessionID string `json:"session_id,omitempty"`
	Content   string `json:"content"`
	Channel   string `json:"channel,omitempty"`
	ReplyTo   string `json:"reply_to,omitempty"`
}

type Client struct {
	ID string
	// contains filtered or unexported fields
}

type Config struct {
	Address         string
	ReadTimeout     time.Duration
	WriteTimeout    time.Duration
	PingInterval    time.Duration
	Logger          *slog.Logger
	Agent           AgentProcessor
	WebhookHandlers map[string]http.Handler // Path -> Handler for webhook endpoints
}

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

type EventMessage struct {
	Event   string                 `json:"event"`
	Channel string                 `json:"channel,omitempty"`
	Data    map[string]interface{} `json:"data,omitempty"`
}

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

type Message struct {
	ID        string                 `json:"id,omitempty"`
	Type      MessageType            `json:"type"`
	Channel   string                 `json:"channel,omitempty"`
	Content   string                 `json:"content,omitempty"`
	Data      map[string]interface{} `json:"data,omitempty"`
	Error     string                 `json:"error,omitempty"`
	Timestamp time.Time              `json:"timestamp,omitempty"`
}

type MessageHandler func(ctx context.Context, client *Client, msg *Message) (*Message, error)

type MessageType string

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

type ObservabilityConfig struct {
	// ServiceName is the name of this gateway service.
	ServiceName string

	// ServiceVersion is the version of this gateway service.
	ServiceVersion string

	// ObservopsProvider is the observops provider for metrics/traces.
	ObservopsProvider observops.Provider

	// AgentopsStore is the agentops store for workflow/task tracking.
	AgentopsStore agentops.Store

	// Logger is the logger for observability events.
	Logger *slog.Logger
}

type ToolInfo struct {
	Name        string `json:"name"`
	Description string `json:"description"`
	Parameters  any    `json:"parameters,omitempty"`
}

type ToolInvokeRequest struct {
	// Tool is the name of the tool to invoke.
	Tool string `json:"tool"`

	// Arguments are the tool arguments as JSON.
	Arguments json.RawMessage `json:"arguments"`

	// WorkflowID is the optional workflow ID for tracking.
	WorkflowID string `json:"workflow_id,omitempty"`

	// TaskID is the optional task ID for tracking.
	TaskID string `json:"task_id,omitempty"`

	// AgentID is the agent invoking the tool.
	AgentID string `json:"agent_id,omitempty"`

	// TraceID is the optional trace ID for correlation.
	TraceID string `json:"trace_id,omitempty"`

	// Metadata contains optional request metadata.
	Metadata map[string]any `json:"metadata,omitempty"`
}

type ToolInvokeResponse struct {
	// Result is the tool execution result.
	Result string `json:"result,omitempty"`

	// Error is the error message if execution failed.
	Error string `json:"error,omitempty"`

	// ToolName is the name of the tool that was invoked.
	ToolName string `json:"tool_name"`

	// DurationMs is the execution duration in milliseconds.
	DurationMs int64 `json:"duration_ms"`

	// InvocationID is the unique ID for this invocation.
	InvocationID string `json:"invocation_id,omitempty"`

	// TraceID is the trace ID for correlation.
	TraceID string `json:"trace_id,omitempty"`
}

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

type ToolsListResponse struct {
	Tools []ToolInfo `json:"tools"`
}

type ToolsRPCConfig struct {
	// ToolRegistry is the registry of available tools.
	ToolRegistry *agent.ToolRegistry

	// Observability provides tracing and metrics.
	Observability *Observability

	// Logger is the logger for RPC events.
	Logger *slog.Logger

	// MaxRequestSize is the maximum request body size in bytes.
	// Default is 1MB.
	MaxRequestSize int64

	// Timeout is the maximum time for a tool invocation.
	// Default is 30 seconds.
	Timeout time.Duration
}

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

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