GenerativeConfig.cs

using System.Text.Json.Serialization;

namespace Weaviate.Client.Models;

public interface IGenerativeConfig
{
    [JsonIgnore]
    string Type { get; }
}

public static class GenerativeConfig
{
    public record Custom : IGenerativeConfig
    {
        public required string Type { get; set; }

        public dynamic? Config { get; set; } = new { };
    }

    public abstract record OpenAIBase : IGenerativeConfig
    {
        public string? BaseURL { get; set; }
        public int? FrequencyPenalty { get; set; }
        public int? MaxTokens { get; set; }
        public int? PresencePenalty { get; set; }
        public double? Temperature { get; set; }
        public double? TopP { get; set; }
        public abstract string Type { get; }
    }

    public record AWS : IGenerativeConfig
    {
        public const string TypeValue = "generative-aws";
        public string Type => TypeValue;

        public string Region { get; set; } = string.Empty;
        public string Service { get; set; } = string.Empty;
        public string? Model { get; set; }
        public string? Endpoint { get; set; }
    }

    public record Anthropic : IGenerativeConfig
    {
        public const string TypeValue = "generative-anthropic";
        public string Type => TypeValue;

        public string? BaseURL { get; set; }
        public int? MaxTokens { get; set; }
        public string? Model { get; set; }
        public string[]? StopSequences { get; set; }
        public double? Temperature { get; set; }
        public int? TopK { get; set; }
        public double? TopP { get; set; }
    }

    public record Anyscale : IGenerativeConfig
    {
        public const string TypeValue = "generative-anyscale";
        public string Type => TypeValue;

        public string? BaseURL { get; set; }
        public string? Model { get; set; }
        public double? Temperature { get; set; }
    }

    public record Cohere : IGenerativeConfig
    {
        public const string TypeValue = "generative-cohere";
        public string Type => TypeValue;

        public int? K { get; set; }
        public string? Model { get; set; }
        public int? MaxTokens { get; set; }
        public string? ReturnLikelihoods { get; set; }
        public string[]? StopSequences { get; set; }
        public double? Temperature { get; set; }
    }

    public record Databricks : IGenerativeConfig
    {
        public const string TypeValue = "generative-databricks";
        public string Type => TypeValue;

        public string Endpoint { get; set; } = string.Empty;
        public int? MaxTokens { get; set; }
        public double? Temperature { get; set; }
        public int? TopK { get; set; }
        public double? TopP { get; set; }
    }

    public record FriendliAI : IGenerativeConfig
    {
        public const string TypeValue = "generative-friendliai";
        public string Type => TypeValue;

        public string? BaseURL { get; set; }
        public int? MaxTokens { get; set; }
        public string? Model { get; set; }
        public double? Temperature { get; set; }
    }

    public record Mistral : IGenerativeConfig
    {
        public const string TypeValue = "generative-mistral";
        public string Type => TypeValue;

        public string? BaseURL { get; set; }
        public int? MaxTokens { get; set; }
        public string? Model { get; set; }
        public double? Temperature { get; set; }
    }

    public record Nvidia : IGenerativeConfig
    {
        public const string TypeValue = "generative-nvidia";
        public string Type => TypeValue;

        public string? BaseURL { get; set; }
        public int? MaxTokens { get; set; }
        public string? Model { get; set; }
        public double? Temperature { get; set; }
    }

    public record Ollama : IGenerativeConfig
    {
        public const string TypeValue = "generative-ollama";
        public string Type => TypeValue;

        public string? ApiEndpoint { get; set; }
        public string? Model { get; set; }
    }

    public record OpenAI : OpenAIBase
    {
        public const string TypeValue = "generative-openai";
        public override string Type => TypeValue;

        public string? Model { get; set; }
    }

    public record AzureOpenAI : OpenAIBase
    {
        public const string TypeValue = "generative-azure-openai";
        public override string Type => TypeValue;

        public string ResourceName { get; set; } = string.Empty;
        public string DeploymentId { get; set; } = string.Empty;
    }

    public record Google : IGenerativeConfig
    {
        public const string TypeValue = "generative-google";
        public string Type => TypeValue;

        public string? ApiEndpoint { get; set; }
        public int? MaxOutputTokens { get; set; }
        public string? ModelId { get; set; }
        public string? ProjectId { get; set; }
        public double? Temperature { get; set; }
        public int? TopK { get; set; }
        public double? TopP { get; set; }
    }

    public record XAI : IGenerativeConfig
    {
        public const string TypeValue = "generative-xai";
        public string Type => TypeValue;

        public string? BaseURL { get; set; }
        public int? MaxTokens { get; set; }
        public string? Model { get; set; }
        public double? Temperature { get; set; }
        public double? TopP { get; set; }
    }
}