feat: add dimension check for HNSW index creation in common.py

Author: liuruibin

apps/knowledge/serializers/common.py

@@ -256,9 +256,11 @@ def create_knowledge_index(knowledge_id=None, document_id=None):
         if len(result) == 0:
             return
         dims = result[0]['dims']
-        sql = f"""CREATE INDEX "embedding_hnsw_idx_{k_id}" ON embedding USING hnsw ((embedding::vector({dims})) vector_cosine_ops) WHERE knowledge_id = '{k_id}'"""
-        update_execute(sql, [])
-        maxkb_logger.info(f'Created index for knowledge ID: {k_id}')
+        # 超过2000维度不创建索引，pgvector hnsw索引不支持超过2000维度
+        if dims < 2000:
+            sql = f"""CREATE INDEX "embedding_hnsw_idx_{k_id}" ON embedding USING hnsw ((embedding::vector({dims})) vector_cosine_ops) WHERE knowledge_id = '{k_id}'"""
+            update_execute(sql, [])
+            maxkb_logger.info(f'Created index for knowledge ID: {k_id}')
 
 
 def drop_knowledge_index(knowledge_id=None, document_id=None):