From f33131a9c39b8d44768f554e2689ec26fc48ccee Mon Sep 17 00:00:00 2001 From: Brandon Hancock Date: Fri, 10 Jan 2025 14:36:25 -0500 Subject: [PATCH] Trying to fix tests --- tests/llm_test.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/llm_test.py b/tests/llm_test.py index f2f8011a7..551309389 100644 --- a/tests/llm_test.py +++ b/tests/llm_test.py @@ -20,14 +20,18 @@ def test_llm_callback_replacement(): messages=[{"role": "user", "content": "Hello, world!"}], callbacks=[calc_handler_1], ) + print("result1:", result1) usage_metrics_1 = calc_handler_1.token_cost_process.get_summary() + print("usage_metrics_1:", usage_metrics_1) result2 = llm2.call( messages=[{"role": "user", "content": "Hello, world from another agent!"}], callbacks=[calc_handler_2], ) sleep(5) + print("result2:", result2) usage_metrics_2 = calc_handler_2.token_cost_process.get_summary() + print("usage_metrics_2:", usage_metrics_2) # The first handler should not have been updated assert usage_metrics_1.successful_requests == 1