From 21e496fb0aada689dfc0bdd57e0d70ea4949e3b4 Mon Sep 17 00:00:00 2001
From: Wendong-Fan <133094783+Wendong-Fan@users.noreply.github.com>
Date: Mon, 13 Jan 2025 07:13:58 +0800
Subject: [PATCH] fix: camel app AttributeError (#1439)

---
 apps/agents/agents.py           | 49 +++++++++++++++++----------------
 apps/agents/test/test_agents.py |  1 -
 2 files changed, 26 insertions(+), 24 deletions(-)

diff --git a/apps/agents/agents.py b/apps/agents/agents.py
index eda55369ed..17b1ca0a44 100644
--- a/apps/agents/agents.py
+++ b/apps/agents/agents.py
@@ -563,25 +563,33 @@ def change_society(society_name: str) -> Tuple[Dict, Dict, str]:
         lambda v: gr.update(visible=v), task_specifier_cb, ts_word_limit_nb
     )
 
+    def start_chain():
+        state = cleanup_on_launch(session_state.value)
+        state, specified_task, task_prompt, chat, progress = (
+            role_playing_start(
+                state,
+                society_dd.value,
+                assistant_ta.value,
+                user_ta.value,
+                original_task_ta.value,
+                num_messages_sl.value,
+                task_specifier_cb.value,
+                ts_word_limit_nb.value,
+                language_ta.value,
+            )
+        )
+        state, chat, progress = role_playing_chat_init(state)
+        return (
+            state,
+            specified_task,
+            task_prompt,
+            chat,
+            progress,
+        )
+
     start_bn.click(
-        cleanup_on_launch,
-        session_state,
-        [session_state, chatbot, start_bn],
-        queue=False,
-    ).then(
-        role_playing_start,
-        [
-            session_state,
-            society_dd,
-            assistant_ta,
-            user_ta,
-            original_task_ta,
-            num_messages_sl,
-            task_specifier_cb,
-            ts_word_limit_nb,
-            language_ta,
-        ],
-        [
+        fn=start_chain,
+        outputs=[
             session_state,
             specified_task_ta,
             task_prompt_ta,
@@ -589,11 +597,6 @@ def change_society(society_name: str) -> Tuple[Dict, Dict, str]:
             progress_sl,
         ],
         queue=False,
-    ).then(
-        role_playing_chat_init,
-        session_state,
-        [session_state, chatbot, progress_sl],
-        queue=False,
     )
 
     blocks.load(
diff --git a/apps/agents/test/test_agents.py b/apps/agents/test/test_agents.py
index 4abc4a9613..27068dd4ba 100644
--- a/apps/agents/test/test_agents.py
+++ b/apps/agents/test/test_agents.py
@@ -26,7 +26,6 @@
 )
 
 
-@pytest.mark.skip(reason="Skipping this test temporarily")
 def test_construct_blocks():
     blocks = construct_blocks(None)
     assert isinstance(blocks, gr.Blocks)