CoACT initialize (#292)
This commit is contained in:
82
mm_agents/coact/autogen/interop/langchain/langchain_tool.py
Normal file
82
mm_agents/coact/autogen/interop/langchain/langchain_tool.py
Normal file
@@ -0,0 +1,82 @@
|
||||
# Copyright (c) 2023 - 2025, AG2ai, Inc., AG2ai open-source projects maintainers and core contributors
|
||||
#
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
|
||||
import sys
|
||||
from typing import Any, Optional
|
||||
|
||||
from ...doc_utils import export_module
|
||||
from ...import_utils import optional_import_block, require_optional_import
|
||||
from ...tools import Tool
|
||||
from ..registry import register_interoperable_class
|
||||
|
||||
__all__ = ["LangChainInteroperability"]
|
||||
|
||||
with optional_import_block():
|
||||
from langchain_core.tools import BaseTool as LangchainTool
|
||||
|
||||
|
||||
@register_interoperable_class("langchain")
|
||||
@export_module("autogen.interop")
|
||||
class LangChainInteroperability:
|
||||
"""A class implementing the `Interoperable` protocol for converting Langchain tools
|
||||
into a general `Tool` format.
|
||||
|
||||
This class takes a `LangchainTool` and converts it into a standard `Tool` object,
|
||||
ensuring compatibility between Langchain tools and other systems that expect
|
||||
the `Tool` format.
|
||||
"""
|
||||
|
||||
@classmethod
|
||||
@require_optional_import("langchain_core", "interop-langchain")
|
||||
def convert_tool(cls, tool: Any, **kwargs: Any) -> Tool:
|
||||
"""Converts a given Langchain tool into a general `Tool` format.
|
||||
|
||||
This method verifies that the provided tool is a valid `LangchainTool`,
|
||||
processes the tool's input and description, and returns a standardized
|
||||
`Tool` object.
|
||||
|
||||
Args:
|
||||
tool (Any): The tool to convert, expected to be an instance of `LangchainTool`.
|
||||
**kwargs (Any): Additional arguments, which are not supported by this method.
|
||||
|
||||
Returns:
|
||||
Tool: A standardized `Tool` object converted from the Langchain tool.
|
||||
|
||||
Raises:
|
||||
ValueError: If the provided tool is not an instance of `LangchainTool`, or if
|
||||
any additional arguments are passed.
|
||||
"""
|
||||
if not isinstance(tool, LangchainTool):
|
||||
raise ValueError(f"Expected an instance of `langchain_core.tools.BaseTool`, got {type(tool)}")
|
||||
if kwargs:
|
||||
raise ValueError(f"The LangchainInteroperability does not support any additional arguments, got {kwargs}")
|
||||
|
||||
# needed for type checking
|
||||
langchain_tool: LangchainTool = tool # type: ignore[no-any-unimported]
|
||||
|
||||
model_type = langchain_tool.get_input_schema()
|
||||
|
||||
def func(tool_input: model_type) -> Any: # type: ignore[valid-type]
|
||||
return langchain_tool.run(tool_input.model_dump()) # type: ignore[attr-defined]
|
||||
|
||||
return Tool(
|
||||
name=langchain_tool.name,
|
||||
description=langchain_tool.description,
|
||||
func_or_tool=func,
|
||||
)
|
||||
|
||||
@classmethod
|
||||
def get_unsupported_reason(cls) -> Optional[str]:
|
||||
if sys.version_info < (3, 9):
|
||||
return "This submodule is only supported for Python versions 3.9 and above"
|
||||
|
||||
with optional_import_block() as result:
|
||||
import langchain_core.tools # noqa: F401
|
||||
|
||||
if not result.is_successful:
|
||||
return (
|
||||
"Please install `interop-langchain` extra to use this module:\n\n\tpip install ag2[interop-langchain]"
|
||||
)
|
||||
|
||||
return None
|
||||
Reference in New Issue
Block a user