post migrate cleanup

2024-03-18 23:10:20 +07:00
parent df12dec732
commit a8f92b3f9e
2 changed files with 5 additions and 10 deletions
--- a/libs/kotaemon/tests/test_llms_chat_models.py
+++ b/libs/kotaemon/tests/test_llms_chat_models.py
@@ -3,12 +3,7 @@ from unittest.mock import patch

 import pytest

-from kotaemon.base.schema import (
-    AIMessage,
-    HumanMessage,
-    LLMInterface,
-    SystemMessage,
-)
+from kotaemon.base.schema import AIMessage, HumanMessage, LLMInterface, SystemMessage
 from kotaemon.llms import AzureChatOpenAI, LlamaCppChat

 try: