From 29d8326c06899535bb2d8953246aa30466b8f72f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=8E=98=E6=9D=83=20=E9=A9=AC?= Date: Wed, 10 Jan 2024 21:13:36 +0800 Subject: [PATCH] feat: +ver feat: Moderation + llm arg feat: +log --- metagpt/document_store/faiss_store.py | 6 +++++- metagpt/tools/moderation.py | 4 ++-- requirements.txt | 2 +- setup.py | 2 +- 4 files changed, 9 insertions(+), 5 deletions(-) diff --git a/metagpt/document_store/faiss_store.py b/metagpt/document_store/faiss_store.py index 1271f1c23..6f97141c2 100644 --- a/metagpt/document_store/faiss_store.py +++ b/metagpt/document_store/faiss_store.py @@ -40,7 +40,11 @@ class FaissStore(LocalStore): return FAISS.load_local(self.raw_data_path.parent, self.embedding, self.fname) def _write(self, docs, metadatas): - store = FAISS.from_texts(docs, self.embedding, metadatas=metadatas) + try: + store = FAISS.from_texts(docs, self.embedding, metadatas=metadatas) + except Exception as e: + logger.error(f"Failed to write. error: {e}") + raise e return store def persist(self): diff --git a/metagpt/tools/moderation.py b/metagpt/tools/moderation.py index cda164ec5..8effc0e8b 100644 --- a/metagpt/tools/moderation.py +++ b/metagpt/tools/moderation.py @@ -11,8 +11,8 @@ from metagpt.llm import LLM class Moderation: - def __init__(self): - self.llm = LLM() + def __init__(self, llm=None): + self.llm = llm or LLM() def handle_moderation_results(self, results): resp = [] diff --git a/requirements.txt b/requirements.txt index 0a54236f0..f8e4d2585 100644 --- a/requirements.txt +++ b/requirements.txt @@ -14,7 +14,7 @@ lancedb==0.4.0 langchain==0.0.352 loguru==0.6.0 meilisearch==0.21.0 -numpy==1.24.3 +numpy>=1.24.3 openai==1.6.0 openpyxl beautifulsoup4==4.12.2 diff --git a/setup.py b/setup.py index d997b5f62..ea84fe299 100644 --- a/setup.py +++ b/setup.py @@ -57,7 +57,7 @@ extras_require["dev"] = (["pylint~=3.0.3", "black~=23.3.0", "isort~=5.12.0", "pr setup( name="metagpt", - version="0.6.3", + version="0.6.4", description="The Multi-Agent Framework", long_description=long_description, long_description_content_type="text/markdown",