feat: Add ALTK Agent with tool validation and comprehensive tests

src/lfx/src/lfx/base/agents/altk_base_agent.py

@@ -4,30 +4,30 @@
 on user-facing configuration and small customizations.
 """
 
 from __future__ import annotations
 
 import uuid
 from abc import ABC, abstractmethod
 from typing import TYPE_CHECKING, Any, cast
 
 from altk.core.llm import get_llm
 from langchain.agents import AgentExecutor, BaseMultiActionAgent, BaseSingleActionAgent
 from langchain_anthropic.chat_models import ChatAnthropic
 from langchain_core.language_models.chat_models import BaseChatModel
 from langchain_core.messages import BaseMessage, HumanMessage
 from langchain_core.runnables import Runnable, RunnableBinding
 from langchain_core.tools import BaseTool
 from langchain_openai.chat_models.base import ChatOpenAI
-from pydantic import Field
 
 from lfx.base.agents.callback import AgentAsyncHandler
 from lfx.base.agents.events import ExceptionWithMessageError, process_agent_events
 from lfx.base.agents.utils import data_to_messages, get_chat_output_sender_name
 from lfx.components.agents import AgentComponent
 from lfx.log.logger import logger
 from lfx.memory import delete_message
 from lfx.schema.content_block import ContentBlock
 from lfx.schema.data import Data
+from pydantic import Field
 
 if TYPE_CHECKING:
     from collections.abc import Sequence

src/lfx/src/lfx/base/agents/altk_tool_wrappers.py

@@ -1,24 +1,24 @@
 import ast
 import json
 import uuid
 from typing import Any
 
 from altk.core.toolkit import AgentPhase, ComponentConfig
 from altk.post_tool.code_generation.code_generation import (
     CodeGenerationComponent,
     CodeGenerationComponentConfig,
 )
 from altk.post_tool.core.toolkit import CodeGenerationRunInput
 from altk.pre_tool.core import SPARCExecutionMode, SPARCReflectionRunInput, Track
 from altk.pre_tool.sparc import SPARCReflectionComponent
 from langchain_core.messages import BaseMessage
 from langchain_core.messages.base import message_to_dict
 from langchain_core.tools import BaseTool
-from pydantic import Field
 
 from lfx.base.agents.altk_base_agent import ALTKBaseTool, BaseToolWrapper
 from lfx.log.logger import logger
 from lfx.schema.data import Data
+from pydantic import Field
 
 # Maximum wrapper nesting depth to prevent infinite loops
 _MAX_WRAPPER_DEPTH = 10
@@ -141,6 +141,15 @@
         )
         return self._validate_and_run(*args, **kwargs)
 
+    @staticmethod
+    def _custom_message_to_dict(message: BaseMessage) -> dict:
+        """Convert a BaseMessage to a dictionary."""
+        if isinstance(message,BaseMessage):
+            return message_to_dict(message)
+        msg = f"Invalid message type: {type(message)}"
+        logger.error(msg,exc_info=True)
+        raise ValueError(msg) from None
+
     def _validate_and_run(self, *args, **kwargs) -> str:
         """Validate the tool call using SPARC and execute if valid."""
         # Check if validation should be bypassed
@@ -163,7 +172,7 @@
             and isinstance(self.conversation_context[0], BaseMessage)
         ):
             logger.debug("Converting BaseMessages to list of dictionaries for conversation context of SPARC")
-            self.conversation_context = [message_to_dict(msg) for msg in self.conversation_context]
+            self.conversation_context = [self._custom_message_to_dict(msg) for msg in self.conversation_context]
 
         logger.debug(
             f"Converted conversation context for SPARC for tool call:\n"