apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-30 03:16:23 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

replace langchain with llama-index

Browse source
This commit is contained in:
better629 2024-01-19 17:37:12 +08:00
parent 7005a1e86f
commit 4fcf724797
15 changed files with 175 additions and 71 deletions
Download patch file Download diff file Expand all files Collapse all files

BIN
examples/example.faiss
View file

Binary file not shown.

BIN
examples/example.pkl
View file

Binary file not shown.

4
examples/search_kb.py
View file

@ -6,7 +6,7 @@
"""
import asyncio
from langchain.embeddings import OpenAIEmbeddings
from llama_index.embeddings import OpenAIEmbedding
from metagpt.config2 import config
from metagpt.const import DATA_PATH, EXAMPLE_PATH
@ -17,7 +17,7 @@ from metagpt.roles import Sales
def get_store():
llm = config.get_openai_llm()
embedding = OpenAIEmbeddings(openai_api_key=llm.api_key, openai_api_base=llm.base_url)
embedding = OpenAIEmbedding(api_key=llm.api_key, api_base=llm.base_url)
return FaissStore(DATA_PATH / "example.json", embedding=embedding)