update passing moderations v3

langchain-ai · Nov 18, 2024 · 2f89630 · 2f89630
1 parent dc5f9a2
commit 2f89630
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/libs/ibm/langchain_ibm/llms.py b/libs/ibm/langchain_ibm/llms.py
@@ -323,9 +323,11 @@ def _create_llm_result(self, response: List[dict]) -> LLMResult:
                 Generation(
                     text=result.get("generated_text", ""),
                     generation_info={"finish_reason": result.get("stop_reason")}
-                    | {"moderations": moderations}
-                    if (moderations := result.get("moderations"))
-                    else {},
+                    | (
+                        {"moderations": moderations}
+                        if (moderations := result.get("moderations"))
+                        else {}
+                    ),
                 )
             ]
             for res in response