update chatgpt_long_term_memory

ElmiraGhorbani · Jul 25, 2023 · 74b2456 · 74b2456
1 parent 1505bf6
commit 74b2456
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/chatgpt_long_term_memory/__init__.py b/chatgpt_long_term_memory/__init__.py
@@ -1,11 +1,16 @@
-from chatgpt_long_term_memory.openai_engine import (OpenAIChatBot,
-                                                  OpenAIChatConfig,
-                                                  retry_on_openai_errors)
+from chatgpt_long_term_memory.conversation.chatgpt_chatbot_client import \
+    ChatbotClient
+from chatgpt_long_term_memory.conversation.chatgpt_index_client import \
+    ChatGPTClient
 from chatgpt_long_term_memory.llama_index_helpers import DocIndexer, Retrievers
 from chatgpt_long_term_memory.llama_index_helpers.config import (
     IndexConfig, RetrieversConfig)
 from chatgpt_long_term_memory.memory import ChatMemory
 from chatgpt_long_term_memory.memory.chat_memory import ChatMemoryConfig
+from chatgpt_long_term_memory.openai_engine import (OpenAIChatBot,
+                                                    OpenAIChatConfig,
+                                                    TokenCounterConfig,
+                                                    retry_on_openai_errors)
 
 __all__ = [
     "DocIndexer",
@@ -16,5 +21,8 @@
     "ChatMemoryConfig",
     "OpenAIChatConfig",
     "OpenAIChatBot",
-    "retry_on_openai_errors"
+    "retry_on_openai_errors",
+    "TokenCounterConfig",
+    "ChatGPTClient",
+    "ChatbotClient"
 ]