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

add sales test

Browse source
This commit is contained in:
geekan 2023-12-25 22:13:36 +08:00
parent 9f653ea60b
commit bd1014e19a
5 changed files with 33 additions and 84 deletions
Download patch file Download diff file Expand all files Collapse all files

BIN
examples/faq.xlsx Normal file
View file

Binary file not shown.

20
examples/search_kb.py
View file

@ -6,28 +6,18 @@
"""
import asyncio
from langchain.embeddings import OpenAIEmbeddings
from metagpt.config import CONFIG
from metagpt.const import EXAMPLE_PATH
from metagpt.document_store import FaissStore
from metagpt.logs import logger
from metagpt.roles import Sales
def get_store():
embedding = OpenAIEmbeddings(openai_api_key=CONFIG.openai_api_key, openai_api_base=CONFIG.openai_base_url)
return FaissStore(EXAMPLE_PATH / "example.json", embedding=embedding)
async def search():
role = Sales(profile="Sales", store=get_store())
queries = ["Which facial cleanser is good for oily skin?", "Is L'Oreal good to use?"]
for query in queries:
logger.info(f"User: {query}")
result = await role.run(query)
logger.info(result)
store = FaissStore(EXAMPLE_PATH / "example.json")
role = Sales(profile="Sales", store=store)
query = "Which facial cleanser is good for oily skin?"
result = await role.run(query)
logger.info(result)
if __name__ == "__main__":