mirror of
https://github.com/MindWorkAI/AI-Studio.git
synced 2026-10-18 00:18:00 +00:00
Port the Anthropic and Google families
This commit is contained in:
8 files changed
+250
-1
No files matched your search
@@ -0,0 +1,35 @@
|
||||
using AIStudio.Provider;
|
||||
|
||||
using static AIStudio.Provider.Capability;
|
||||
|
||||
namespace AIStudio.Models.Google;
|
||||
|
||||
/// <summary>
|
||||
/// Google's embedding models, which turn text into a vector and answer nothing.
|
||||
/// </summary>
|
||||
/// <remarks>
|
||||
/// The previous rules answered for these with the Google default: images in, text out, tool
|
||||
/// calling. None of it is true, and the app already knows better -- it asks every provider for its
|
||||
/// embedding models through a method of its own.
|
||||
///
|
||||
/// The Gemini one needs a rule of its own for another reason: its name begins with "gemini", so
|
||||
/// without one it would be read as a chat model of the family.
|
||||
/// </remarks>
|
||||
public sealed class GoogleEmbeddingFamily : ModelFamily
|
||||
{
|
||||
/// <inheritdoc />
|
||||
public override ModelVendor Vendor => ModelVendor.GOOGLE;
|
||||
|
||||
/// <inheritdoc />
|
||||
public override ModelSource Source => new("https://ai.google.dev/gemini-api/docs/embeddings", new DateOnly(2026, 9, 11), "The app lists these under IProvider.GetEmbeddingModels, which is where the statement that they embed comes from.");
|
||||
|
||||
/// <inheritdoc />
|
||||
protected override void Declare(ModelFamilyBuilder builder)
|
||||
{
|
||||
builder.Rule("text-embedding-004").AsExact()
|
||||
.Capabilities(TEXT_INPUT | EMBEDDING)
|
||||
.Kind(ModelKind.EMBEDDING);
|
||||
|
||||
builder.Rule("gemini-embedding").AsPrefix().Inherits();
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user