diff --git a/tests/ra_aid/test_llm.py b/tests/ra_aid/test_llm.py index 831ed1c..69a72d5 100644 --- a/tests/ra_aid/test_llm.py +++ b/tests/ra_aid/test_llm.py @@ -128,12 +128,12 @@ def test_estimate_tokens(): # Test string content assert CiaynAgent._estimate_tokens("test") == 1 # 4 bytes - assert CiaynAgent._estimate_tokens("hello world") == 2 # 11 bytes + assert CiaynAgent._estimate_tokens("hello world") == 3 # 11 bytes assert CiaynAgent._estimate_tokens("🚀") == 1 # 4 bytes # Test message content msg = HumanMessage(content="test message") - assert CiaynAgent._estimate_tokens(msg) == 3 # 11 bytes + assert CiaynAgent._estimate_tokens(msg) == 4 # 11 bytes def test_initialize_openai(clean_env, mock_openai):