update

client.py

@@ -0,0 +1,28 @@
+from gradio_client import Client
+
+client = Client("xu-song/self-chat")
+
+
+def self_chat_demo(system_message, num_turn=4):
+    # 1. set system message for the agent
+    client.predict(
+        system=system_message,
+        api_name="/reset_state"
+    )
+    messages = []
+    # 2. self chat
+    for num in range(num_turn):
+        # 注意：history 是 gr.state类型，不能通过API传参
+        messages = client.predict(
+            chatbot=messages,
+            api_name="/generate"
+        )
+        if num % 2 == 0:
+            assert messages[-1][1] is None
+            print(f"Q: {messages[-1][0]}")
+        else:
+            print(f"A: {messages[-1][1]}")
+
+
+if __name__ == "__main__":
+    self_chat_demo(system_message="你是一个小说家，擅长写武侠小说")