'commit'

dsLightRag/Util/LightRagUtil.py

@@ -1,11 +1,12 @@
 import logging
 import logging.config
 import os
+
 import numpy as np
 from lightrag import LightRAG
 from lightrag.kg.shared_storage import initialize_pipeline_status
 from lightrag.llm.openai import openai_complete_if_cache, openai_embed
-from lightrag.rerank import jina_rerank, custom_rerank
+from lightrag.rerank import custom_rerank
 from lightrag.utils import EmbeddingFunc, logger, set_verbose_debug
 
 import Config.Config
@@ -144,8 +145,8 @@ def create_llm_model_func():
 
 def create_embedding_func():
     return EmbeddingFunc(
-        embedding_dim=1024,
-        max_token_size=8192,
+        embedding_dim=EMBED_DIM,
+        max_token_size=EMBED_MAX_TOKEN_SIZE,
         func=lambda texts: openai_embed(
             texts,
             model=EMBED_MODEL_NAME,