// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System;
using System.Threading.Tasks;
using Microsoft.TestUtilities;
using Xunit;
 
namespace Microsoft.Extensions.AI;
 
public class OllamaEmbeddingGeneratorIntegrationTests : EmbeddingGeneratorIntegrationTests
{
    protected override IEmbeddingGenerator<string, Embedding<float>>? CreateEmbeddingGenerator() =>
        IntegrationTestHelpers.GetOllamaUri() is Uri endpoint ?
            new OllamaEmbeddingGenerator(endpoint, "all-minilm") :
            null;
 
    [ConditionalFact]
    public async Task InvalidModelParameter_ThrowsInvalidOperationException()
    {
        SkipIfNotEnabled();
 
        var endpoint = IntegrationTestHelpers.GetOllamaUri();
        Assert.NotNull(endpoint);
 
        using var generator = new OllamaEmbeddingGenerator(endpoint, modelId: "inexistent-model");
 
        InvalidOperationException ex;
        ex = await Assert.ThrowsAsync<InvalidOperationException>(() => generator.GenerateAsync(["Hello, world!"]));
        Assert.Contains("inexistent-model", ex.Message);
    }
}