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 1/2] 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", From b275f1a3f8c33ea5832d1656e26e9e4b8831b631 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=8E=98=E6=9D=83=20=E9=A9=AC?= Date: Tue, 16 Jan 2024 10:45:01 +0800 Subject: [PATCH 2/2] feat: +ver fixbug: RPC think --- metagpt/roles/role.py | 1 + setup.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/metagpt/roles/role.py b/metagpt/roles/role.py index 3d5e55057..36d007f3b 100644 --- a/metagpt/roles/role.py +++ b/metagpt/roles/role.py @@ -494,6 +494,7 @@ class Role(SerializationMixin, is_polymorphic_base=True): async def think(self) -> Action: """The exported `think` function""" + await self._observe() await self._think() return self.rc.todo diff --git a/setup.py b/setup.py index ea84fe299..ca8bb3980 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.4", + version="0.6.5", description="The Multi-Agent Framework", long_description=long_description, long_description_content_type="text/markdown",