agentos

type AddContentRequest struct {
	// Content 文本内容
	// Content is the text content
	Content string `json:"content" binding:"required"`

	// Metadata 元数据
	// Metadata is the metadata
	Metadata map[string]interface{} `json:"metadata,omitempty"`

	// ChunkerType 分块器类型（character/sentence/paragraph）
	// ChunkerType is the chunker type
	ChunkerType string `json:"chunker_type,omitempty"`

	// ChunkSize 块大小
	// ChunkSize is the chunk size
	ChunkSize int `json:"chunk_size,omitempty"`

	// ChunkOverlap specifies overlap between consecutive chunks
	ChunkOverlap int `json:"chunk_overlap,omitempty"`

	// CollectionName 集合名称
	// CollectionName is the collection name
	CollectionName string `json:"collection_name,omitempty"`
}

type AddContentResponse struct {
	// DocumentIDs 添加的文档 ID 列表
	// DocumentIDs is the list of added document IDs
	DocumentIDs []string `json:"document_ids"`

	// ChunkCount 生成的块数量
	// ChunkCount is the number of chunks generated
	ChunkCount int `json:"chunk_count"`

	// ChunkerType 使用的分块器类型
	// ChunkerType is the chunker type used
	ChunkerType string `json:"chunker_type"`

	// Collection 使用的集合名称
	// Collection is the collection that stored the documents
	Collection string `json:"collection,omitempty"`

	// Message 处理消息
	// Message is the processing message
	Message string `json:"message"`
}

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

type AgentRunRequest struct {
	Input      string             `json:"input"`
	SessionID  string             `json:"session_id,omitempty"`
	Stream     bool               `json:"stream,omitempty"`
	Media      interface{}        `json:"media,omitempty"`
	RunContext *RunContextRequest `json:"run_context,omitempty"`
}

type AgentRunResponse struct {
	RunID     string                 `json:"run_id,omitempty"`
	Status    agent.RunStatus        `json:"status,omitempty"`
	Content   string                 `json:"content"`
	SessionID string                 `json:"session_id,omitempty"`
	Metadata  map[string]interface{} `json:"metadata,omitempty"`
}

type ChunkerInfo struct {
	// Name 分块器名称
	// Name is the chunker name
	Name string `json:"name"`

	// Description 分块器描述
	// Description is the chunker description
	Description string `json:"description"`

	// DefaultChunkSize 默认块大小
	// DefaultChunkSize is the default chunk size
	DefaultChunkSize int `json:"default_chunk_size,omitempty"`

	// DefaultOverlap 默认重叠大小
	// DefaultOverlap is the default overlap size
	DefaultOverlap int `json:"default_overlap,omitempty"`

	// Metadata describes additional chunker-specific configuration.
	Metadata map[string]interface{} `json:"metadata,omitempty"`
}

type CompleteData struct {
	// Output 输出内容
	// Output is the output content
	Output string `json:"output"`

	// Duration 运行时长（秒）
	// Duration is the run duration in seconds
	Duration float64 `json:"duration"`

	// TokenCount 令牌数量（可选）
	// TokenCount is the token count (optional)
	TokenCount int `json:"token_count,omitempty"`

	// Reasoning 推理摘要（可选）
	// Reasoning is an optional reasoning summary
	Reasoning *ReasoningSummary `json:"reasoning,omitempty"`

	// Usage 用量统计（可选）
	// Usage provides usage statistics (optional)
	Usage *UsageMetrics `json:"usage,omitempty"`

	// Status 运行状态
	Status string `json:"status,omitempty"`

	// CacheHit 缓存命中
	CacheHit bool `json:"cache_hit,omitempty"`

	// RunID 运行标识
	RunID string `json:"run_id,omitempty"`

	// CancellationReason 取消原因
	CancellationReason string `json:"cancellation_reason,omitempty"`
}

type Config struct {
	// Server address (default: :8080)
	// 服务器地址 (默认: :8080)
	Address string

	// API route prefix (e.g., "/api/v1", "/chat", empty for no prefix)
	// API 路由前缀 (例如: "/api/v1", "/chat", 空字符串表示无前缀)
	Prefix string

	// Session storage
	// Session 存储
	SessionStorage session.Storage

	// Logger
	// 日志记录器
	Logger *slog.Logger

	// Enable debug mode
	// 启用调试模式
	Debug bool

	// CORS settings
	// CORS 设置
	AllowOrigins []string
	AllowMethods []string
	AllowHeaders []string

	// Request timeout
	// 请求超时时间
	RequestTimeout time.Duration

	// Max request size (in bytes)
	// 最大请求大小 (字节)
	MaxRequestSize int64

	// VectorDBConfig 向量数据库配置（新增）
	// VectorDBConfig is the vector database configuration (new)
	VectorDBConfig *VectorDBConfig

	// EmbeddingConfig 嵌入模型配置（新增）
	// EmbeddingConfig is the embedding model configuration (new)
	EmbeddingConfig *EmbeddingConfig

	// KnowledgeAPI 知识 API 行为控制
	// KnowledgeAPI configures knowledge endpoints behaviour
	KnowledgeAPI *KnowledgeAPIOptions

	// SummaryManager 会话摘要管理器
	SummaryManager *session.SummaryManager

	// HealthPath allows overriding the health check endpoint path.
	HealthPath string
}

type CreateSessionRequest struct {
	AgentID    string                 `json:"agent_id" binding:"required"`
	UserID     string                 `json:"user_id,omitempty"`
	TeamID     string                 `json:"team_id,omitempty"`
	WorkflowID string                 `json:"workflow_id,omitempty"`
	Name       string                 `json:"name,omitempty"`
	Metadata   map[string]interface{} `json:"metadata,omitempty"`
}

type EmbeddingConfig struct {
	// Provider 提供商（openai）
	// Provider is the provider
	Provider string

	// APIKey API 密钥
	// APIKey is the API key
	APIKey string

	// Model 模型名称
	// Model is the model name
	Model string

	// BaseURL API 基础 URL
	// BaseURL is the API base URL
	BaseURL string
}

type EmbeddingModelInfo struct {
	// Provider 提供商（openai, etc.）
	// Provider is the model provider
	Provider string `json:"provider"`

	// Model 模型名称
	// Model is the model name
	Model string `json:"model"`

	// Dimensions 嵌入维度
	// Dimensions is the embedding dimensions
	Dimensions int `json:"dimensions"`
}

type ErrorData struct {
	// Error 错误消息
	// Error is the error message
	Error string `json:"error"`

	// Code 错误代码
	// Code is the error code
	Code string `json:"code,omitempty"`

	// Details 错误详情
	// Details are error details
	Details interface{} `json:"details,omitempty"`
}

type ErrorResponse struct {
	Status  string `json:"status,omitempty"` // "error"
	Error   string `json:"error"`
	Message string `json:"message,omitempty"`
	Code    string `json:"code,omitempty"`
}

type Event struct {
	// Type 事件类型
	// Type is the event type
	Type EventType `json:"type"`

	// Timestamp 事件时间戳
	// Timestamp is the event timestamp
	Timestamp time.Time `json:"timestamp"`

	// Data 事件数据
	// Data is the event data
	Data interface{} `json:"data,omitempty"`

	// SessionID 会话 ID（可选）
	// SessionID is the session ID (optional)
	SessionID string `json:"session_id,omitempty"`

	// AgentID Agent ID（可选）
	// AgentID is the agent ID (optional)
	AgentID string `json:"agent_id,omitempty"`

	// RunContextID 运行上下文 ID（可选，用于事件关联）
	// RunContextID is the run context identifier for correlating events (optional)
	RunContextID string `json:"run_context_id,omitempty"`
}

type EventFilter struct {
	// Types 允许的事件类型（如果为空则允许所有）
	// Types are the allowed event types (if empty, all types are allowed)
	Types map[EventType]bool
}

type EventType string

type KnowledgeAPIOptions struct {
	// DisableSearch 禁用搜索端点
	DisableSearch bool

	// DisableIngestion 禁用入库端点
	DisableIngestion bool

	// EnableHealth 暴露健康检查端点
	EnableHealth bool

	// DefaultLimit 搜索默认条数
	DefaultLimit int

	// MaxLimit 搜索最大条数
	MaxLimit int

	// SearchTimeout 搜索超时时间
	SearchTimeout time.Duration

	// IngestionTimeout 入库超时时间
	IngestionTimeout time.Duration

	// HealthTimeout 健康检查超时时间
	HealthTimeout time.Duration

	// AllowedCollections 允许的集合列表 (为空时仅默认集合)
	AllowedCollections []string

	// AllowAllCollections 是否允许任意集合
	AllowAllCollections bool

	// AllowedSourceSchemes 允许的来源 URL scheme（为空表示不限制）
	AllowedSourceSchemes []string
}

type KnowledgeConfigResponse struct {
	// AvailableChunkers 可用的分块器列表
	// AvailableChunkers is the list of available chunkers
	AvailableChunkers []ChunkerInfo `json:"available_chunkers"`

	// AvailableVectorDBs 可用的向量数据库列表
	// AvailableVectorDBs is the list of available vector databases
	AvailableVectorDBs []string `json:"available_vector_dbs"`

	// DefaultChunker 默认分块器
	// DefaultChunker is the default chunker
	DefaultChunker string `json:"default_chunker"`

	// DefaultVectorDB 默认向量数据库
	// DefaultVectorDB is the default vector database
	DefaultVectorDB string `json:"default_vector_db"`

	// EmbeddingModel 嵌入模型信息
	// EmbeddingModel is the embedding model information
	EmbeddingModel EmbeddingModelInfo `json:"embedding_model"`

	// Features 功能开关
	Features KnowledgeFeatures `json:"features"`

	// Limits 搜索限制
	Limits KnowledgeLimits `json:"limits"`

	// DefaultCollection 默认集合
	DefaultCollection string `json:"default_collection"`

	// AllowedCollections 允许的集合列表
	AllowedCollections []string `json:"allowed_collections,omitempty"`

	// AllowedSourceSchemes 允许的来源 URL scheme
	AllowedSourceSchemes []string `json:"allowed_source_schemes,omitempty"`
}

type KnowledgeFeatures struct {
	SearchEnabled    bool `json:"search_enabled"`
	IngestionEnabled bool `json:"ingestion_enabled"`
	HealthEnabled    bool `json:"health_enabled"`
}

type KnowledgeLimits struct {
	DefaultLimit int `json:"default_limit"`
	MaxLimit     int `json:"max_limit"`
}

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

type KnowledgeServiceConfig struct {
	// DefaultLimit 默认返回结果数量
	// DefaultLimit is the default number of results to return
	DefaultLimit int

	// MaxLimit 最大返回结果数量
	// MaxLimit is the maximum number of results to return
	MaxLimit int

	// DefaultChunkerType 默认分块器类型
	// DefaultChunkerType is the default chunker type
	DefaultChunkerType string

	// DefaultChunkSize 默认块大小
	// DefaultChunkSize is the default chunk size
	DefaultChunkSize int

	// DefaultOverlap 默认重叠大小
	// DefaultOverlap is the default overlap size
	DefaultOverlap int

	// EmbeddingProvider 嵌入模型提供商
	// EmbeddingProvider is the embedding model provider
	EmbeddingProvider string

	// EmbeddingModel 嵌入模型名称
	// EmbeddingModel is the embedding model name
	EmbeddingModel string

	// EmbeddingDimensions 嵌入维度
	// EmbeddingDimensions is the embedding dimensions
	EmbeddingDimensions int

	// EnableSearch 是否启用搜索
	EnableSearch bool

	// EnableIngestion 是否启用知识入库
	EnableIngestion bool

	// EnableHealth 是否启用健康检查
	EnableHealth bool

	// SearchTimeout 搜索超时时间
	SearchTimeout time.Duration

	// IngestionTimeout 入库超时时间
	IngestionTimeout time.Duration

	// HealthTimeout 健康检查超时时间
	HealthTimeout time.Duration

	// DefaultCollection 默认集合名称
	DefaultCollection string

	// AllowedCollections 允许访问的集合列表
	AllowedCollections []string

	// AllowAllCollections 是否允许任意集合
	AllowAllCollections bool

	// AllowedSourceSchemes 允许的来源 URL scheme
	AllowedSourceSchemes []string
}

type PaginationMeta struct {
	// TotalCount 总结果数（注意：向量数据库可能不支持精确计数）
	// TotalCount is the total number of results
	TotalCount int `json:"total_count"`

	// Page 当前页码（基于 offset 和 limit 计算）
	// Page is the current page number
	Page int `json:"page"`

	// PageSize 每页大小
	// PageSize is the number of results per page
	PageSize int `json:"page_size"`

	// TotalPages 总页数（基于 total_count 和 page_size 计算）
	// TotalPages is the total number of pages
	TotalPages int `json:"total_pages"`

	// Offset 当前偏移量
	Offset int `json:"offset"`

	// HasMore 是否还有更多结果
	HasMore bool `json:"has_more"`

	// NextOffset 下一页的偏移量
	NextOffset int `json:"next_offset,omitempty"`
}

type ReasoningData struct {
	Content         string  `json:"content"`
	TokenCount      *int    `json:"token_count,omitempty"`
	RedactedContent *string `json:"redacted_content,omitempty"`
	MessageIndex    int     `json:"message_index"`
	Model           string  `json:"model,omitempty"`
	Provider        string  `json:"provider,omitempty"`
}

type ReasoningSummary struct {
	Content         string  `json:"content"`
	TokenCount      *int    `json:"token_count,omitempty"`
	RedactedContent *string `json:"redacted_content,omitempty"`
	Model           string  `json:"model,omitempty"`
	Provider        string  `json:"provider,omitempty"`
}

type ReuseSessionRequest struct {
	AgentID    string `json:"agent_id,omitempty"`
	TeamID     string `json:"team_id,omitempty"`
	WorkflowID string `json:"workflow_id,omitempty"`
	UserID     string `json:"user_id,omitempty"`
}

type RunContextRequest struct {
	RunID       string                 `json:"run_id,omitempty"`
	ParentRunID string                 `json:"parent_run_id,omitempty"`
	SessionID   string                 `json:"session_id,omitempty"`
	UserID      string                 `json:"user_id,omitempty"`
	WorkflowID  string                 `json:"workflow_id,omitempty"`
	TeamID      string                 `json:"team_id,omitempty"`
	Metadata    map[string]interface{} `json:"metadata,omitempty"`
}

type RunStartData struct {
	// Input 输入内容
	// Input is the input content
	Input string `json:"input"`

	// SessionID 会话 ID
	// SessionID is the session ID
	SessionID string `json:"session_id,omitempty"`

	// Media 附带的媒体资源
	// Media contains normalized media attachments
	Media []media.Attachment `json:"media,omitempty"`
}

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

type SessionResponse struct {
	SessionID  string                  `json:"session_id"`
	AgentID    string                  `json:"agent_id,omitempty"`
	UserID     string                  `json:"user_id,omitempty"`
	TeamID     string                  `json:"team_id,omitempty"`
	WorkflowID string                  `json:"workflow_id,omitempty"`
	Name       string                  `json:"name,omitempty"`
	Metadata   map[string]interface{}  `json:"metadata,omitempty"`
	State      map[string]interface{}  `json:"state,omitempty"`
	RunCount   int                     `json:"run_count"`
	CreatedAt  int64                   `json:"created_at"`
	UpdatedAt  int64                   `json:"updated_at"`
	Summary    *session.SessionSummary `json:"summary,omitempty"`
	Runs       []SessionRunMetadata    `json:"runs,omitempty"`
}

type SessionRunMetadata struct {
	RunID              string          `json:"run_id"`
	Status             agent.RunStatus `json:"status"`
	StartedAt          int64           `json:"started_at,omitempty"`
	CompletedAt        int64           `json:"completed_at,omitempty"`
	CancellationReason string          `json:"cancellation_reason,omitempty"`
	CacheHit           bool            `json:"cache_hit,omitempty"`
}

type SessionSummaryResult struct {
	Summary *session.SessionSummary `json:"summary"`
}

type StepData struct {
	// StepName 步骤名称
	// StepName is the step name
	StepName string `json:"step_name"`

	// StepIndex 步骤索引
	// StepIndex is the step index
	StepIndex int `json:"step_index"`

	// Description 步骤描述
	// Description is the step description
	Description string `json:"description,omitempty"`
}

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

type TeamToolsResponse struct {
	TeamID string                  `json:"team_id"`
	Tools  []models.ToolDefinition `json:"tools"`
}

type TokenData struct {
	// Token 生成的令牌
	// Token is the generated token
	Token string `json:"token"`

	// Index 令牌索引
	// Index is the token index
	Index int `json:"index"`
}

type ToolCallData struct {
	// ToolName 工具名称
	// ToolName is the tool name
	ToolName string `json:"tool_name"`

	// Arguments 工具参数
	// Arguments are the tool arguments
	Arguments map[string]interface{} `json:"arguments"`

	// Result 工具结果（可选）
	// Result is the tool result (optional)
	Result interface{} `json:"result,omitempty"`
}

type UpdateSessionRequest struct {
	Name     string                 `json:"name,omitempty"`
	Metadata map[string]interface{} `json:"metadata,omitempty"`
	State    map[string]interface{} `json:"state,omitempty"`
}

type UsageMetrics struct {
	PromptTokens     int `json:"prompt_tokens,omitempty"`
	CompletionTokens int `json:"completion_tokens,omitempty"`
	TotalTokens      int `json:"total_tokens,omitempty"`
	ReasoningTokens  int `json:"reasoning_tokens,omitempty"`
}

type VectorDBConfig struct {
	// Type 向量数据库类型（chromadb）
	// Type is the vector database type
	Type string

	// BaseURL 向量数据库 URL
	// BaseURL is the vector database URL
	BaseURL string

	// CollectionName 集合名称
	// CollectionName is the collection name
	CollectionName string

	// Database 数据库名称
	// Database is the database name
	Database string

	// Tenant 租户名称
	// Tenant is the tenant name
	Tenant string
}

type VectorSearchRequest struct {
	// Query 查询文本
	// Query is the search query text
	Query string `json:"query" binding:"required"`

	// Limit 返回结果数量限制（默认: 10, 最大: 100）
	// Limit is the maximum number of results to return
	Limit int `json:"limit,omitempty"`

	// Offset 分页偏移量（默认: 0）
	// Offset is the pagination offset
	Offset int `json:"offset,omitempty"`

	// Filters 元数据过滤条件
	// Filters are metadata filter conditions
	Filters map[string]interface{} `json:"filters,omitempty"`

	// CollectionName 集合名称（可选，默认使用配置的集合）
	// CollectionName is the collection to search in
	CollectionName string `json:"collection_name,omitempty"`
}

type VectorSearchResponse struct {
	// Results 搜索结果列表
	// Results is the list of search results
	Results []VectorSearchResult `json:"results"`

	// Meta 分页元数据
	// Meta is pagination metadata
	Meta PaginationMeta `json:"meta"`
}

type VectorSearchResult struct {
	// ID 文档 ID
	// ID is the document identifier
	ID string `json:"id"`

	// Content 文档内容
	// Content is the document content
	Content string `json:"content"`

	// Metadata 文档元数据
	// Metadata is the document metadata
	Metadata map[string]interface{} `json:"metadata,omitempty"`

	// Score 相似度得分（越高越相似）
	// Score is the similarity score (higher is better)
	Score float32 `json:"score"`

	// Distance 距离度量（越低越相似）
	// Distance is the distance metric (lower is better)
	Distance float32 `json:"distance"`
}