wehub-resource-sync
80 lines
4.2 KiB
C#
80 lines
4.2 KiB
C#
// Copyright (c) Microsoft. All rights reserved.
|
|
|
|
using Microsoft.Extensions.AI;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.Extensions.VectorData;
|
|
using Pinecone.ConformanceTests.Support;
|
|
using VectorData.ConformanceTests;
|
|
using VectorData.ConformanceTests.Support;
|
|
using Xunit;
|
|
|
|
namespace Pinecone.ConformanceTests;
|
|
|
|
public class PineconeEmbeddingGenerationTests(PineconeEmbeddingGenerationTests.StringVectorFixture stringVectorFixture, PineconeEmbeddingGenerationTests.RomOfFloatVectorFixture romOfFloatVectorFixture)
|
|
: EmbeddingGenerationTests<string>(stringVectorFixture, romOfFloatVectorFixture), IClassFixture<PineconeEmbeddingGenerationTests.StringVectorFixture>, IClassFixture<PineconeEmbeddingGenerationTests.RomOfFloatVectorFixture>
|
|
{
|
|
// Overriding since Pinecone requires collection names to only contain ASCII lowercase letters, digits and dashes.
|
|
public override async Task SearchAsync_string_without_generator_throws()
|
|
{
|
|
// The database doesn't support embedding generation, and no client-side generator has been configured at any level,
|
|
// so SearchAsync should throw.
|
|
var collection = stringVectorFixture.GetCollection<RawRecord>(stringVectorFixture.TestStore.DefaultVectorStore, stringVectorFixture.CollectionName + "-without-generator");
|
|
|
|
var exception = await Assert.ThrowsAsync<NotSupportedException>(() => collection.SearchAsync("foo", top: 1).ToListAsync().AsTask());
|
|
|
|
Assert.StartsWith(
|
|
"A value of type 'string' was passed to 'SearchAsync', but that isn't a supported vector type by your provider and no embedding generator was configured. The supported vector types are:",
|
|
exception.Message);
|
|
}
|
|
|
|
public new class StringVectorFixture : EmbeddingGenerationTests<string>.StringVectorFixture
|
|
{
|
|
public override TestStore TestStore => PineconeTestStore.Instance;
|
|
|
|
public override VectorStore CreateVectorStore(IEmbeddingGenerator? embeddingGenerator)
|
|
=> PineconeTestStore.Instance.GetVectorStore(new() { EmbeddingGenerator = embeddingGenerator });
|
|
|
|
public override Func<IServiceCollection, IServiceCollection>[] DependencyInjectionStoreRegistrationDelegates =>
|
|
[
|
|
services => services
|
|
.AddSingleton(PineconeTestStore.Instance.Client)
|
|
.AddPineconeVectorStore(),
|
|
services => services
|
|
.AddPineconeVectorStore("ForPineconeLocalTheApiKeysAreIgnored", PineconeTestStore.Instance.ClientOptions)
|
|
];
|
|
|
|
public override Func<IServiceCollection, IServiceCollection>[] DependencyInjectionCollectionRegistrationDelegates =>
|
|
[
|
|
services => services
|
|
.AddSingleton(PineconeTestStore.Instance.Client)
|
|
.AddPineconeCollection<RecordWithAttributes>(this.CollectionName),
|
|
services => services
|
|
.AddPineconeCollection<RecordWithAttributes>(this.CollectionName, "ForPineconeLocalTheApiKeysAreIgnored", PineconeTestStore.Instance.ClientOptions),
|
|
services => services
|
|
.AddPineconeCollection<RecordWithAttributes>(this.CollectionName, _ => new PineconeClient("ForPineconeLocalTheApiKeysAreIgnored", PineconeTestStore.Instance.ClientOptions))
|
|
];
|
|
}
|
|
|
|
public new class RomOfFloatVectorFixture : EmbeddingGenerationTests<string>.RomOfFloatVectorFixture
|
|
{
|
|
public override TestStore TestStore => PineconeTestStore.Instance;
|
|
|
|
public override VectorStore CreateVectorStore(IEmbeddingGenerator? embeddingGenerator)
|
|
=> PineconeTestStore.Instance.GetVectorStore(new() { EmbeddingGenerator = embeddingGenerator });
|
|
|
|
public override Func<IServiceCollection, IServiceCollection>[] DependencyInjectionStoreRegistrationDelegates =>
|
|
[
|
|
services => services
|
|
.AddSingleton(PineconeTestStore.Instance.Client)
|
|
.AddPineconeVectorStore()
|
|
];
|
|
|
|
public override Func<IServiceCollection, IServiceCollection>[] DependencyInjectionCollectionRegistrationDelegates =>
|
|
[
|
|
services => services
|
|
.AddSingleton(PineconeTestStore.Instance.Client)
|
|
.AddPineconeCollection<RecordWithAttributes>(this.CollectionName)
|
|
];
|
|
}
|
|
}
|