minor tweaks

ShoggothAI · Jul 29, 2024 · 2971c5f · 2971c5f
1 parent 7f90b49
commit 2971c5f
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 3 deletions.
diff --git a/motleycrew/agents/__init__.py b/motleycrew/agents/__init__.py
@@ -3,9 +3,11 @@
 from .abstract_parent import MotleyAgentAbstractParent
 from .output_handler import MotleyOutputHandler
 from .parent import MotleyAgentParent
+from .langchain import LangchainMotleyAgent
 
 __all__ = [
     "MotleyAgentAbstractParent",
     "MotleyAgentParent",
     "MotleyOutputHandler",
+    "LangchainMotleyAgent",
 ]
diff --git a/motleycrew/agents/langchain/langchain.py b/motleycrew/agents/langchain/langchain.py
@@ -6,6 +6,7 @@
 from langchain_core.chat_history import InMemoryChatMessageHistory
 from langchain_core.runnables import RunnableConfig
 from langchain_core.runnables.history import RunnableWithMessageHistory, GetSessionHistoryCallable
+from langchain_core.prompts.chat import ChatPromptTemplate
 
 from motleycrew.agents.mixins import LangchainOutputHandlingAgentMixin
 from motleycrew.agents.parent import MotleyAgentParent
@@ -21,7 +22,7 @@ def __init__(
         self,
         description: str | None = None,
         name: str | None = None,
-        prompt_prefix: str | None = None,
+        prompt_prefix: str | ChatPromptTemplate | None = None,
         agent_factory: MotleyAgentFactory[AgentExecutor] | None = None,
         tools: Sequence[MotleySupportedTool] | None = None,
         output_handler: MotleySupportedTool | None = None,

diff --git a/motleycrew/agents/langchain/tool_calling_react.py b/motleycrew/agents/langchain/tool_calling_react.py
@@ -101,7 +101,7 @@ def __init__(
         tools: Sequence[MotleySupportedTool],
         description: str | None = None,
         name: str | None = None,
-        prompt_prefix: str | None = None,
+        prompt_prefix: str | ChatPromptTemplate | None = None,
         prompt: ChatPromptTemplate | None = None,
         chat_history: bool | GetSessionHistoryCallable = True,
         output_handler: MotleySupportedTool | None = None,

diff --git a/motleycrew/agents/parent.py b/motleycrew/agents/parent.py
@@ -55,7 +55,7 @@ class MotleyAgentParent(MotleyAgentAbstractParent, ABC):
 
     def __init__(
         self,
-        prompt_prefix: str | None = None,
+        prompt_prefix: str | ChatPromptTemplate | None = None,
         description: str | None = None,
         name: str | None = None,
         agent_factory: MotleyAgentFactory | None = None,