Add dspy as a tool

small-thinking · Jun 10, 2024 · 95c578d · 95c578d
1 parent 2b6490a
commit 95c578d
Show file tree

Hide file tree

Showing 6 changed files with 1,206 additions and 337 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/polymind/__init__.py b/polymind/__init__.py
@@ -3,10 +3,10 @@
 from .core.agent import Agent, ThoughtProcess
 from .core.message import Message
 from .core.task import AtomTask, BaseTask, CompositeTask, SequentialTask
-from .core.tool import BaseTool
+from .core.tool import BaseTool, LLMTool, DspyPipelineTool
 from .core_tools.retrieve_tool import KnowledgeRetrieveTool, ToolRetriever
 from .core_tools.index_tool import KnowledgeIndexTool, ToolIndexer
 from .core.logger import Logger
 
 # Expose the tools
-from .core_tools.llm_tool import LLMTool, OpenAIChatTool, OpenAIEmbeddingTool
+from .core_tools.llm_tool import OpenAIChatTool, OpenAIEmbeddingTool