BotSharp/src/Infrastructure/BotSharp.Abstraction/VectorStorage/Models/VectorCollectionConfigModel.cs
2024-09-12 11:51:23 -05:00

46 lines
1.1 KiB
C#

namespace BotSharp.Abstraction.VectorStorage.Models;
public class VectorCollectionConfigsModel
{
[JsonPropertyName("collections")]
public List<VectorCollectionConfig> Collections { get; set; } = new();
}
public class VectorCollectionConfig
{
/// <summary>
/// Must be unique
/// </summary>
[JsonPropertyName("name")]
public string Name { get; set; }
/// <summary>
/// Collection type, e.g., question-answer, document, etc.
/// </summary>
[JsonPropertyName("type")]
public string Type { get; set; }
[JsonPropertyName("vector_store")]
public VectorStoreConfig VectorStore { get; set; }
[JsonPropertyName("text_embedding")]
public KnowledgeEmbeddingConfig TextEmbedding { get; set; }
}
public class KnowledgeEmbeddingConfig
{
[JsonPropertyName("provider")]
public string Provider { get; set; }
[JsonPropertyName("model")]
public string Model { get; set; }
[JsonPropertyName("dimension")]
public int Dimension { get; set; }
}
public class VectorStoreConfig
{
[JsonPropertyName("provider")]
public string Provider { get; set; }
}