refacto

lunary-ai · Jan 11, 2025 · 2bcae30 · 2bcae30
1 parent d4cc23f
commit 2bcae30
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 36 deletions.
diff --git a/examples/evals.py b/examples/evals.py
diff --git a/examples/langchain-rag.py → examples/langchain/v0.0.1/langchain-rag.py b/examples/langchain-rag.py → examples/langchain/v0.0.1/langchain-rag.py
diff --git a/examples/multiple_cb.py → examples/langchain/v0.0.1/multiple_cb.py b/examples/multiple_cb.py → examples/langchain/v0.0.1/multiple_cb.py
@@ -7,4 +7,4 @@
 chat = ChatOpenAI(
   callbacks=[handler1, handler2],
 )
-chat.invoke("Write a random string of 4 letters")
+chat.stream("Write a random string of 4 letters")
diff --git a/lunary/__init__.py b/lunary/__init__.py
@@ -1127,6 +1127,7 @@ def on_llm_end(
                 run_id = run_manager.end_run(run_id)
 
                 token_usage = (response.llm_output or {}).get("token_usage", {})
+
                 parsed_output: Any = [
                     (
                         _parse_lc_message(generation.message)
@@ -1136,7 +1137,6 @@ def on_llm_end(
                     for generation in response.generations[0]
                 ]
 
-                # if it's an array of 1, just parse the first element
                 if len(parsed_output) == 1:
                     parsed_output = parsed_output[0]