fmt

DataDog · Oct 2, 2024 · b2eef10 · b2eef10
1 parent e1a684d
commit b2eef10
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/contrib/langchain/test_langchain_llmobs.py b/tests/contrib/langchain/test_langchain_llmobs.py
@@ -97,7 +97,7 @@ def _invoke_llm(cls, llm, prompt, mock_tracer, cassette_name):
                     llm(prompt)
                 else:
                     llm.invoke(prompt)
-        else: # streams do not use casettes
+        else:  # streams do not use casettes
             for _ in llm.stream(prompt):
                 pass
         LLMObs.disable()
@@ -116,7 +116,7 @@ def _invoke_chat(cls, chat_model, prompt, mock_tracer, cassette_name, role="user
                     chat_model(messages)
                 else:
                     chat_model.invoke(messages)
-        else: # streams do not use casettes
+        else:  # streams do not use casettes
             for _ in chat_model.stream(prompt):
                 pass
         LLMObs.disable()
@@ -134,7 +134,7 @@ def _invoke_chain(cls, chain, prompt, mock_tracer, cassette_name, batch=False):
                 else:
                     chain.invoke(prompt)
 
-        else: # streams do not use casettes
+        else:  # streams do not use casettes
             for _ in chain.stream(prompt):
                 pass
         LLMObs.disable()