types

type BatchSearchConfig struct {
	MaxBatchSize        int           `json:"max_batch_size"`
	MaxWaitTime         time.Duration `json:"max_wait_time"`
	EnableDeduplication bool          `json:"enable_deduplication"`
	MaxConcurrency      int           `json:"max_concurrency"`
	CacheEnabled        bool          `json:"cache_enabled"`
	CacheTTL            time.Duration `json:"cache_ttl"`
}

type BatchSearchRequest struct {
	Queries           []*SearchQuery  `json:"queries"`
	MaxConcurrency    int             `json:"max_concurrency"`
	EnableAggregation bool            `json:"enable_aggregation"`
	DeduplicationKey  string          `json:"deduplication_key"`
	Metadata          json.RawMessage `json:"metadata"`
}

type BatchSearchResponse struct {
	Results             []*SearchResponse `json:"results"`
	AggregatedResults   []*SearchResult   `json:"aggregated_results"`
	TotalProcessingTime time.Duration     `json:"total_processing_time"`
	ParallelQueries     int               `json:"parallel_queries"`
	CacheHits           int               `json:"cache_hits"`
	Metadata            json.RawMessage   `json:"metadata"`
}

type ChatMessage struct {
	Role    string `json:"role"` // "system", "user", "assistant"
	Content string `json:"content"`
}

type CircuitBreakerConfig struct {
	FailureThreshold    int64         `json:"failure_threshold"`
	FailureRate         float64       `json:"failure_rate"`
	MinimumRequestCount int64         `json:"minimum_request_count"`
	Timeout             time.Duration `json:"timeout"`
	HalfOpenTimeout     time.Duration `json:"half_open_timeout"`
	SuccessThreshold    int64         `json:"success_threshold"`
	HalfOpenMaxRequests int64         `json:"half_open_max_requests"`
	ResetTimeout        time.Duration `json:"reset_timeout"`
	SlidingWindowSize   int           `json:"sliding_window_size"`
	EnableHealthCheck   bool          `json:"enable_health_check"`
	HealthCheckInterval time.Duration `json:"health_check_interval"`
	HealthCheckTimeout  time.Duration `json:"health_check_timeout"`
	// Advanced features
	EnableAdaptiveTimeout bool `json:"enable_adaptive_timeout"`
	MaxConcurrentRequests int  `json:"max_concurrent_requests"`
}

type ComponentStatus = contracts.ComponentStatus

type ComponentType = contracts.ComponentType

type EmbeddingCacheEntry struct {
	Text        string        `json:"text"`
	Vector      []float32     `json:"vector"`
	CreatedAt   time.Time     `json:"created_at"`
	AccessCount int64         `json:"access_count"`
	TTL         time.Duration `json:"ttl"`
}

type EmbeddingCacheInterface interface {
	Get(key string) ([]float32, bool)
	Set(key string, embedding []float32, ttl time.Duration) error
	Delete(key string) error
	Clear() error
	Stats() EmbeddingCacheStats
}

type EmbeddingCacheStats struct {
	Size    int64   `json:"size"`
	Hits    int64   `json:"hits"`
	Misses  int64   `json:"misses"`
	HitRate float64 `json:"hit_rate"`
}

type HealthStatus struct {
	Healthy   bool            `json:"healthy"`
	Message   string          `json:"message"`
	LastCheck time.Time       `json:"last_check"`
	Details   json.RawMessage `json:"details"`
}

type LLMConfig struct {
	Provider    string            `json:"provider"`
	Model       string            `json:"model"`
	APIKey      string            `json:"api_key"`
	Endpoint    string            `json:"endpoint"`
	MaxTokens   int               `json:"max_tokens"`
	Temperature float32           `json:"temperature"`
	TopP        float32           `json:"top_p"`
	Timeout     time.Duration     `json:"timeout"`
	Headers     map[string]string `json:"headers"`
}

type LLMRequest struct {
	Model       string          `json:"model"`
	Messages    []ChatMessage   `json:"messages"`
	MaxTokens   int             `json:"max_tokens"`
	Temperature float32         `json:"temperature"`
	TopP        float32         `json:"top_p"`
	Stream      bool            `json:"stream"`
	Stop        []string        `json:"stop"`
	Metadata    json.RawMessage `json:"metadata"`
}

type LLMResponse struct {
	Content      string        `json:"content"`
	Model        string        `json:"model"`
	Usage        TokenUsage    `json:"usage"`
	FinishReason string        `json:"finish_reason"`
	Error        string        `json:"error,omitempty"`
	Duration     time.Duration `json:"duration"`
	RequestID    string        `json:"request_id"`
}

type NetworkIntent struct {
	ID              string          `json:"id"`
	Type            string          `json:"type"`
	Priority        int             `json:"priority"`
	Description     string          `json:"description"`
	Parameters      json.RawMessage `json:"parameters"`
	TargetResources []string        `json:"target_resources"`
	Constraints     json.RawMessage `json:"constraints"`
	CreatedAt       time.Time       `json:"created_at"`
	UpdatedAt       time.Time       `json:"updated_at"`
	Status          string          `json:"status"`
}

type PoolConfig struct {
	MinConnections int           `json:"min_connections"`
	MaxConnections int           `json:"max_connections"`
	MaxIdleTime    time.Duration `json:"max_idle_time"`
	MaxLifetime    time.Duration `json:"max_lifetime"`
	HealthCheck    bool          `json:"health_check"`
	HealthInterval time.Duration `json:"health_interval"`
}

type PoolMetrics struct {
	ActiveConnections    int64         `json:"active_connections"`
	IdleConnections      int64         `json:"idle_connections"`
	TotalConnections     int64         `json:"total_connections"`
	ConnectionsCreated   int64         `json:"connections_created"`
	ConnectionsDestroyed int64         `json:"connections_destroyed"`
	HealthChecksRun      int64         `json:"health_checks_run"`
	HealthChecksFailed   int64         `json:"health_checks_failed"`
	AverageLatency       time.Duration `json:"average_latency"`
}

type ResourceUsage struct {
	CPUPercent        float64 `json:"cpuPercent"`
	MemoryPercent     float64 `json:"memoryPercent"`
	DiskPercent       float64 `json:"diskPercent"`
	NetworkInMBps     float64 `json:"networkInMBps"`
	NetworkOutMBps    float64 `json:"networkOutMBps"`
	ActiveConnections int     `json:"activeConnections"`
}

type ScalingIntent struct {
	ID             string          `json:"id"`
	ResourceType   string          `json:"resource_type"`
	ResourceName   string          `json:"resource_name"`
	TargetScale    int             `json:"target_scale"`
	CurrentScale   int             `json:"current_scale"`
	ScaleDirection string          `json:"scale_direction"` // "up", "down", "auto"
	Reason         string          `json:"reason"`
	Metadata       json.RawMessage `json:"metadata"`
	CreatedAt      time.Time       `json:"created_at"`
	CompletedAt    *time.Time      `json:"completed_at,omitempty"`
	Status         string          `json:"status"`
}

type SearchQuery struct {
	Text       string          `json:"text"`
	MaxResults int             `json:"max_results"`
	Threshold  float64         `json:"threshold"`
	Filters    json.RawMessage `json:"filters"`
	Namespace  string          `json:"namespace"`
}

type SearchResponse struct {
	Query         string          `json:"query"`
	Results       []*SearchResult `json:"results"`
	TotalResults  int             `json:"total_results"`
	ProcessedTime time.Duration   `json:"processed_time"`
}

type SearchResult struct {
	ID        string          `json:"id"`
	Content   string          `json:"content"`
	Score     float64         `json:"score"`
	Metadata  json.RawMessage `json:"metadata"`
	Embedding []float32       `json:"embedding,omitempty"`
}

type ServiceConfig struct {
	Name           string                `json:"name"`
	Port           int                   `json:"port"`
	Host           string                `json:"host"`
	TLS            bool                  `json:"tls"`
	Timeout        time.Duration         `json:"timeout"`
	Retries        int                   `json:"retries"`
	CircuitBreaker *CircuitBreakerConfig `json:"circuit_breaker,omitempty"`
	Metadata       json.RawMessage       `json:"metadata"`
}

type SystemHealth struct {
	OverallStatus  string                      `json:"overallStatus"`
	Healthy        bool                        `json:"healthy"`
	Components     map[string]*ComponentStatus `json:"components"`
	LastUpdate     time.Time                   `json:"lastUpdate"`
	ActiveIntents  int                         `json:"activeIntents"`
	ProcessingRate float64                     `json:"processingRate"`
	ErrorRate      float64                     `json:"errorRate"`
	ResourceUsage  ResourceUsage               `json:"resourceUsage"`
}

type TokenUsage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}