using System.Collections.Generic;

namespace Gott.Models;

public class ChatContext
{
    public string ChainId { get; set; }
    public string SessionId { get; set; }
    public string ModelId { get; set; }
    public string Workspace { get; set; }
    public string AgentId { get; set; }
    public string SystemPromptText { get; set; }
    public List<object> SystemPrompt { get; set; }
    public string ToolChoice { get; set; }
    public bool UseMemory { get; set; }
    public string WebSearchVersion { get; set; }
    public int MaxIterations { get; set; }
    public AgentRecord Agent { get; set; }
    public SessionRecord Session { get; set; }
    public bool PlanMode { get; set; }

    // Review-Agent
    public ReviewContext Review { get; set; }

    // Finalizer-Agent
    public FinalizerContext Finalizer { get; set; }
}

public class FinalizerContext
{
    public string ModelId { get; set; }
    public List<object> SystemPrompt { get; set; }
    public string AgentId { get; set; }
    public string Workspace { get; set; }
    public bool UseMemory { get; set; }
    public string WebSearchVersion { get; set; }
    public string ToolChoice { get; set; }
    public AgentRecord Agent { get; set; }
}

public class ReviewContext
{
    public string ModelId { get; set; }
    public List<object> SystemPrompt { get; set; }
    public string AgentId { get; set; }
    public string Workspace { get; set; }
    public bool UseMemory { get; set; }
    public string WebSearchVersion { get; set; }
    public string ToolChoice { get; set; }
    public int MaxReviewIterations { get; set; }
    public AgentRecord Agent { get; set; }
}