From f5024245a7dfe11ca2a8ffa62d58d4c531f80236 Mon Sep 17 00:00:00 2001
From: liunux4odoo <liunu@qq.com>
Date: Thu, 10 Aug 2023 13:33:40 +0800
Subject: [PATCH] make api.py and webui.py working offline

---
 server/api.py | 2 +-
 webui.py      | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/server/api.py b/server/api.py
index 900ba00..21c9c1f 100644
--- a/server/api.py
+++ b/server/api.py
@@ -6,7 +6,7 @@ sys.path.append(os.path.dirname(os.path.dirname(__file__)))
 from configs.model_config import NLTK_DATA_PATH, OPEN_CROSS_DOMAIN
 import argparse
 import uvicorn
-from fastapi import FastAPI
+from fastapi_offline import FastAPIOffline as FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from starlette.responses import RedirectResponse
 from server.chat import (chat, knowledge_base_chat, openai_chat,
diff --git a/webui.py b/webui.py
index 07e3900..c883b10 100644
--- a/webui.py
+++ b/webui.py
@@ -44,7 +44,7 @@ if __name__ == "__main__":
     pages = {"对话": {"icon": "chat",
                       "func": dialogue_page,
                       },
-             "知识库管理": {"icon": "database-fill-gear",
+             "知识库管理": {"icon": "hdd-stack",
                             "func": knowledge_base_page,
                             },
              "模型配置": {"icon": "gear",