From 7913df4db10acc85e96bd4540f7752416c5f3667 Mon Sep 17 00:00:00 2001
From: AI Christianson <ai.christianson@christianson.ai>
Date: Fri, 31 Jan 2025 14:46:01 -0500
Subject: [PATCH] fix tests

---
 tests/ra_aid/test_llm.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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):