fix for LlmChain OutputKey. The result of CallAsync was always contai…

…ning only ["text'] key ignoring OutputKey. This was causing an error when used with SequentialChain
tryAGI · Oct 17, 2023 · 9ea137c · 9ea137c
1 parent fdebe46
commit 9ea137c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/libs/LangChain.Core/Chains/LLM/LLMChain.cs b/src/libs/LangChain.Core/Chains/LLM/LLMChain.cs
@@ -58,8 +58,9 @@ public override async Task<IChainValues> CallAsync(IChainValues values)
 
         BasePromptValue promptValue = await Prompt.FormatPromptValue(new InputValues(values.Value));
         var response = await Llm.GenerateAsync(new ChatRequest(promptValue.ToChatMessages(), stop));
-
-        return new ChainValues(response.Messages.Last().Content);
+        if(string.IsNullOrEmpty(OutputKey))
+            return new ChainValues(response.Messages.Last().Content);
+        return new ChainValues(OutputKey,response.Messages.Last().Content);
     }
 
     public async Task<object> Predict(ChainValues values)