alihan/HiRAG
1
0
Fork 0
mirror of https://github.com/hhy-huang/HiRAG.git synced 2025-09-16 23:52:00 +03:00
Code Issues Packages Projects Releases Wiki Activity

[fix]:bug, cannot async for some embedding function

Browse Source
This commit is contained in:
haoyuhuang
2025-08-17 16:45:43 +08:00
parent d5c039d1a3
commit ca5ff7a2f9
6 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
eval/eval_mhqa.py
View File

@@ -68,7 +68,7 @@ def wrap_embedding_func_with_attrs(**kwargs):
@wrap_embedding_func_with_attrs(embedding_dim=2048, max_token_size=8192)
async def GLM_embedding(texts: list[str]) -> np.ndarray:
model_name = GLM_MODEL # "embedding-3"
client = OpenAI(
client = AsyncOpenAI(
api_key=GLM_API_KEY,
base_url=GLM_URL
)
@@ -82,7 +82,7 @@ async def GLM_embedding(texts: list[str]) -> np.ndarray:
@wrap_embedding_func_with_attrs(embedding_dim=4096, max_token_size=8192)
async def NV_embedding(texts: list[str]) -> np.ndarray:
model_name = NVIDIA_MODEL # "nvidia/nv-embed-v1"
client = OpenAI(
client = AsyncOpenAI(
api_key=NVIDIA_API_KEY,
base_url=NVIDIA_URL,
)

2
eval/test_deepseek.py
View File

@@ -59,7 +59,7 @@ def wrap_embedding_func_with_attrs(**kwargs):
@wrap_embedding_func_with_attrs(embedding_dim=2048, max_token_size=8192)
async def GLM_embedding(texts: list[str]) -> np.ndarray:
model_name = "embedding-3"
client = OpenAI(
client = AsyncOpenAI(
api_key=GLM_API_KEY,
base_url="https://open.bigmodel.cn/api/paas/v4/"
)

2
eval/test_glm.py
View File

@@ -50,7 +50,7 @@ def wrap_embedding_func_with_attrs(**kwargs):
@wrap_embedding_func_with_attrs(embedding_dim=2048, max_token_size=8192)
async def GLM_embedding(texts: list[str]) -> np.ndarray:
model_name = "embedding-3"
client = OpenAI(
client = AsyncOpenAI(
api_key=GLM_API_KEY,
base_url="https://open.bigmodel.cn/api/paas/v4/"
)

2
hi_Search_deepseek.py
View File

@@ -42,7 +42,7 @@ def wrap_embedding_func_with_attrs(**kwargs):
@wrap_embedding_func_with_attrs(embedding_dim=config['model_params']['glm_embedding_dim'], max_token_size=config['model_params']['max_token_size'])
async def GLM_embedding(texts: list[str]) -> np.ndarray:
model_name = "embedding-3"
client = OpenAI(
client = AsyncOpenAI(
api_key=GLM_API_KEY,
base_url=GLM_URL
)

2
hi_Search_glm.py
View File

@@ -39,7 +39,7 @@ def wrap_embedding_func_with_attrs(**kwargs):
@wrap_embedding_func_with_attrs(embedding_dim=config['model_params']['glm_embedding_dim'], max_token_size=config['model_params']['max_token_size'])
async def GLM_embedding(texts: list[str]) -> np.ndarray:
model_name = config['glm']['embedding_model']
client = OpenAI(
client = AsyncOpenAI(
api_key=GLM_API_KEY,
base_url=GLM_URL
)

2
hi_Search_openai.py
View File

@@ -50,7 +50,7 @@ async def OPENAI_embedding(texts: list[str]) -> np.ndarray:
@wrap_embedding_func_with_attrs(embedding_dim=config['model_params']['glm_embedding_dim'], max_token_size=config['model_params']['max_token_size'])
async def GLM_embedding(texts: list[str]) -> np.ndarray:
model_name = config['glm']['embedding_model']
client = OpenAI(
client = AsyncOpenAI(
api_key=GLM_API_KEY,
base_url=GLM_URL
)