apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-30 11:26: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

6
tests/metagpt/document_store/test_faiss_store.py
View file

@ -25,7 +25,7 @@ async def test_search_json():
@pytest.mark.asyncio
async def test_search_xlsx():
store = FaissStore(EXAMPLE_PATH / "example.xlsx")
store = FaissStore(EXAMPLE_PATH / "example.xlsx", meta_col="Answer", content_col="Question")
role = Sales(profile="Sales", store=store)
query = "Which facial cleanser is good for oily skin?"
result = await role.run(query)
@ -36,5 +36,5 @@ async def test_search_xlsx():
async def test_write():
store = FaissStore(EXAMPLE_PATH / "example.xlsx", meta_col="Answer", content_col="Question")
_faiss_store = store.write()
assert _faiss_store.docstore
assert _faiss_store.index
assert _faiss_store.storage_context.docstore
assert _faiss_store.storage_context.vector_store.client