Update context when generating a response.

buhe · May 25, 2024 · 68cf719 · 68cf719
1 parent d3155bc
commit 68cf719
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/Sources/LangChain/llms/Ollama.swift b/Sources/LangChain/llms/Ollama.swift
@@ -100,6 +100,7 @@ public class Ollama: LLM {
             return LLMResult()
         }
         let llmResponse = try JSONDecoder().decode(GenerateResponse.self, from: data)
+        context = llmResponse.context
         return LLMResult(llm_output: llmResponse.response)
     }