klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-09 13:25:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Resolve Pydantic deprecation warnings

Browse Source 
Signed-off-by: Emmanuel Ferdman <emmanuelferdman@gmail.com>
This commit is contained in:
Emmanuel Ferdman
2025-07-25 16:55:00 -07:00
parent f5d12b7546
commit 655a1b38f2
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_conversation_utils.py
View File

@@ -91,7 +91,7 @@ class TestTruncateMessage:
# Arrange # Arrange
chat_history = generate_chat_history(5) chat_history = generate_chat_history(5)
big_chat_message = ChatMessage(role="user", content=generate_content(100, suffix="Question?")) big_chat_message = ChatMessage(role="user", content=generate_content(100, suffix="Question?"))
copy_big_chat_message = big_chat_message.copy() copy_big_chat_message = big_chat_message.model_copy()
chat_history.insert(0, big_chat_message) chat_history.insert(0, big_chat_message)
initial_tokens = sum([utils.count_tokens(message.content, self.encoder) for message in chat_history]) initial_tokens = sum([utils.count_tokens(message.content, self.encoder) for message in chat_history])
@@ -114,7 +114,7 @@ class TestTruncateMessage:
chat_history = generate_chat_history(5) chat_history = generate_chat_history(5)
chat_history[0].role = "system" # Mark the first message as system message chat_history[0].role = "system" # Mark the first message as system message
big_chat_message = ChatMessage(role="user", content=generate_content(100, suffix="Question?")) big_chat_message = ChatMessage(role="user", content=generate_content(100, suffix="Question?"))
copy_big_chat_message = big_chat_message.copy() copy_big_chat_message = big_chat_message.model_copy()
chat_history.insert(0, big_chat_message) chat_history.insert(0, big_chat_message)
initial_tokens = sum([utils.count_tokens(message.content, self.encoder) for message in chat_history]) initial_tokens = sum([utils.count_tokens(message.content, self.encoder) for message in chat_history])
@@ -138,7 +138,7 @@ class TestTruncateMessage:
def test_truncate_single_large_non_system_message(self): def test_truncate_single_large_non_system_message(self):
# Arrange # Arrange
big_chat_message = ChatMessage(role="user", content=generate_content(100, suffix="Question?")) big_chat_message = ChatMessage(role="user", content=generate_content(100, suffix="Question?"))
copy_big_chat_message = big_chat_message.copy() copy_big_chat_message = big_chat_message.model_copy()
chat_messages = [big_chat_message] chat_messages = [big_chat_message]
initial_tokens = sum([utils.count_tokens(message.content, self.encoder) for message in chat_messages]) initial_tokens = sum([utils.count_tokens(message.content, self.encoder) for message in chat_messages])
@@ -160,7 +160,7 @@ class TestTruncateMessage:
# Arrange # Arrange
big_chat_message_content = [{"type": "text", "text": " ".join(["hi"] * (self.max_prompt_size + 1))}] big_chat_message_content = [{"type": "text", "text": " ".join(["hi"] * (self.max_prompt_size + 1))}]
big_chat_message = ChatMessage(role="user", content=big_chat_message_content) big_chat_message = ChatMessage(role="user", content=big_chat_message_content)
copy_big_chat_message = big_chat_message.copy() copy_big_chat_message = big_chat_message.model_copy()
chat_messages = [big_chat_message] chat_messages = [big_chat_message]
initial_tokens = sum([utils.count_tokens(message.content, self.encoder) for message in chat_messages]) initial_tokens = sum([utils.count_tokens(message.content, self.encoder) for message in chat_messages])