"Update cache only if llmResult has non-nil llm_output."

buhe · Jan 21, 2024 · 27a817b · 27a817b
1 parent e91e4d1
commit 27a817b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Sources/LangChain/llms/LLM.swift b/Sources/LangChain/llms/LLM.swift
@@ -36,7 +36,9 @@ public class LLM {
             }
             let llmResult = try await _send(text: text, stops: stops)
             if let cache = self.cache {
-                await cache.update(prompt: text, return_val: llmResult)
+                if llmResult.llm_output != nil {
+                    await cache.update(prompt: text, return_val: llmResult)
+                }
             }
             cost = Date.now.timeIntervalSince1970 - now
             if !llmResult.stream {