From 2a25610dbfc6c20d781014d8b877a9b008783a86 Mon Sep 17 00:00:00 2001
From: TesAnti <8780022+TesAnti@users.noreply.github.com>
Date: Thu, 18 Jan 2024 22:47:44 +0100
Subject: [PATCH] fix: Small fixes (#101)

* fixed error in Retreival chain

* set ollama instruction model url to default value
---
 src/libs/LangChain.Core/Chains/Chain.cs                       | 2 +-
 .../OllamaLanguageModelInstruction.cs                         | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/libs/LangChain.Core/Chains/Chain.cs b/src/libs/LangChain.Core/Chains/Chain.cs
index bc7a2b23..6f6caee3 100644
--- a/src/libs/LangChain.Core/Chains/Chain.cs
+++ b/src/libs/LangChain.Core/Chains/Chain.cs
@@ -95,7 +95,7 @@ public static RetrieveDocumentsChain RetrieveDocuments(
         VectorStoreIndexWrapper index,
         int amount = 4,
         string inputKey = "text",
-        string outputKey = "text")
+        string outputKey = "docs")
     {
         return new RetrieveDocumentsChain(index, inputKey, outputKey, amount);
     }
diff --git a/src/libs/Providers/LangChain.Providers.Ollama/OllamaLanguageModelInstruction.cs b/src/libs/Providers/LangChain.Providers.Ollama/OllamaLanguageModelInstruction.cs
index 738491b2..0c4fbbbd 100644
--- a/src/libs/Providers/LangChain.Providers.Ollama/OllamaLanguageModelInstruction.cs
+++ b/src/libs/Providers/LangChain.Providers.Ollama/OllamaLanguageModelInstruction.cs
@@ -37,12 +37,12 @@ public class OllamaLanguageModelInstruction : IChatModel
     /// <param name="options"></param>
     public OllamaLanguageModelInstruction(
         string modelName,
-        string? url = null,
+        string url = "http://localhost:11434",
         OllamaLanguageModelOptions? options = null)
     {
         _modelName = modelName;
         Options = options ?? new OllamaLanguageModelOptions();
-        _api = new OllamaApiClient(url ?? "http://localhost:11434");
+        _api = new OllamaApiClient(url);
         
     }
     /// <summary>