From ec2e8bdca397a1ff495e5fed0d702ddfc98f163b Mon Sep 17 00:00:00 2001 From: seehi <6580@pm.me> Date: Sat, 16 Mar 2024 09:03:34 +0800 Subject: [PATCH] for consistency, move rag.llm to rag.factories.llm --- metagpt/rag/engines/simple.py | 2 +- metagpt/rag/factories/__init__.py | 3 ++- metagpt/rag/{ => factories}/llm.py | 0 3 files changed, 3 insertions(+), 2 deletions(-) rename metagpt/rag/{ => factories}/llm.py (100%) diff --git a/metagpt/rag/engines/simple.py b/metagpt/rag/engines/simple.py index a9bde8128..02f9ca7b1 100644 --- a/metagpt/rag/engines/simple.py +++ b/metagpt/rag/engines/simple.py @@ -31,11 +31,11 @@ from llama_index.core.schema import ( from metagpt.rag.factories import ( get_index, get_rag_embedding, + get_rag_llm, get_rankers, get_retriever, ) from metagpt.rag.interface import NoEmbedding, RAGObject -from metagpt.rag.llm import get_rag_llm from metagpt.rag.retrievers.base import ModifiableRAGRetriever, PersistableRAGRetriever from metagpt.rag.retrievers.hybrid_retriever import SimpleHybridRetriever from metagpt.rag.schema import ( diff --git a/metagpt/rag/factories/__init__.py b/metagpt/rag/factories/__init__.py index 56101de7c..caa35405f 100644 --- a/metagpt/rag/factories/__init__.py +++ b/metagpt/rag/factories/__init__.py @@ -4,5 +4,6 @@ from metagpt.rag.factories.retriever import get_retriever from metagpt.rag.factories.ranker import get_rankers from metagpt.rag.factories.embedding import get_rag_embedding from metagpt.rag.factories.index import get_index +from metagpt.rag.factories.llm import get_rag_llm -__all__ = ["get_retriever", "get_rankers", "get_rag_embedding", "get_index"] +__all__ = ["get_retriever", "get_rankers", "get_rag_embedding", "get_index", "get_rag_llm"] diff --git a/metagpt/rag/llm.py b/metagpt/rag/factories/llm.py similarity index 100% rename from metagpt/rag/llm.py rename to metagpt/rag/factories/llm.py