diff --git a/.gitignore b/.gitignore index 4c0d0f11c..922116d12 100644 --- a/.gitignore +++ b/.gitignore @@ -175,7 +175,6 @@ output tmp.png .dependencies.json tests/metagpt/utils/file_repo_git -tests/data/rsp_cache.json tests/data/rsp_cache_new.json *.tmp *.png diff --git a/metagpt/rag/rankers/base.py b/metagpt/rag/rankers/base.py index ecb23cf3e..a533a8b90 100644 --- a/metagpt/rag/rankers/base.py +++ b/metagpt/rag/rankers/base.py @@ -3,9 +3,8 @@ from abc import abstractmethod from typing import Optional -from llama_index import QueryBundle from llama_index.core.postprocessor.types import BaseNodePostprocessor -from llama_index.core.schema import NodeWithScore +from llama_index.core.schema import NodeWithScore, QueryBundle class RAGRanker(BaseNodePostprocessor):