Update

esbmc · Sep 12, 2024 · 7c0195d · 7c0195d
1 parent 00346e7
commit 7c0195d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/esbmc_ai/chats/__init__.py b/esbmc_ai/chats/__init__.py
@@ -1,5 +1,8 @@
 # Author: Yiannis Charalambous
 
+"""This module contains different chat interfaces. Along with `BaseChatInterface`
+that provides necessary boilet-plate for implementing an LLM based chat."""
+
 from .base_chat_interface import BaseChatInterface
 from .latest_state_solution_generator import LatestStateSolutionGenerator
 from .solution_generator import SolutionGenerator

diff --git a/esbmc_ai/chats/user_chat.py b/esbmc_ai/chats/user_chat.py
@@ -1,12 +1,12 @@
 # Author: Yiannis Charalambous 2023
 
-from langchain_core.language_models import BaseChatModel
 from typing_extensions import override
 
 from langchain.memory import ConversationSummaryMemory
+from langchain.schema import BaseMessage, SystemMessage
+from langchain_core.language_models import BaseChatModel
 from langchain_community.chat_message_histories import ChatMessageHistory
 
-from langchain.schema import BaseMessage, SystemMessage
 
 from esbmc_ai.config import AIAgentConversation, ChatPromptSettings
 from esbmc_ai.ai_models import AIModel