From 081c1b45c70bc43485374db5f2b4764d3e0e514d Mon Sep 17 00:00:00 2001 From: YangQianli92 <108046369+YangQianli92@users.noreply.github.com> Date: Tue, 16 Apr 2024 11:45:30 +0800 Subject: [PATCH] Update rag_bm.py --- examples/rag_bm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/rag_bm.py b/examples/rag_bm.py index 2b954b385..a67eddc7c 100644 --- a/examples/rag_bm.py +++ b/examples/rag_bm.py @@ -8,7 +8,7 @@ from llama_index.core.schema import NodeWithScore from metagpt.const import DATA_PATH, EXAMPLE_BENCHMARK_PATH, EXAMPLE_DATA_PATH from metagpt.logs import logger -from metagpt.rag.benchmark import RAGBenchMark +from metagpt.rag.benchmark import RAGBenchmark from metagpt.rag.engines import SimpleEngine from metagpt.rag.factories import get_rag_embedding, get_rag_llm from metagpt.rag.schema import ( @@ -40,7 +40,7 @@ class RAGExample: """Show how to use RAG for evaluation.""" def __init__(self): - self.benchmark = RAGBenchMark() + self.benchmark = RAGBenchmark() self.embedding = get_rag_embedding() self.llm = get_rag_llm()