Merge branch 'main' of https://hf-mirror.com/spaces/sanpang/llamaindex_demo into main

app.py

@@ -1,39 +1,38 @@
+import os
 import streamlit as st
 from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, Settings
 from llama_index.embeddings.huggingface import HuggingFaceEmbedding
 from llama_index.legacy.callbacks import CallbackManager
 from llama_index.llms.openai_like import OpenAILike
 
-# Create an instance of CallbackManager
 callback_manager = CallbackManager()
 
 api_base_url =  "https://internlm-chat.intern-ai.org.cn/puyu/api/v1/"
 model = "internlm2.5-latest"
-api_key = ""
 
-# api_base_url =  "https://api.siliconflow.cn/v1"
-# model = "internlm/internlm2_5-7b-chat"
-# api_key = "请填写 API Key"
+# 通过Spaces的secret传入
+api_key = os.environ.get('API_KEY')
 
 llm =OpenAILike(model=model, api_base=api_base_url, api_key=api_key, is_chat_model=True,callback_manager=callback_manager)
 
-
-
 st.set_page_config(page_title="llama_index_demo", page_icon="🦜🔗")
-st.title("llama_index_demo")
+st.title("llamaindex_demo")
+
+os.system('git lfs install')
+os.system('git clone https://www.modelscope.cn/Ceceliachenen/paraphrase-multilingual-MiniLM-L12-v2.git')
 
 # 初始化模型
 @st.cache_resource
 def init_models():
     embed_model = HuggingFaceEmbedding(
-        model_name="/root/model/sentence-transformer"
+        model_name="./paraphrase-multilingual-MiniLM-L12-v2"
     )
     Settings.embed_model = embed_model
 
     #用初始化llm
     Settings.llm = llm
 
-    documents = SimpleDirectoryReader("/root/llamaindex_demo/data").load_data()
+    documents = SimpleDirectoryReader("./data").load_data()
     index = VectorStoreIndex.from_documents(documents)
     query_engine = index.as_query_engine()
 

requirements.txt

@@ -0,0 +1,6 @@
+llama-index==0.11.20
+llama-index-llms-replicate==0.3.0
+llama-index-llms-openai-like==0.2.0
+llama-index-embeddings-huggingface==0.3.1
+llama-index-embeddings-instructor==0.2.1
+sentence-transformers==2.7.0