agentrun-inner-test 0.0.46__py3-none-any.whl

agentrun/integration/builtin/toolset.py

@@ -0,0 +1,47 @@
+"""内置工具集集成函数 / Built-in ToolSet Integration Functions
+
+提供快速创建通用工具集对象的便捷函数。
+Provides convenient functions for quickly creating common toolset objects.
+"""
+
+from typing import Optional, Union
+
+from agentrun.integration.utils.tool import CommonToolSet
+from agentrun.toolset import ToolSet, ToolSetClient
+from agentrun.utils.config import Config
+
+
+def toolset(
+    input: Union[str, ToolSet], config: Optional[Config] = None
+) -> CommonToolSet:
+    """将内置工具集封装为通用工具集 / Wrap built-in toolset as CommonToolSet
+
+    支持从工具集名称或 ToolSet 实例创建通用工具集。
+    Supports creating CommonToolSet from toolset name or ToolSet instance.
+
+    Args:
+        input: 工具集名称或 ToolSet 实例 / Toolset name or ToolSet instance
+        config: 配置对象 / Configuration object
+
+    Returns:
+        CommonToolSet: 通用工具集实例 / CommonToolSet instance
+
+    Examples:
+        >>> # 从工具集名称创建 / Create from toolset name
+        >>> ts = toolset("my-toolset")
+        >>>
+        >>> # 从 ToolSet 实例创建 / Create from ToolSet instance
+        >>> toolset_obj = ToolSetClient().get("my-toolset")
+        >>> ts = toolset(toolset_obj)
+        >>>
+        >>> # 转换为 LangChain 工具 / Convert to LangChain tools
+        >>> lc_tools = ts.to_langchain()
+    """
+
+    toolset = (
+        input
+        if isinstance(input, ToolSet)
+        else ToolSetClient().get(name=input, config=config)
+    )
+
+    return CommonToolSet.from_agentrun_toolset(toolset, config=config)

agentrun/integration/crewai/__init__.py

@@ -0,0 +1,12 @@
+"""CrewAI 集成模块。 / CrewAI 集成 Module
+
+提供 AgentRun 模型与沙箱工具的 CrewAI 适配入口。 / 提供 AgentRun 模型with沙箱工具的 CrewAI 适配入口。
+CrewAI 与 LangChain 兼容，因此直接复用 LangChain 的转换逻辑。 / CrewAI with LangChain 兼容，因此直接复用 LangChain 的转换逻辑。
+"""
+
+from .builtin import model, sandbox_toolset
+
+__all__ = [
+    "model",
+    "sandbox_toolset",
+]

agentrun/integration/crewai/adapter.py

@@ -0,0 +1,9 @@
+"""CrewAI 适配器 / CrewAI Adapters"""
+
+from .model_adapter import CrewAIModelAdapter
+from .tool_adapter import CrewAIToolAdapter
+
+__all__ = [
+    "CrewAIToolAdapter",
+    "CrewAIModelAdapter",
+]

agentrun/integration/crewai/builtin.py

@@ -0,0 +1,65 @@
+"""CrewAI 内置集成函数 / CrewAI Built-in Integration Functions
+
+提供快速创建 CrewAI 兼容模型和工具的便捷函数。
+Provides convenient functions for quickly creating CrewAI-compatible models and tools.
+"""
+
+from typing import Any, Callable, List, Optional, Union
+
+from typing_extensions import Unpack
+
+from agentrun.integration.builtin import model as _model
+from agentrun.integration.builtin import ModelArgs
+from agentrun.integration.builtin import sandbox_toolset as _sandbox_toolset
+from agentrun.integration.builtin import toolset as _toolset
+from agentrun.integration.utils.tool import Tool
+from agentrun.model import ModelProxy, ModelService
+from agentrun.sandbox import TemplateType
+from agentrun.toolset import ToolSet
+from agentrun.utils.config import Config
+
+
+def model(
+    name: Union[str, ModelProxy, ModelService],
+    **kwargs: Unpack[ModelArgs],
+):
+    """获取 AgentRun 模型并转换为 LangChain ``BaseChatModel``。 / CrewAI Built-in Integration Functions"""
+
+    m = _model(input=name, **kwargs)  # type: ignore
+    return m.to_crewai()
+
+
+def toolset(
+    name: Union[str, ToolSet],
+    *,
+    prefix: Optional[str] = None,
+    modify_tool_name: Optional[Callable[[Tool], Tool]] = None,
+    filter_tools_by_name: Optional[Callable[[str], bool]] = None,
+    config: Optional[Config] = None,
+) -> List[Any]:
+    """将内置工具集封装为 LangChain ``StructuredTool`` 列表。 / CrewAI Built-in Integration Functions"""
+
+    ts = _toolset(input=name, config=config)
+    return ts.to_crewai(
+        prefix=prefix,
+        modify_tool_name=modify_tool_name,
+        filter_tools_by_name=filter_tools_by_name,
+    )
+
+
+def sandbox_toolset(
+    template_name: str,
+    *,
+    template_type: TemplateType = TemplateType.CODE_INTERPRETER,
+    config: Optional[Config] = None,
+    sandbox_idle_timeout_seconds: int = 600,
+    prefix: Optional[str] = None,
+) -> List[Any]:
+    """将沙箱模板封装为 LangChain ``StructuredTool`` 列表。 / CrewAI Built-in Integration Functions"""
+
+    return _sandbox_toolset(
+        template_name=template_name,
+        template_type=template_type,
+        config=config,
+        sandbox_idle_timeout_seconds=sandbox_idle_timeout_seconds,
+    ).to_crewai(prefix=prefix)

agentrun/integration/crewai/model_adapter.py

@@ -0,0 +1,31 @@
+"""LangChain 模型适配器 / CrewAI Model Adapter
+
+将 CommonModel 包装为 LangChain BaseChatModel。"""
+
+from typing import Any
+
+from agentrun.integration.utils.adapter import ModelAdapter
+
+
+class CrewAIModelAdapter(ModelAdapter):
+    """CrewAI 模型适配器 / CrewAI Model Adapter
+
+    将 CommonModel 包装为 CrewAI BaseChatModel。"""
+
+    def wrap_model(self, common_model: Any) -> Any:
+        """包装 CommonModel 为 CrewAI BaseChatModel / CrewAI Model Adapter"""
+        from crewai import LLM
+
+        info = common_model.get_model_info()  # 确保模型可用
+
+        # 注意：不在此处设置 stream_options，因为：
+        # 1. CrewAI 内部决定是否使用流式请求
+        # 2. 在非流式请求中传递 stream_options 不符合 OpenAI API 规范
+        # 3. CrewAI 会自行处理 usage 信息
+        return LLM(
+            api_key=info.api_key,
+            model=f"{info.provider or 'openai'}/{info.model}",
+            base_url=info.base_url,
+            default_headers=info.headers,
+            # async_client=AsyncClient(headers=info.headers),
+        )

agentrun/integration/crewai/tool_adapter.py

@@ -0,0 +1,26 @@
+"""LangChain 工具适配器 / CrewAI Tool Adapter
+
+将标准工具定义转换为 LangChain StructuredTool 格式。"""
+
+from typing import Any, List
+
+from agentrun.integration.utils.adapter import ToolAdapter
+from agentrun.integration.utils.canonical import CanonicalTool
+
+
+class CrewAIToolAdapter(ToolAdapter):
+    """CrewAI 工具适配器 / CrewAI Tool Adapter
+
+    实现 CanonicalTool → CrewAI StructuredTool 的转换。"""
+
+    def from_canonical(self, tools: List[CanonicalTool]) -> Any:
+        """将标准格式转换为 CrewAI StructuredTool / CrewAI Tool Adapter"""
+        try:
+            from crewai.tools import tool
+        except ImportError as e:
+            raise ImportError(
+                "LangChain is not installed. "
+                "Install it with: pip install langchain-core"
+            ) from e
+
+        return [tool(t) for t in self.function_tools(tools)]

agentrun/integration/google_adk/__init__.py

@@ -0,0 +1,12 @@
+"""Google ADK 集成模块。 / Google ADK 集成 Module
+
+提供与 Google Agent Development Kit 的模型与沙箱工具集成。 / 提供with Google Agent Development Kit 的模型with沙箱工具集成。
+"""
+
+from .builtin import model, sandbox_toolset, toolset
+
+__all__ = [
+    "model",
+    "toolset",
+    "sandbox_toolset",
+]

agentrun/integration/google_adk/adapter.py

@@ -0,0 +1,15 @@
+"""Google ADK 适配器 / Google ADK Adapters
+
+提供 Google ADK 框架的消息、工具和模型适配器。"""
+
+from agentrun.integration.google_adk.message_adapter import (
+    GoogleADKMessageAdapter,
+)
+from agentrun.integration.google_adk.model_adapter import GoogleADKModelAdapter
+from agentrun.integration.google_adk.tool_adapter import GoogleADKToolAdapter
+
+__all__ = [
+    "GoogleADKMessageAdapter",
+    "GoogleADKToolAdapter",
+    "GoogleADKModelAdapter",
+]

agentrun/integration/google_adk/builtin.py

@@ -0,0 +1,65 @@
+"""Google ADK 内置集成函数 / Google ADK Built-in Integration Functions
+
+提供快速创建 Google ADK 兼容模型和工具的便捷函数。
+Provides convenient functions for quickly creating Google ADK-compatible models and tools.
+"""
+
+from typing import Any, Callable, List, Optional, Union
+
+from typing_extensions import Unpack
+
+from agentrun.integration.builtin import model as _model
+from agentrun.integration.builtin import ModelArgs
+from agentrun.integration.builtin import sandbox_toolset as _sandbox_toolset
+from agentrun.integration.builtin import toolset as _toolset
+from agentrun.integration.utils.tool import Tool
+from agentrun.model import ModelProxy, ModelService
+from agentrun.sandbox import TemplateType
+from agentrun.toolset import ToolSet
+from agentrun.utils.config import Config
+
+
+def model(
+    name: Union[str, ModelProxy, ModelService],
+    **kwargs: Unpack[ModelArgs],
+):
+    """获取 AgentRun 模型并转换为 LangChain ``BaseChatModel``。 / Google ADK Built-in Integration Functions"""
+
+    m = _model(input=name, **kwargs)  # type: ignore
+    return m.to_google_adk()
+
+
+def toolset(
+    name: Union[str, ToolSet],
+    *,
+    prefix: Optional[str] = None,
+    modify_tool_name: Optional[Callable[[Tool], Tool]] = None,
+    filter_tools_by_name: Optional[Callable[[str], bool]] = None,
+    config: Optional[Config] = None,
+) -> List[Any]:
+    """将内置工具集封装为 LangChain ``StructuredTool`` 列表。 / Google ADK Built-in Integration Functions"""
+
+    ts = _toolset(input=name, config=config)
+    return ts.to_google_adk(
+        prefix=prefix,
+        modify_tool_name=modify_tool_name,
+        filter_tools_by_name=filter_tools_by_name,
+    )
+
+
+def sandbox_toolset(
+    template_name: str,
+    *,
+    template_type: TemplateType = TemplateType.CODE_INTERPRETER,
+    config: Optional[Config] = None,
+    sandbox_idle_timeout_seconds: int = 600,
+    prefix: Optional[str] = None,
+) -> List[Any]:
+    """将沙箱模板封装为 LangChain ``StructuredTool`` 列表。 / Google ADK Built-in Integration Functions"""
+
+    return _sandbox_toolset(
+        template_name=template_name,
+        template_type=template_type,
+        config=config,
+        sandbox_idle_timeout_seconds=sandbox_idle_timeout_seconds,
+    ).to_google_adk(prefix=prefix)

agentrun/integration/google_adk/message_adapter.py

@@ -0,0 +1,144 @@
+"""Google ADK 消息适配器 / Google ADK Message Adapter
+
+将 Google ADK LlmRequest 转换为标准格式,供 ModelAdapter 内部使用。
+Converts Google ADK LlmRequest to canonical format for internal use by ModelAdapter.
+"""
+
+import json
+from typing import Any, List
+
+from agentrun.integration.utils.adapter import MessageAdapter
+from agentrun.integration.utils.canonical import (
+    CanonicalMessage,
+    CanonicalToolCall,
+    MessageRole,
+)
+
+
+class GoogleADKMessageAdapter(MessageAdapter):
+    """Google ADK 消息适配器 / Google ADK Message Adapter
+
+    实现 Google ADK LlmRequest → CanonicalMessage 的转换。
+    Implements conversion from Google ADK LlmRequest to CanonicalMessage.
+    """
+
+    def to_canonical(self, messages: Any) -> List[CanonicalMessage]:
+        """将 Google ADK LlmRequest 转换为标准格式 / Google ADK Message Adapter"""
+        canonical = []
+
+        # Google ADK 使用 LlmRequest，包含 contents 列表
+        if hasattr(messages, "contents"):
+            contents = messages.contents
+        elif isinstance(messages, list):
+            contents = messages
+        else:
+            contents = [messages]
+
+        # 处理 system_instruction (在 config 中)
+        if hasattr(messages, "config") and messages.config:
+            if (
+                hasattr(messages.config, "system_instruction")
+                and messages.config.system_instruction
+            ):
+                canonical.append(
+                    CanonicalMessage(
+                        role=MessageRole.SYSTEM,
+                        content=messages.config.system_instruction,
+                    )
+                )
+
+        # 处理 contents
+        for content in contents:
+            # 确定角色
+            role = MessageRole.USER
+            if hasattr(content, "role"):
+                role_str = str(content.role).lower()
+                if "model" in role_str or "assistant" in role_str:
+                    role = MessageRole.ASSISTANT
+                elif "system" in role_str:
+                    role = MessageRole.SYSTEM
+                elif "tool" in role_str:
+                    role = MessageRole.TOOL
+                elif "function" in role_str:
+                    role = MessageRole.TOOL
+                else:
+                    role = MessageRole.USER
+
+            # 处理 parts
+            if hasattr(content, "parts"):
+                text_parts = []
+                tool_calls = []
+
+                for part in content.parts:
+                    # 处理文本
+                    if hasattr(part, "text") and part.text:
+                        text_parts.append(part.text)
+
+                    # 处理 function_call
+                    elif hasattr(part, "function_call") and part.function_call:
+                        func_call = part.function_call
+                        args = {}
+                        if hasattr(func_call, "args"):
+                            if isinstance(func_call.args, dict):
+                                args = func_call.args
+                            else:
+                                try:
+                                    args = json.loads(str(func_call.args))
+                                except (json.JSONDecodeError, TypeError):
+                                    args = {}
+
+                        tool_calls.append(
+                            CanonicalToolCall(
+                                id=getattr(
+                                    func_call, "id", f"call_{len(tool_calls)}"
+                                ),
+                                name=getattr(func_call, "name", ""),
+                                arguments=args,
+                            )
+                        )
+
+                    # 处理 function_response
+                    elif (
+                        hasattr(part, "function_response")
+                        and part.function_response
+                    ):
+                        func_resp = part.function_response
+                        response_content = ""
+                        if hasattr(func_resp, "response"):
+                            if isinstance(func_resp.response, dict):
+                                response_content = json.dumps(
+                                    func_resp.response
+                                )
+                            else:
+                                response_content = str(func_resp.response)
+                        else:
+                            response_content = str(func_resp)
+
+                        # function_response 表示工具返回结果
+                        canonical.append(
+                            CanonicalMessage(
+                                role=MessageRole.TOOL,
+                                content=response_content,
+                                tool_call_id=getattr(func_resp, "id", "call_0"),
+                            )
+                        )
+                        continue
+
+                # 构建消息
+                if text_parts or tool_calls:
+                    content_text = " ".join(text_parts) if text_parts else None
+                    canonical.append(
+                        CanonicalMessage(
+                            role=role,
+                            content=content_text,
+                            tool_calls=tool_calls if tool_calls else None,
+                        )
+                    )
+            else:
+                # 没有 parts，直接使用字符串内容
+                content_text = str(content) if content else None
+                canonical.append(
+                    CanonicalMessage(role=role, content=content_text)
+                )
+
+        return canonical

agentrun/integration/google_adk/model_adapter.py

@@ -0,0 +1,46 @@
+"""Google ADK 模型适配器 / Google ADK Model Adapter
+
+将 CommonModel 包装为 Google ADK BaseLlm。"""
+
+from typing import Any
+
+from agentrun.integration.google_adk.message_adapter import (
+    GoogleADKMessageAdapter,
+)
+from agentrun.integration.utils.adapter import ModelAdapter
+from agentrun.integration.utils.model import CommonModel
+
+
+class GoogleADKModelAdapter(ModelAdapter):
+    """Google ADK 模型适配器 / Google ADK Model Adapter
+
+    将 CommonModel 包装为 Google ADK BaseLlm。"""
+
+    def __init__(self):
+        """初始化适配器，创建内部的消息适配器 / Google ADK Message Adapter"""
+        self._message_adapter = GoogleADKMessageAdapter()
+
+    def wrap_model(self, common_model: CommonModel) -> Any:
+        """包装 CommonModel 为 Google ADK BaseLlm / Google ADK Model Adapter"""
+
+        try:
+            from google.adk.models.lite_llm import LiteLlm  # type: ignore
+        except ImportError as e:
+            raise ImportError(
+                "import google.adk.models.lite_llm failed."
+                "Google ADK may not installed, "
+                "Install it with: pip install google-adk"
+            ) from e
+
+        info = common_model.get_model_info()
+
+        # 注意：不在此处设置 stream_options，因为：
+        # 1. Google ADK 内部决定是否使用流式请求
+        # 2. 在非流式请求中传递 stream_options 不符合 OpenAI API 规范
+        # 3. Google ADK 会自行处理 usage 信息
+        return LiteLlm(
+            model=f"{info.provider or 'openai'}/{info.model}",
+            api_base=info.base_url,
+            api_key=info.api_key,
+            extra_headers=info.headers,
+        )