refactor(memory): Rename store folder to chat_message_history (#126)

davidmigloz · Aug 18, 2023 · fa54c7e · fa54c7e
1 parent f87a738
commit fa54c7e
Show file tree

Hide file tree

Showing 10 changed files with 9 additions and 13 deletions.
diff --git a/packages/langchain/lib/src/memory/buffer_window.dart b/packages/langchain/lib/src/memory/buffer_window.dart
@@ -3,7 +3,6 @@ import '../model_io/chat_models/utils.dart';
 import 'base.dart';
 import 'chat.dart';
 import 'models/models.dart';
-import 'stores/message/in_memory.dart';
 
 /// {@template conversation_buffer_window_memory}
 /// Buffer for storing a conversation in-memory inside a limited size window

diff --git a/packages/langchain/lib/src/memory/chat.dart b/packages/langchain/lib/src/memory/chat.dart
@@ -3,9 +3,8 @@ import 'dart:async';
 import '../model_io/chat_models/models/models.dart';
 import '../utils/exception.dart';
 import 'base.dart';
+import 'chat_message_history/chat_message_history.dart';
 import 'models/models.dart';
-import 'stores/message/history.dart';
-import 'stores/message/in_memory.dart';
 import 'utils.dart';
 
 /// {@template base_chat_memory}

diff --git a/...ib/src/memory/stores/message/history.dart → ...src/memory/chat_message_history/base.dart b/...ib/src/memory/stores/message/history.dart → ...src/memory/chat_message_history/base.dart
@@ -1,4 +1,4 @@
-import '../../../model_io/chat_models/models/models.dart';
+import '../../model_io/chat_models/models/models.dart';
 
 /// {@template base_chat_message_history}
 /// Base interface for chat message history.

diff --git a/packages/langchain/lib/src/memory/chat_message_history/chat_message_history.dart b/packages/langchain/lib/src/memory/chat_message_history/chat_message_history.dart
@@ -0,0 +1,2 @@
+export 'base.dart';
+export 'in_memory.dart';
diff --git a/.../src/memory/stores/message/in_memory.dart → ...emory/chat_message_history/in_memory.dart b/.../src/memory/stores/message/in_memory.dart → ...emory/chat_message_history/in_memory.dart
@@ -1,7 +1,7 @@
 import 'dart:collection';
 
-import '../../../model_io/chat_models/chat_models.dart';
-import 'history.dart';
+import '../../model_io/chat_models/chat_models.dart';
+import 'base.dart';
 
 /// {@template chat_message_history}
 /// A simple in-memory implementation of a chat message history.

diff --git a/packages/langchain/lib/src/memory/memory.dart b/packages/langchain/lib/src/memory/memory.dart
@@ -2,7 +2,7 @@ export 'base.dart';
 export 'buffer.dart';
 export 'buffer_window.dart';
 export 'chat.dart';
+export 'chat_message_history/chat_message_history.dart';
 export 'models/models.dart';
 export 'simple.dart';
-export 'stores/stores.dart';
 export 'vector_store.dart';
diff --git a/packages/langchain/lib/src/memory/stores/stores.dart b/packages/langchain/lib/src/memory/stores/stores.dart
diff --git a/packages/langchain/lib/src/memory/summary.dart b/packages/langchain/lib/src/memory/summary.dart
@@ -5,9 +5,8 @@ import '../model_io/language_models/language_models.dart';
 import '../model_io/prompts/prompts.dart';
 import 'base.dart';
 import 'chat.dart';
+import 'chat_message_history/chat_message_history.dart';
 import 'models/models.dart';
-import 'stores/message/history.dart';
-import 'stores/message/in_memory.dart';
 
 const _template = '''
 Progressively summarize the lines of conversation provided, adding onto the previous summary returning a new summary.

diff --git a/packages/langchain/lib/src/memory/token_buffer.dart b/packages/langchain/lib/src/memory/token_buffer.dart
@@ -6,7 +6,6 @@ import 'base.dart';
 import 'buffer_window.dart';
 import 'chat.dart';
 import 'models/models.dart';
-import 'stores/message/in_memory.dart';
 
 /// {@template conversation_token_buffer_memory}
 /// Rolling buffer for storing a conversation and then retrieving the messages

diff --git a/...memory/stores/message/in_memory_test.dart → .../chat_message_history/in_memory_test.dart b/...memory/stores/message/in_memory_test.dart → .../chat_message_history/in_memory_test.dart
@@ -1,5 +1,4 @@
-import 'package:langchain/src/memory/stores/stores.dart';
-import 'package:langchain/src/model_io/chat_models/chat_models.dart';
+import 'package:langchain/langchain.dart';
 import 'package:test/test.dart';
 
 void main() {