zach
/
RA.Aid
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix tests

This commit is contained in:
AI Christianson 2025-01-31 14:46:01 -05:00
parent e059ba3c49
commit 7913df4db1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/ra_aid/test_llm.py
View File

@ -128,12 +128,12 @@ def test_estimate_tokens():
# Test string content # Test string content
assert CiaynAgent._estimate_tokens("test") == 1 # 4 bytes 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 assert CiaynAgent._estimate_tokens("🚀") == 1 # 4 bytes
# Test message content # Test message content
msg = HumanMessage(content="test message") 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): def test_initialize_openai(clean_env, mock_openai):