Skip to content

Commit

Permalink
update test results for no tags by default (#573)
Browse files Browse the repository at this point in the history
  • Loading branch information
granawkins authored Apr 25, 2024
1 parent b830d83 commit 1200fb7
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
6 changes: 3 additions & 3 deletions tests/code_context_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -213,20 +213,20 @@ def func_4(string):
mock_session_context.config.auto_context_tokens = 8000

code_message = await code_context.get_code_message(0, prompt="prompt")
assert mock_session_context.llm_api_handler.spice.count_tokens(code_message, "gpt-4", is_message=True) == 95 # Code
assert mock_session_context.llm_api_handler.spice.count_tokens(code_message, "gpt-4", is_message=True) == 85 # Code
assert (
code_message
== """\
Code Files:
file_1.py (search-result, user-included)
file_1.py
1:def func_1(x, y):
2: return x + y
3:
4:def func_2():
5: return 3
file_2.py (search-result)
file_2.py
1:def func_3(a, b, c):
2: return a * b ** c
3:
Expand Down
2 changes: 1 addition & 1 deletion tests/code_file_manager_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ async def test_partial_files(mocker, mock_session_context):
"""\
Code Files:
dir/file.txt (user-included)
dir/file.txt
1:I am a file
...
3:third
Expand Down

0 comments on commit 1200fb7

Please sign in to comment.