fix: system message

LyzrCore · Apr 28, 2024 · a3325f5 · a3325f5
1 parent f849c59
commit a3325f5
Show file tree

Hide file tree

Showing 7 changed files with 38 additions and 26 deletions.
diff --git a/build/lib/lyzr/utils/chat_utils.py b/build/lib/lyzr/utils/chat_utils.py
@@ -88,11 +88,12 @@ def pdf_chat_(
 
     memory = ChatMemoryBuffer.from_defaults(token_limit=4000)
 
-    chat_engine = ContextChatEngine(
+    chat_engine = ContextChatEngine.from_defaults(
         llm=llm,
         memory=memory,
+        system_prompt=system_prompt,
+        service_context=service_context,
         retriever=retriever,
-        prefix_messages=list(),
         **chat_engine_params,
     )
 
@@ -167,11 +168,12 @@ def txt_chat_(
 
     memory = ChatMemoryBuffer.from_defaults(token_limit=4000)
 
-    chat_engine = ContextChatEngine(
+    chat_engine = ContextChatEngine.from_defaults(
         llm=llm,
         memory=memory,
+        system_prompt=system_prompt,
+        service_context=service_context,
         retriever=retriever,
-        prefix_messages=list(),
         **chat_engine_params,
     )
 
@@ -246,11 +248,12 @@ def docx_chat_(
 
     memory = ChatMemoryBuffer.from_defaults(token_limit=4000)
 
-    chat_engine = ContextChatEngine(
+    chat_engine = ContextChatEngine.from_defaults(
         llm=llm,
         memory=memory,
+        system_prompt=system_prompt,
+        service_context=service_context,
         retriever=retriever,
-        prefix_messages=list(),
         **chat_engine_params,
     )
 
@@ -315,11 +318,12 @@ def webpage_chat_(
 
     memory = ChatMemoryBuffer.from_defaults(token_limit=4000)
 
-    chat_engine = ContextChatEngine(
+    chat_engine = ContextChatEngine.from_defaults(
         llm=llm,
         memory=memory,
+        system_prompt=system_prompt,
+        service_context=service_context,
         retriever=retriever,
-        prefix_messages=list(),
         **chat_engine_params,
     )
 
@@ -384,11 +388,12 @@ def website_chat_(
 
     memory = ChatMemoryBuffer.from_defaults(token_limit=4000)
 
-    chat_engine = ContextChatEngine(
+    chat_engine = ContextChatEngine.from_defaults(
         llm=llm,
         memory=memory,
+        system_prompt=system_prompt,
+        service_context=service_context,
         retriever=retriever,
-        prefix_messages=list(),
         **chat_engine_params,
     )
 
@@ -453,11 +458,12 @@ def youtube_chat_(
 
     memory = ChatMemoryBuffer.from_defaults(token_limit=4000)
 
-    chat_engine = ContextChatEngine(
+    chat_engine = ContextChatEngine.from_defaults(
         llm=llm,
         memory=memory,
+        system_prompt=system_prompt,
+        service_context=service_context,
         retriever=retriever,
-        prefix_messages=list(),
         **chat_engine_params,
     )
 

diff --git a/dist/lyzr-0.1.37.tar.gz b/dist/lyzr-0.1.37.tar.gz
diff --git a/dist/lyzr-0.1.37-py3-none-any.whl → dist/lyzr-0.1.38-py3-none-any.whl b/dist/lyzr-0.1.37-py3-none-any.whl → dist/lyzr-0.1.38-py3-none-any.whl
diff --git a/dist/lyzr-0.1.38.tar.gz b/dist/lyzr-0.1.38.tar.gz
diff --git a/lyzr.egg-info/PKG-INFO b/lyzr.egg-info/PKG-INFO
@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lyzr
-Version: 0.1.37
+Version: 0.1.38
 Summary: UNKNOWN
 Home-page: UNKNOWN
 Author: lyzr

diff --git a/lyzr/utils/chat_utils.py b/lyzr/utils/chat_utils.py
@@ -88,11 +88,12 @@ def pdf_chat_(
 
     memory = ChatMemoryBuffer.from_defaults(token_limit=4000)
 
-    chat_engine = ContextChatEngine(
+    chat_engine = ContextChatEngine.from_defaults(
         llm=llm,
         memory=memory,
+        system_prompt=system_prompt,
+        service_context=service_context,
         retriever=retriever,
-        prefix_messages=list(),
         **chat_engine_params,
     )
 
@@ -167,11 +168,12 @@ def txt_chat_(
 
     memory = ChatMemoryBuffer.from_defaults(token_limit=4000)
 
-    chat_engine = ContextChatEngine(
+    chat_engine = ContextChatEngine.from_defaults(
         llm=llm,
         memory=memory,
+        system_prompt=system_prompt,
+        service_context=service_context,
         retriever=retriever,
-        prefix_messages=list(),
         **chat_engine_params,
     )
 
@@ -246,11 +248,12 @@ def docx_chat_(
 
     memory = ChatMemoryBuffer.from_defaults(token_limit=4000)
 
-    chat_engine = ContextChatEngine(
+    chat_engine = ContextChatEngine.from_defaults(
         llm=llm,
         memory=memory,
+        system_prompt=system_prompt,
+        service_context=service_context,
         retriever=retriever,
-        prefix_messages=list(),
         **chat_engine_params,
     )
 
@@ -315,11 +318,12 @@ def webpage_chat_(
 
     memory = ChatMemoryBuffer.from_defaults(token_limit=4000)
 
-    chat_engine = ContextChatEngine(
+    chat_engine = ContextChatEngine.from_defaults(
         llm=llm,
         memory=memory,
+        system_prompt=system_prompt,
+        service_context=service_context,
         retriever=retriever,
-        prefix_messages=list(),
         **chat_engine_params,
     )
 
@@ -384,11 +388,12 @@ def website_chat_(
 
     memory = ChatMemoryBuffer.from_defaults(token_limit=4000)
 
-    chat_engine = ContextChatEngine(
+    chat_engine = ContextChatEngine.from_defaults(
         llm=llm,
         memory=memory,
+        system_prompt=system_prompt,
+        service_context=service_context,
         retriever=retriever,
-        prefix_messages=list(),
         **chat_engine_params,
     )
 
@@ -453,11 +458,12 @@ def youtube_chat_(
 
     memory = ChatMemoryBuffer.from_defaults(token_limit=4000)
 
-    chat_engine = ContextChatEngine(
+    chat_engine = ContextChatEngine.from_defaults(
         llm=llm,
         memory=memory,
+        system_prompt=system_prompt,
+        service_context=service_context,
         retriever=retriever,
-        prefix_messages=list(),
         **chat_engine_params,
     )
 

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name="lyzr",
-    version="0.1.37",
+    version="0.1.38",
     author="lyzr",
     description="",
     long_description=open("README.md").read(),