feat: rag

13transformers.py

@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+# 说明：验证 Sentence Transformers 是否安装成功
+
+# 说明：导入 SentenceTransformer 类
+from sentence_transformers import SentenceTransformer
+
+# 说明：尝试加载一个轻量级模型进行测试
+# "all-MiniLM-L6-v2" 是一个小型的通用模型，适合快速测试
+# 首次运行时会自动下载模型（可能需要一些时间）
+print("正在加载模型进行测试...")
+model = SentenceTransformer("all-MiniLM-L6-v2")
+
+# 说明：对一个简单的句子进行编码测试
+sentence = "这是一个测试句子"
+embedding = model.encode(sentence)
+
+# 说明：检查嵌入向量的形状
+print(f"安装成功！嵌入向量维度：{embedding.shape}")
+print(f"前 5 个维度值：{embedding[:5]}")
+
+# 说明：如果没有报错并输出了维度信息，说明安装成功