using System.Text.Json.Serialization;

namespace Toak.Api.Models;

public class LlamaRequestMessage
{
    [JsonPropertyName("role")]
    public string Role { get; set; } = string.Empty;
    [JsonPropertyName("content")]
    public string Content { get; set; } = string.Empty;
}

public class LlamaRequest
{
    [JsonPropertyName("model")]
    public string Model { get; set; } = "llama-3.1-8b-instant";
    [JsonPropertyName("messages")]
    public LlamaRequestMessage[] Messages { get; set; } = Array.Empty<LlamaRequestMessage>();
    [JsonPropertyName("temperature")]
    public double Temperature { get; set; } = 0.0;
}

public class LlamaResponse
{
    [JsonPropertyName("choices")]
    public LlamaChoice[] Choices { get; set; } = Array.Empty<LlamaChoice>();
}

public class LlamaChoice
{
    [JsonPropertyName("message")]
    public LlamaRequestMessage Message { get; set; } = new();
}