core

type ContextKey string

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

type FunctionCall struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments"`
}

type FunctionDefinition struct {
	Name        string                         `json:"name"`
	Description string                         `json:"description"`
	Parameters  map[string]ParameterDefinition `json:"parameters"`
}

type ImageContent struct {
	Path        string `json:"path"`
	Description string `json:"description"`
}

type LLMConfig struct {
	Model            string
	Temperature      float64
	MaxTokens        int
	TopP             float64
	FrequencyPenalty float64
	PresencePenalty  float64
	Stop             []string
	APIKey           string
	BaseURL          string
	APIVersion       string
	Timeout          float64
}

type LLMProvider interface {
	// Complete generates a completion for a prompt
	Complete(ctx context.Context, prompt string) (string, error)

	// CompleteWithFunctions generates a completion with function calling support
	CompleteWithFunctions(ctx context.Context, prompt string, functions []types.Function) (string, error)

	// Chat generates a response in a chat conversation
	Chat(ctx context.Context, messages []types.Message) (string, error)

	// GetAPIKey returns the API key for the provider
	GetAPIKey() string

	// GetConfig returns the current configuration
	GetConfig() *LLMConfig

	// SupportsFunction returns whether the provider supports function calling
	SupportsFunction() bool

	// GetContextWindowSize returns the context window size for the model
	GetContextWindowSize() int
}

type Message struct {
	Role         string // system, user, or assistant
	Content      string
	FunctionCall *FunctionCall  `json:"function_call,omitempty"`
	Images       []ImageContent `json:"images,omitempty"`
}

type ParameterDefinition struct {
	Type        string   `json:"type"`
	Description string   `json:"description"`
	Required    bool     `json:"required"`
	Enum        []string `json:"enum,omitempty"`
}

type Permission struct {
	Resource   ResourceType
	Action     PermissionAction
	Conditions map[string]interface{} // Additional constraints like time windows, resource tags
}

type PermissionAction string

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

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

type ResourceType string

type Role struct {
	Name        string
	Description string
	Permissions []Permission
	Hierarchy   int    // Role level in hierarchy (higher number = more privileges)
	ParentRole  string // Name of parent role for inheritance
}

type Tool interface {
	// Name returns the name of the tool
	Name() string

	// Description returns a description of what the tool does
	Description() string

	// Parameters returns the parameters that the tool accepts
	Parameters() map[string]ParameterDefinition

	// Execute executes the tool with the given input
	Execute(input string) (string, error)
}