From 5ce2484af086d763962dd369c9fd03f87abe852f Mon Sep 17 00:00:00 2001
From: imClumsyPanda <littlepanda0716@gmail.com>
Date: Tue, 1 Aug 2023 15:08:19 +0800
Subject: [PATCH] update webui.py

---
 webui.py                                     | 2 +-
 webui_pages/dialogue/dialogue.py             | 8 +++++---
 webui_pages/knowledge_base/knowledge_base.py | 2 +-
 webui_pages/model_config/model_config.py     | 2 +-
 4 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/webui.py b/webui.py
index c06646e..bd02735 100644
--- a/webui.py
+++ b/webui.py
@@ -26,4 +26,4 @@ if __name__ == "__main__":
                                     menu_icon="chat-quote",
                                     default_index=0)
 
-    pages[selected_page]["func"]()
+    pages[selected_page]["func"](api)
diff --git a/webui_pages/dialogue/dialogue.py b/webui_pages/dialogue/dialogue.py
index 08b348c..f01c0ac 100644
--- a/webui_pages/dialogue/dialogue.py
+++ b/webui_pages/dialogue/dialogue.py
@@ -4,7 +4,8 @@ from streamlit_chatbox import *
 
 chat_box = ChatBox()
 
-def dialogue_page():
+
+def dialogue_page(api: ApiRequest):
     with st.sidebar:
         dialogue_mode = st.radio("请选择对话模式",
                                  ["LLM 对话",
@@ -22,7 +23,8 @@ def dialogue_page():
     if prompt := st.chat_input("请输入对话内容，换行请使用Ctrl+Enter"):
         chat_box.user_say(prompt)
         chat_box.ai_say("正在思考...")
-        # with api.chat_fastchat([{"role": "user", "content": "prompt"}], stream=streaming) as r: # todo: support history len
+        # with api.chat_fastchat([{"role": "user", "content": "prompt"}], stream=streaming) as r:
+        # todo: support history len
         text = ""
         r = api.chat_chat(prompt, no_remote_api=True)
         for t in r:
@@ -33,4 +35,4 @@ def dialogue_page():
         #     for t in r.iter_text(None):
         #         text += t
         #         chat_box.update_msg(text)
-        #     chat_box.update_msg(text, streaming=False)
\ No newline at end of file
+        #     chat_box.update_msg(text, streaming=False)
diff --git a/webui_pages/knowledge_base/knowledge_base.py b/webui_pages/knowledge_base/knowledge_base.py
index 8515b62..119a4a2 100644
--- a/webui_pages/knowledge_base/knowledge_base.py
+++ b/webui_pages/knowledge_base/knowledge_base.py
@@ -1,6 +1,6 @@
 import streamlit as st
 from webui_pages.utils import *
 
-def knowledge_base_page():
+def knowledge_base_page(api: ApiRequest):
     st.write(123)
     pass
\ No newline at end of file
diff --git a/webui_pages/model_config/model_config.py b/webui_pages/model_config/model_config.py
index cc19d93..119a6ca 100644
--- a/webui_pages/model_config/model_config.py
+++ b/webui_pages/model_config/model_config.py
@@ -1,5 +1,5 @@
 import streamlit as st
 from webui_pages.utils import *
 
-def model_config_page():
+def model_config_page(api: ApiRequest):
     pass
\ No newline at end of file