llm

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

type AgentConfig struct {
	// LLM settings
	BaseURL         string
	APIKey          string
	Model           string
	ReasoningEffort string // Reasoning effort level

	// Agent settings
	DeviceID        string
	Lang            string // "zh" or "en"
	MaxSteps        int
	Stream          bool
	SystemPrompt    string // Custom system prompt (optional)
	UseTaskPlanning bool   // Whether to use task decomposition planning

	// Callbacks for streaming
	OnToken func(token string)
}

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

type Config struct {
	BaseURL string // Custom API base URL (e.g., for proxies or alternative providers)
	APIKey  string // API key for authentication
	Model   string // Model name (e.g., "gpt-4-vision-preview")
}

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

type LLMInteractionRecord struct {
	TaskID       string                 `json:"task_id"`
	StepIndex    int                    `json:"step_index"` // -1 for PlanTask, >=0 for step actions
	RequestData  map[string]interface{} `json:"request_data"`
	ResponseData map[string]interface{} `json:"response_data"`
	LatencyMs    int64                  `json:"latency_ms"`
	Success      bool                   `json:"success"`
	Error        string                 `json:"error,omitempty"`
}

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

type LoggingMeta struct {
	BaseURL string
	Model   string
}

type MemoryConfig struct {
	Type       MemoryType // Type of memory to use
	MaxHistory int        // Maximum number of messages to keep (for window type)
}

type MemoryType string

type OperationRecord struct {
	StepIndex int
	Thinking  string
	Action    string
	Result    string
	Timestamp time.Time
}

type ParsedAction struct {
	Name string
	Args map[string]interface{}
	Raw  string
}

type PlanStepActionsResult struct {
	Actions     []string              `json:"actions"` // Planned actions for the step
	Interaction *LLMInteractionRecord `json:"-"`       // LLM interaction record for database storage
}

type PlanTaskResult struct {
	Steps       []string              `json:"steps"` // Planned task steps
	Interaction *LLMInteractionRecord `json:"-"`     // LLM interaction record for database storage
}

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

type PlatformCapturer interface {
	CaptureState(ctx context.Context) (uiagent.ScreenInfo, error)
}

type PlatformWithApps interface {
	GetInstalledApps(ctx context.Context) (map[string]string, error)
}

type PromptConfig struct {
	Language    PromptLanguage
	OSVersion   string
	DeviceModel string
}

type PromptLanguage string

type RunLog struct {
	Task      string
	DeviceID  string
	Steps     []StepExecution
	Variables map[string]interface{}
}

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

type StepExecution struct {
	Index     int
	Action    string
	Result    uiagent.StepExecResult
	Timestamp time.Time
}

type StepResult struct {
	Success  bool
	Finished bool
	Thinking string
	Message  string
	Action   *ParsedAction
	Log      *RunLog
	Err      error
}

type StepRunner interface {
	RunStep(ctx context.Context, step uiagent.ActionStep) (uiagent.StepExecResult, error)
}

type UIAgent interface {
	Run(ctx context.Context, task string) (*StepResult, error)
	Step(ctx context.Context, task string) (*StepResult, error)
	PlanTask(ctx context.Context, task string) (*PlanTaskResult, error)
	PlanStepActions(ctx context.Context, step string) (*PlanStepActionsResult, error)
	Reset()
	GetLog() *RunLog
	SetTaskID(taskID string)
	GetPlanSteps() []string
	SetPlanSteps(steps []string)
}