feat: Add base tool calling agent class

Author: dsowinski2

plugins/enthusiast-agent-re-act/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "enthusiast-agent-re-act"
-version = "1.3.0"
+version = "1.3.1"
 description = "Base implementation of a ReAct agent for Enthusiast"
 authors = [
     {name = "Rafal Cymerys",email = "[email protected]"}

plugins/enthusiast-agent-re-act/src/enthusiast_agent_re_act/base_re_act_agent.py

@@ -1,6 +1,6 @@
 from typing import Any
 
-from enthusiast_common.agents import BaseAgent
+from enthusiast_common.agents import AgentType, BaseAgent
 from langchain.agents import AgentExecutor, create_react_agent
 from langchain_core.memory import BaseMemory
 from langchain_core.tools import BaseTool, render_text_description_and_args
@@ -9,7 +9,7 @@
 
 
 class BaseReActAgent(BaseAgent):
-    IS_REACT = True
+    AGENT_TYPE = AgentType.RE_ACT
 
     def get_answer(self, input_text: str) -> str:
         agent_executor = self._build_agent_executor()

plugins/enthusiast-agent-tool-calling/README.md

@@ -0,0 +1,21 @@
+[project]
+name = "enthusiast-agent-tool-calling"
+version = "1.0.0"
+description = "Base implementation of a Tool Calling agent for Enthusiast"
+authors = [
+    {name = "Damian Sowiński",email = "[email protected]"}
+]
+readme = "README.md"
+requires-python = ">=3.10,<4"
+dependencies = [
+    "enthusiast-common (>=1.4.0,<2.0.0)",
+    "langchain (>=0.3.26,<0.4.0)"
+]
+
+[tool.poetry]
+packages = [{include = "enthusiast_agent_tool_calling", from = "src"}]
+
+
+[build-system]
+requires = ["poetry-core>=2.0.0,<3.0.0"]
+build-backend = "poetry.core.masonry.api"

plugins/enthusiast-agent-tool-calling/poetry.lock

@@ -0,0 +1 @@
+from .base_tool_calling_agent import BaseToolCallingAgent as BaseToolCallingAgent

plugins/enthusiast-agent-tool-calling/pyproject.toml

@@ -0,0 +1,38 @@
+from typing import Any
+
+from enthusiast_common.agents import AgentType, BaseAgent
+from langchain.agents import AgentExecutor, create_tool_calling_agent
+from langchain_core.memory import BaseMemory
+from langchain_core.tools import BaseTool
+
+
+class BaseToolCallingAgent(BaseAgent):
+    AGENT_TYPE = AgentType.TOOL_CALLING
+
+    def get_answer(self, input_text: str) -> str:
+        agent_executor = self._build_agent_executor()
+        response = agent_executor.invoke({"input": input_text}, config=self._build_invoke_config())
+        return response["output"]
+
+    def _build_tools(self) -> list[BaseTool]:
+        return [tool.as_tool() for tool in self._tools]
+
+    def _build_memory(self) -> BaseMemory:
+        return self._injector.chat_limited_memory
+
+    def _build_invoke_config(self) -> dict[str, Any]:
+        if self._callback_handler:
+            return {"callbacks": [self._callback_handler]}
+
+        return {}
+
+    def _build_agent_executor(self) -> AgentExecutor:
+        tools = self._build_tools()
+        agent = create_tool_calling_agent(
+            tools=tools,
+            llm=self._llm,
+            prompt=self._prompt,
+        )
+        return AgentExecutor(
+            agent=agent, tools=tools, verbose=True, memory=self._build_memory(), return_intermediate_steps=True
+        )

plugins/enthusiast-agent-tool-calling/src/enthusiast_agent_tool_calling/__init__.py

@@ -1,3 +1,3 @@
-from .base import BaseAgent
+from .base import AgentType, BaseAgent
 
-__all__ = ["BaseAgent"]
+__all__ = ["BaseAgent", "AgentType"]

plugins/enthusiast-agent-tool-calling/src/enthusiast_agent_tool_calling/base_tool_calling_agent.py

@@ -1,4 +1,5 @@
 from abc import ABC, ABCMeta, abstractmethod
+from enum import Enum
 from typing import Any
 
 from langchain_core.callbacks import BaseCallbackHandler
@@ -13,6 +14,12 @@
 from ..utils import RequiredFieldsModel, validate_required_vars
 
 
+class AgentType(str, Enum):
+    BASE = "base"
+    RE_ACT = "re_act"
+    TOOL_CALLING = "tool_calling"
+
+
 class ExtraArgsClassBaseMeta(ABCMeta):
     REQUIRED_VARS = {}
 
@@ -42,7 +49,7 @@ class BaseAgent(ABC, ExtraArgsClassBase):
     PROMPT_EXTENSION = None
     TOOLS = []
 
-    IS_REACT = False
+    AGENT_TYPE: AgentType = AgentType.BASE
     FILE_UPLOAD = False
     DEFAULT_FILE_TOOLS = [FileListTool, FileRetrievalTool]
 

plugins/enthusiast-common/enthusiast_common/agents/__init__.py

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "enthusiast-common"
-version = "1.4.0"
+version = "1.4.1"
 description = "Core interfaces for developing custom Enthusiast plugins and integrations."
 authors = ["Rafal Cymerys <[email protected]>"]
 readme = "README.md"