Merge pull request #84 from small-thinking/refactor-prompt

Add dspy as a tool
small-thinking · Jun 10, 2024 · 74b5ee4 · 74b5ee4
2 parents 2b6490a + eb16551
commit 74b5ee4
Show file tree

Hide file tree

Showing 7 changed files with 1,207 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