Browse Source

[Feature] Add support for optionally fetch all chat history for app (#1249)

Deshraj Yadav 1 năm trước cách đây
mục cha
commit
7ad4af250f
2 tập tin đã thay đổi với 13 bổ sung3 xóa
  1. 12 2
      embedchain/embedchain.py
  2. 1 1
      pyproject.toml

+ 12 - 2
embedchain/embedchain.py

@@ -696,9 +696,19 @@ class EmbedChain(JSONSerializable):
         # Send anonymous telemetry
         self.telemetry.capture(event_name="reset", properties=self._telemetry_props)
 
-    def get_history(self, num_rounds: int = 10, display_format: bool = True, session_id: Optional[str] = "default"):
+    def get_history(
+        self,
+        num_rounds: int = 10,
+        display_format: bool = True,
+        session_id: Optional[str] = "default",
+        fetch_all: bool = False,
+    ):
         history = self.llm.memory.get(
-            app_id=self.config.id, session_id=session_id, num_rounds=num_rounds, display_format=display_format
+            app_id=self.config.id,
+            session_id=session_id,
+            num_rounds=num_rounds,
+            display_format=display_format,
+            fetch_all=fetch_all,
         )
         return history
 

+ 1 - 1
pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "embedchain"
-version = "0.1.75"
+version = "0.1.76"
 description = "Simplest open source retrieval(RAG) framework"
 authors = [
     "Taranjeet Singh <taranjeet@embedchain.ai>",