Skip to content

Commit 3a078c3

Browse files
fix: fix embedding config load (#309)
* fix: fix embedding config load * Update examples/rag/e2e_RAG_eval_with_mockRAG_fiqa.py Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> --------- Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
1 parent 3e11143 commit 3a078c3

File tree

2 files changed

+10
-0
lines changed

2 files changed

+10
-0
lines changed

dingo/model/llm/base_openai.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,15 @@ def create_client(cls):
4040

4141
# 如果配置了 embedding_config,初始化 Embedding 客户端
4242
if cls.dynamic_config.embedding_config:
43+
from dingo.config.input_args import EmbeddingConfigArgs
44+
4345
embedding_cfg = cls.dynamic_config.embedding_config
46+
47+
# 处理 embedding_config 可能是字典或对象的情况
48+
if isinstance(embedding_cfg, dict):
49+
# 如果是字典,转换为 EmbeddingConfigArgs 对象
50+
embedding_cfg = EmbeddingConfigArgs(**embedding_cfg)
51+
4452
if not embedding_cfg.api_url:
4553
raise ValueError("embedding_config must provide api_url")
4654

examples/rag/e2e_RAG_eval_with_mockRAG_fiqa.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -406,6 +406,8 @@ async def main():
406406
print("FiQA 端到端 RAG 系统评测")
407407
print("=" * 80)
408408
print(f"数据集: {FIQA_DATASET} (从 HuggingFace 自动下载)")
409+
print(f"API Key: {('sk-...' + OPENAI_API_KEY[-4:]) if OPENAI_API_KEY else 'Not set'}")
410+
print(f"API Base URL: {OPENAI_BASE_URL}")
409411
print(f"模型: {OPENAI_MODEL}")
410412
print(f"Top-K: {args.top_k}")
411413
print("=" * 80)

0 commit comments

Comments
 (0)