'commit'

dsSchoolBuddy/ElasticSearch/T1_RebuildMapping.py

@@ -0,0 +1,47 @@
+import warnings
+
+from elasticsearch import Elasticsearch
+
+from Config import Config
+
+# 抑制HTTPS相关警告
+warnings.filterwarnings('ignore', message='Connecting to .* using TLS with verify_certs=False is insecure')
+warnings.filterwarnings('ignore', message='Unverified HTTPS request is being made to host')
+
+# 初始化ES连接
+es = Elasticsearch(
+    hosts=Config.ES_CONFIG['hosts'],
+    basic_auth=Config.ES_CONFIG['basic_auth'],
+    verify_certs=False
+)
+
+# 定义mapping结构
+mapping = {
+    "mappings": {
+        "properties": {
+            "embedding": {
+                "type": "dense_vector",
+                "dims": 200,  # embedding维度为200
+                "index": True,
+                "similarity": "l2_norm"  # 使用L2距离
+            },
+            "user_input": {"type": "text"},
+            "tags": {
+                "type": "object",
+                "properties": {
+                    "tags": {"type": "keyword"},
+                    "full_content": {"type": "text"}
+                }
+            }
+        }
+    }
+}
+
+# 创建索引
+index_name = Config.ES_CONFIG['index_name']
+if es.indices.exists(index=index_name):
+    es.indices.delete(index=index_name)
+    print(f"删除已存在的索引 '{index_name}'")
+
+es.indices.create(index=index_name, body=mapping)
+print(f"索引 '{index_name}' 创建成功，mapping结构已设置。")