更新依赖版本

ChuanhuChatbot.py

@@ -287,6 +287,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
             chatbot,
             use_streaming_checkbox,
             use_websearch_checkbox,
+            autogpt_mode,
             index_files,
             language_select_dropdown,
         ],
@@ -349,6 +350,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
             chatbot,
             use_streaming_checkbox,
             use_websearch_checkbox,
+            autogpt_mode,
             index_files,
             language_select_dropdown,
         ],

modules/llama_func.py

@@ -107,7 +107,7 @@ def construct_index(
 ):
     from langchain.chat_models import ChatOpenAI
     from langchain.embeddings.huggingface import HuggingFaceEmbeddings
-    from llama_index import GPTSimpleVectorIndex, ServiceContext, LangchainEmbedding, OpenAIEmbedding
+    from llama_index import GPTVectorStoreIndex, ServiceContext, LangchainEmbedding, OpenAIEmbedding
 
     if api_key:
         os.environ["OPENAI_API_KEY"] = api_key
@@ -129,7 +129,7 @@ def construct_index(
     index_name = get_index_name(file_src)
     if os.path.exists(f"./index/{index_name}.json"):
         logging.info("找到了缓存的索引文件，加载中……")
-        return GPTSimpleVectorIndex.load_from_disk(f"./index/{index_name}.json")
+        return GPTVectorStoreIndex.load_from_disk(f"./index/{index_name}.json")
     else:
         try:
             documents = get_documents(file_src)
@@ -144,12 +144,12 @@ def construct_index(
                     chunk_size_limit=chunk_size_limit,
                     embed_model=embed_model,
                 )
-                index = GPTSimpleVectorIndex.from_documents(
+                index = GPTVectorStoreIndex.from_documents(
                     documents, service_context=service_context
                 )
             logging.debug("索引构建完成！")
             os.makedirs("./index", exist_ok=True)
-            index.save_to_disk(f"./index/{index_name}.json")
+            index.storage_context.persist(f"./index/{index_name}")
             logging.debug("索引已保存至本地!")
             return index
 

modules/models/base_model.py

@@ -277,6 +277,7 @@ class BaseLLMModel:
         chatbot,
         stream=False,
         use_websearch=False,
+        autogpt_mode=False,
         files=None,
         reply_language="中文",
         should_check_token_count=True,
@@ -383,6 +384,7 @@ class BaseLLMModel:
         chatbot,
         stream=False,
         use_websearch=False,
+        autogpt_mode=False,
         files=None,
         reply_language="中文",
     ):
@@ -402,6 +404,7 @@ class BaseLLMModel:
             chatbot,
             stream=stream,
             use_websearch=use_websearch,
+            autogpt_mode=autogpt_mode,
             files=files,
             reply_language=reply_language,
         )

requirements.txt

@@ -1,4 +1,4 @@
-gradio==3.28.0
+gradio==3.30.0
 gradio_client==0.1.4
 mdtex2html
 pypinyin
@@ -8,8 +8,8 @@ tqdm
 colorama
 duckduckgo_search==2.9.5
 Pygments
-llama_index==0.5.25
-langchain<0.0.150
+llama_index==0.6.8
+langchain==0.0.170
 markdown
 PyPDF2
 pdfplumber