Merge pull request #133 from Integration-Automation/dev

Fix remove option that not in options list
Integration-Automation · Apr 12, 2024 · 437fb52 · 437fb52
2 parents 9f261d5 + 8a07bee
commit 437fb52
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/dev.toml b/dev.toml
@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "re_edge_gpt_dev"
-version = "0.0.42"
+version = "0.0.43"
 authors = [
     { name = "JE-Chen", email = "jechenmailman@gmail.com" },
 ]

diff --git a/pyproject.toml b/pyproject.toml
@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "re_edge_gpt"
-version = "0.0.35"
+version = "0.0.36"
 authors = [
     { name = "JE-Chen", email = "jechenmailman@gmail.com" },
 ]

diff --git a/re_edge_gpt/chat/chathub.py b/re_edge_gpt/chat/chathub.py
@@ -121,7 +121,7 @@ async def ask_stream(
         # Construct a ChatHub request
         if remove_options is not None:
             for option in remove_options:
-                if option in remove_options:
+                if option in conversation_style.value:
                     conversation_style.value.remove(option)
         if add_options is not None:
             for option in add_options:

diff --git a/test/unit_test/back-end/manual_test/test_bot_manual.py b/test/unit_test/back-end/manual_test/test_bot_manual.py
@@ -21,7 +21,7 @@ async def test_ask() -> None:
         bot = await Chatbot.create(cookies=cookies, mode=mode)
         response = await bot.ask(
             prompt="What version u are using GPT-4 turbo?",
-            conversation_style=ConversationStyle.creative_classic,
+            conversation_style=ConversationStyle.creative_classic, # ConversationStyle.creative
             simplify_response=True,
             search_result=True,
         )