Merge pull request #97 from small-thinking/fix-metaclass

Update package
small-thinking · Jul 13, 2024 · dfee4f8 · dfee4f8
2 parents 274f53f + 1d1dede
commit dfee4f8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/polymind/core/tool.py b/polymind/core/tool.py
@@ -278,7 +278,7 @@ async def _execute(self, input: Message) -> Message:
         pass
 
 
-class OptimizableBaseTool(AbstractTool, dspy.Predict):
+class OptimizableBaseTool(AbstractTool, dspy.Predict, metaclass=type(BaseModel)):
     def __call__(self, input: Message) -> Message:
         self._validate_input_message(input)
         output_message = self.forward(**input.content)

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "polymind"
-version = "0.0.55"  # Update this version before publishing to PyPI
+version = "0.0.56"  # Update this version before publishing to PyPI
 description = "PolyMind is a customizable collaborative multi-agent framework for collective intelligence and distributed problem solving."
 authors = ["TechTao"]
 license = "MIT License"
@@ -49,7 +49,7 @@ skip = ["__init__.py"]
 [tool.flake8]
 max-line-length = 120
 exclude = ["__init__.py"]
-ignore = ["Q000"]
+ignore = ["Q000", "E203"]
 
 [build-system]
 requires = ["poetry-core"]