Update app.py

app.py

@@ -28,8 +28,24 @@ with st.sidebar:
 
 def main() -> None:
     st.header("X-Pipe Wiki 机器人 💬")
-    robot_manager = XPipeRobotManagerFactory.get_or_create(XPipeRobotRevision.SIMPLE_OPENAI_VERSION_0)
-    robot = robot_manager.get_robot()
+    # robot_manager = XPipeRobotManagerFactory.get_or_create(XPipeRobotRevision.SIMPLE_OPENAI_VERSION_0)
+    # robot = robot_manager.get_robot()
+    from llama_index import StorageContext
+    store = StorageContext.from_defaults(persist_dir="./dataset")
+    from langchain_manager.manager import LangChainAzureManager
+    lc_manager = LangChainAzureManager()
+    from llama_index import ServiceContext
+    from llama_index import LangchainEmbedding
+    service = ServiceContext.from_defaults(llm=lc_manager.get_llm(), embed_model=LangchainEmbedding(lc_manager.get_embedding()))
+
+    from llama_index import load_index_from_storage
+    index = load_index_from_storage(store)
+    query_engine = index.as_query_engine(service_context=service)
+    robot = AzureOpenAIXPipeWikiRobot(query_engine)
+    query = st.text_input("X-Pipe Wiki 问题:")
+    if query:
+        response = robot.ask(question=query)
+        st.write(response)
     query = st.text_input("X-Pipe Wiki 问题:")
     if query:
         response = robot.ask(question=query)