fix(Ollama): Fixed #349.

tryAGI · Jun 19, 2024 · ca20d77 · ca20d77
1 parent ab445a9
commit ca20d77
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 3 deletions.
diff --git a/src/Providers/Ollama/src/OllamaChatModel.cs b/src/Providers/Ollama/src/OllamaChatModel.cs
@@ -32,7 +32,14 @@ public override async Task<ChatResponse> GenerateAsync(
     {
         request = request ?? throw new ArgumentNullException(nameof(request));
 
-        await Provider.Api.Models.PullModelAndEnsureSuccessAsync(Id, cancellationToken: cancellationToken).ConfigureAwait(false);
+        try
+        {
+            await Provider.Api.Models.PullModelAndEnsureSuccessAsync(Id, cancellationToken: cancellationToken).ConfigureAwait(false);
+        }
+        catch (HttpRequestException)
+        {
+            // Ignore
+        }
 
         var prompt = ToPrompt(request.Messages);
         var watch = Stopwatch.StartNew();

diff --git a/src/Providers/Ollama/src/OllamaEmbeddingModel.cs b/src/Providers/Ollama/src/OllamaEmbeddingModel.cs
@@ -25,8 +25,15 @@ public async Task<EmbeddingResponse> CreateEmbeddingsAsync(
     {
         request = request ?? throw new ArgumentNullException(nameof(request));
 
-        await Provider.Api.Models.PullModelAndEnsureSuccessAsync(Id, cancellationToken: cancellationToken).ConfigureAwait(false);
-
+        try
+        {
+            await Provider.Api.Models.PullModelAndEnsureSuccessAsync(Id, cancellationToken: cancellationToken).ConfigureAwait(false);
+        }
+        catch (HttpRequestException)
+        {
+            // Ignore
+        }
+
         var results = new List<IList<double>>(capacity: request.Strings.Count);
         foreach (var prompt in request.Strings)
         {