apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-09 15:52:38 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

rag add chromadb save&load example

Browse source
This commit is contained in:
seehi 2024-03-06 17:39:41 +08:00
parent c657af448b
commit 525b47b033
3 changed files with 32 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
metagpt/rag/engines/simple.py
View file

@ -100,7 +100,7 @@ class SimpleEngine(RetrieverQueryEngine):
llm: LLM = None,
retriever_configs: list[BaseRetrieverConfig] = None,
ranker_configs: list[BaseRankerConfig] = None,
):
) -> "SimpleEngine":
"""Load from previously maintained"""
index = get_index(index_config, embed_model=embed_model or get_rag_embedding())
return cls._from_index(index, llm=llm, retriever_configs=retriever_configs, ranker_configs=ranker_configs)
@ -112,7 +112,7 @@ class SimpleEngine(RetrieverQueryEngine):
llm: LLM = None,
retriever_configs: list[BaseRetrieverConfig] = None,
ranker_configs: list[BaseRankerConfig] = None,
):
) -> "SimpleEngine":
llm = llm or get_rag_llm()
retriever = get_retriever(configs=retriever_configs, index=index)
rankers = get_rankers(configs=ranker_configs, llm=llm)