Initial commit

2026-05-12 09:41:56 +08:00
commit 572283e101
936 changed files with 133949 additions and 0 deletions
@@ -0,0 +1,33 @@
+import os
+# os.environ['HF_ENDPOINT']='https://hf-mirror.com'
+from dotenv import load_dotenv
+from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, Settings 
+from llama_index.llms.openai_like import OpenAILike
+from llama_index.embeddings.huggingface import HuggingFaceEmbedding
+
+load_dotenv()
+
+# 使用 AIHubmix
+Settings.llm = OpenAILike(
+    model="glm-4.7-flash-free",
+    api_key=os.getenv("DEEPSEEK_API_KEY"),
+    api_base="https://aihubmix.com/v1",
+    is_chat_model=True
+)
+
+# Settings.llm = OpenAI(
+#     model="deepseek-chat",
+#     api_key=os.getenv("DEEPSEEK_API_KEY"),
+#     api_base="https://api.deepseek.com"
+# )
+Settings.embed_model = HuggingFaceEmbedding("BAAI/bge-small-zh-v1.5")
+
+docs = SimpleDirectoryReader(input_files=["../../data/C1/markdown/easy-rl-chapter1.md"]).load_data()
+
+index = VectorStoreIndex.from_documents(docs)
+
+query_engine = index.as_query_engine()
+
+print(query_engine.get_prompts())
+
+print(query_engine.query("文中举了哪些例子?"))