using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace OllamaStudy.Core.OllamaResponse
{
    public class ResponseModelInfo
    {
        public string? license { get; set; }

        public string? modelfile { get; set; }

        public string? parameters { get; set; }

        public string? template { get; set; }

        public ModelDetail? details { get; set; }

        public ModelInfo? model_info { get; set; }

        public List<Tensor> tensors { get; set; } = new List<Tensor>();

        public List<string> capabilities { get; set; } = new List<string>();

        public DateTime modified_at { get; set; }
    }

    public class ModelDetail
    { 
        public string? parent_model { get; set; }
        public string? format { get; set; }
        public string? family { get; set; }
        public List<string> families { get; set; } = new List<string>();

        public string? parameter_size { get; set; }

        public string? quantization_level { get; set; }
    }

    public class Tensor
    { 
        public string? name { get; set; }

        public string? type  { get; set; }

        public List<long>? shape { get; set; }
    }

    public class ModelInfo
    {
        [JsonPropertyName("general.architecture")]
        public string? General_Architecture { get; set; }

        [JsonPropertyName("general.file_type")]
        public int General_File_Type { get; set; }

        [JsonPropertyName("general.parameter_count")]
        public long General_Parameter_Count { get; set; }

        [JsonPropertyName("general.quantization_version")]
        public int General_Quantization_Version { get; set; }

        [JsonPropertyName("tokenizer.ggml.bos_token_id")]
        public int tokenizer_ggml_bos_token_id { get; set; }

        [JsonPropertyName("tokenizer.ggml.eos_token_id")]
        public int tokenizer_ggml_eos_token_id { get; set; }

        [JsonPropertyName("tokenizer.ggml.merges")]
        public List<string> tokenizer_ggml_merges { get; set; } = new List<string>();

        [JsonPropertyName("tokenizer.ggml.model")]
        public string? tokenizer_ggml_model { get; set; }

        [JsonPropertyName("tokenizer.ggml.pre")]
        public string? tokenizer_ggml_pre { get; set; }

        [JsonPropertyName("tokenizer.ggml.token_type")]
        public List<string> tokenizer_ggml_token_type {get;set;} = new List<string>();

        [JsonPropertyName("tokenizer.ggml.tokens")]
        public List<string> tokenizer_ggml_tokens { get; set; } = new List<string>();

    }
}