agentscope-runtime 0.1.0__tar.gz → 0.1.2__tar.gz

{agentscope_runtime-0.1.0 → agentscope_runtime-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentscope-runtime
-Version: 0.1.0
+Version: 0.1.2
 Summary: A production-ready runtime framework for agent applications, providing secure sandboxed execution environments and scalable deployment solutions with multi-framework support.
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
@@ -17,14 +17,18 @@ Requires-Dist: pytest-asyncio>=0.23.0; extra == "dev"
 Requires-Dist: pre-commit>=4.2.0; extra == "dev"
 Requires-Dist: jupyter-book>=1.0.4.post1; extra == "dev"
 Requires-Dist: furo>=2025.7.19; extra == "dev"
+Requires-Dist: pytest-cov>=6.2.1; extra == "dev"
+Requires-Dist: fakeredis>=2.31.0; extra == "dev"
+Requires-Dist: sphinx-autoapi>=3.6.0; extra == "dev"
 Provides-Extra: sandbox
 Requires-Dist: docker>=7.1.0; extra == "sandbox"
 Requires-Dist: steel-sdk>=0.1.0; extra == "sandbox"
 Requires-Dist: redis>=6.0.0b2; extra == "sandbox"
 Requires-Dist: oss2>=2.19.1; extra == "sandbox"
-Requires-Dist: requests>=2.32.4; extra == "sandbox"
 Requires-Dist: pydantic-settings>=2.9.1; extra == "sandbox"
 Requires-Dist: dotenv>=0.9.9; extra == "sandbox"
+Requires-Dist: kubernetes>=33.1.0; extra == "sandbox"
+Requires-Dist: shortuuid>=1.0.13; extra == "sandbox"
 Provides-Extra: agentscope
 Requires-Dist: agentscope==1.0.0; extra == "agentscope"
 Provides-Extra: langgraph
@@ -33,17 +37,35 @@ Provides-Extra: agno
 Requires-Dist: agno>=1.7.5; extra == "agno"
 Provides-Extra: a2a
 Requires-Dist: a2a-sdk>=0.3.0; extra == "a2a"
+Provides-Extra: autogen
+Requires-Dist: autogen-agentchat>=0.7.4; extra == "autogen"
+Requires-Dist: autogen-ext[openai]>=0.7.4; extra == "autogen"
+Provides-Extra: langchain-rag
+Requires-Dist: langchain>=0.3.25; extra == "langchain-rag"
+Requires-Dist: pymilvus>=2.6.0; extra == "langchain-rag"
+Requires-Dist: langchain-community>=0.3.27; extra == "langchain-rag"
+Requires-Dist: langchain-milvus>=0.2.1; extra == "langchain-rag"
 Dynamic: license-file
 
 <div align="center">
 
 # AgentScope Runtime
 
-[![Python Version](https://img.shields.io/badge/python-3.10%2B-blue.svg)](https://python.org)
-[![Version](https://img.shields.io/badge/version-0.1.0-green.svg)](https://github.com/agentscope-ai/agentscope-runtime)
-[![License](https://img.shields.io/badge/license-Apache%202.0-yellow.svg)](LICENSE)
-[![Code Style](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
-
+[![PyPI](https://img.shields.io/pypi/v/agentscope-runtime?label=PyPI&color=brightgreen&logo=python)](https://pypi.org/project/agentscope-runtime/)
+[![Python Version](https://img.shields.io/badge/python-3.10%2B-blue.svg?logo=python&label=Python)](https://python.org)
+[![License](https://img.shields.io/badge/license-Apache%202.0-red.svg?logo=apache&label=Liscnese)](LICENSE)
+[![Code Style](https://img.shields.io/badge/code%20style-black-black.svg?logo=python&label=CodeStyle)](https://github.com/psf/black)
+[![GitHub Stars](https://img.shields.io/github/stars/agentscope-ai/agentscope-runtime?style=flat&logo=github&color=yellow&label=Stars)](https://github.com/agentscope-ai/agentscope-runtime/stargazers)
+[![GitHub Forks](https://img.shields.io/github/forks/agentscope-ai/agentscope-runtime?style=flat&logo=github&color=purple&label=Forks)](https://github.com/agentscope-ai/agentscope-runtime/network)
+[![Build Status](https://img.shields.io/badge/build-passing-brightgreen.svg?logo=githubactions&label=Build)](https://github.com/agentscope-ai/agentscope-runtime/actions)
+[![Cookbook](https://img.shields.io/badge/📚_Cookbook-English|中文-teal.svg)](https://runtime.agentscope.io)
+[![DeepWiki](https://img.shields.io/badge/DeepWiki-agentscope--runtime-navy.svg?logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAACwAAAAyCAYAAAAnWDnqAAAAAXNSR0IArs4c6QAAA05JREFUaEPtmUtyEzEQhtWTQyQLHNak2AB7ZnyXZMEjXMGeK/AIi+QuHrMnbChYY7MIh8g01fJoopFb0uhhEqqcbWTp06/uv1saEDv4O3n3dV60RfP947Mm9/SQc0ICFQgzfc4CYZoTPAswgSJCCUJUnAAoRHOAUOcATwbmVLWdGoH//PB8mnKqScAhsD0kYP3j/Yt5LPQe2KvcXmGvRHcDnpxfL2zOYJ1mFwrryWTz0advv1Ut4CJgf5uhDuDj5eUcAUoahrdY/56ebRWeraTjMt/00Sh3UDtjgHtQNHwcRGOC98BJEAEymycmYcWwOprTgcB6VZ5JK5TAJ+fXGLBm3FDAmn6oPPjR4rKCAoJCal2eAiQp2x0vxTPB3ALO2CRkwmDy5WohzBDwSEFKRwPbknEggCPB/imwrycgxX2NzoMCHhPkDwqYMr9tRcP5qNrMZHkVnOjRMWwLCcr8ohBVb1OMjxLwGCvjTikrsBOiA6fNyCrm8V1rP93iVPpwaE+gO0SsWmPiXB+jikdf6SizrT5qKasx5j8ABbHpFTx+vFXp9EnYQmLx02h1QTTrl6eDqxLnGjporxl3NL3agEvXdT0WmEost648sQOYAeJS9Q7bfUVoMGnjo4AZdUMQku50McDcMWcBPvr0SzbTAFDfvJqwLzgxwATnCgnp4wDl6Aa+Ax283gghmj+vj7feE2KBBRMW3FzOpLOADl0Isb5587h/U4gGvkt5v60Z1VLG8BhYjbzRwyQZemwAd6cCR5/XFWLYZRIMpX39AR0tjaGGiGzLVyhse5C9RKC6ai42ppWPKiBagOvaYk8lO7DajerabOZP46Lby5wKjw1HCRx7p9sVMOWGzb/vA1hwiWc6jm3MvQDTogQkiqIhJV0nBQBTU+3okKCFDy9WwferkHjtxib7t3xIUQtHxnIwtx4mpg26/HfwVNVDb4oI9RHmx5WGelRVlrtiw43zboCLaxv46AZeB3IlTkwouebTr1y2NjSpHz68WNFjHvupy3q8TFn3Hos2IAk4Ju5dCo8B3wP7VPr/FGaKiG+T+v+TQqIrOqMTL1VdWV1DdmcbO8KXBz6esmYWYKPwDL5b5FA1a0hwapHiom0r/cKaoqr+27/XcrS5UwSMbQAAAABJRU5ErkJggg==)](https://deepwiki.com/agentscope-ai/agentscope-runtime)
+[![A2A](https://img.shields.io/badge/A2A-Agent_to_Agent-blue.svg?label=A2A)](https://a2a-protocol.org/)
+[![MCP](https://img.shields.io/badge/MCP-Model_Context_Protocol-purple.svg?logo=plug&label=MCP)](https://modelcontextprotocol.io/)
+[![Discord](https://img.shields.io/badge/Discord-Join_Us-blueviolet.svg?logo=discord)](https://discord.gg/eYMpfnkG8h)
+[![DingTalk](https://img.shields.io/badge/DingTalk-Join_Us-orange.svg)](https://qr.dingtalk.com/action/joingroup?code=v1,k1,OmDlBXpjW+I2vWjKDsjvI9dhcXjGZi3bQiojOq3dlDw=&_dt_no_comment=1&origin=11)
+
+[[Cookbook]](https://runtime.agentscope.io/)
 [[中文README]](README_zh.md)
 
 **A Production-Ready Runtime Framework for Intelligent Agent Applications**
@@ -81,7 +103,7 @@ Welcome to join our community on
 
 - [🚀 Quick Start](#-quick-start)
 - [📚 Cookbook](#-cookbook)
-- [🔌Agent Framework Integration](#-agent-framework-integration)
+- [🔌 Agent Framework Integration](#-agent-framework-integration)
 - [🏗️ Deployment](#️-deployment)
 - [🤝 Contributing](#-contributing)
 - [📄 License](#-license)
@@ -96,14 +118,30 @@ Welcome to join our community on
 
 ### Installation
 
+From PyPI:
+
 ```bash
-# Install dependencies
+# Install core dependencies
 pip install agentscope-runtime
 
 # Install sandbox dependencies
 pip install "agentscope-runtime[sandbox]"
 ```
 
+(Optional) From source:
+
+```bash
+# Pull the source code from GitHub
+git clone -b main https://github.com/agentscope-ai/agentscope-runtime.git
+cd agentscope-runtime
+
+# Install core dependencies
+pip install -e .
+
+# Install sandbox dependencies
+pip install -e ".[sandbox]"
+```
+
 ### Basic Agent Usage Example
 
 This example demonstrates how to create a simple LLM agent using AgentScope Runtime and stream responses from the Qwen model.
@@ -164,14 +202,19 @@ with BaseSandbox() as box:
     print(box.run_shell_command(command="echo hello"))
 ```
 
+> [!NOTE]
+>
+> Current version requires Docker or Kubernetes to be installed and running on your system. Please refer to [this tutorial](https://runtime.agentscope.io/en/sandbox.html) for more details.
+
 ---
 
 ## 📚 Cookbook
 
-- **[📖 Cookbook](/cookbook/en/intro.md)**: Comprehensive tutorials
-- **[💡 Concept](/cookbook/en/concept.md)**: Core concepts and architecture overview
-- **[🚀 Quick Start](/cookbook/en/quickstart.md)**: Quick start tutorial
-- **[🏠 Demo House](/cookbook/en/demohouse.md)**: Rich example projects
+- **[📖 Cookbook](https://runtime.agentscope.io/en/intro.html)**: Comprehensive tutorials
+- **[💡 Concept](https://runtime.agentscope.io/en/concept.html)**: Core concepts and architecture overview
+- **[🚀 Quick Start](https://runtime.agentscope.io/en/quickstart.html)**: Quick start tutorial
+- **[🏠 Demo House](https://runtime.agentscope.io/en/demohouse.html)**: Rich example projects
+- **[📋 API Reference](https://runtime.agentscope.io/en/api/index.html)**: Complete API documentation
 
 ---
 
@@ -180,6 +223,7 @@ with BaseSandbox() as box:
 ### AgentScope Integration
 
 ```python
+# pip install "agentscope-runtime[agentscope]"
 import os
 
 from agentscope.agent import ReActAgent
@@ -202,6 +246,7 @@ agent = AgentScopeAgent(
 ### Agno Integration
 
 ```python
+# pip install "agentscope-runtime[agno]"
 from agno.agent import Agent
 from agno.models.openai import OpenAIChat
 from agentscope_runtime.engine.agents.agno_agent import AgnoAgent
@@ -218,9 +263,30 @@ agent = AgnoAgent(
 )
 ```
 
+### AutoGen Integration
+
+```python
+# pip install "agentscope-runtime[autogen]"
+from autogen_agentchat.agents import AssistantAgent
+from autogen_ext.models.openai import OpenAIChatCompletionClient
+from agentscope_runtime.engine.agents.autogen_agent import AutogenAgent
+
+agent = AutogenAgent(
+    name="Friday",
+    model=OpenAIChatCompletionClient(
+        model="gpt-4",
+    ),
+    agent_config={
+        "system_message": "You're a helpful assistant",
+    },
+    agent_builder=AssistantAgent,
+)
+```
+
 ### LangGraph Integration
 
 ```python
+# pip install "agentscope-runtime[langgraph]"
 from typing import TypedDict
 from langgraph import graph, types
 from agentscope_runtime.engine.agents.langgraph_agent import LangGraphAgent

{agentscope_runtime-0.1.0 → agentscope_runtime-0.1.2}/README.md

@@ -2,11 +2,21 @@
 
 # AgentScope Runtime
 
-[![Python Version](https://img.shields.io/badge/python-3.10%2B-blue.svg)](https://python.org)
-[![Version](https://img.shields.io/badge/version-0.1.0-green.svg)](https://github.com/agentscope-ai/agentscope-runtime)
-[![License](https://img.shields.io/badge/license-Apache%202.0-yellow.svg)](LICENSE)
-[![Code Style](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
-
+[![PyPI](https://img.shields.io/pypi/v/agentscope-runtime?label=PyPI&color=brightgreen&logo=python)](https://pypi.org/project/agentscope-runtime/)
+[![Python Version](https://img.shields.io/badge/python-3.10%2B-blue.svg?logo=python&label=Python)](https://python.org)
+[![License](https://img.shields.io/badge/license-Apache%202.0-red.svg?logo=apache&label=Liscnese)](LICENSE)
+[![Code Style](https://img.shields.io/badge/code%20style-black-black.svg?logo=python&label=CodeStyle)](https://github.com/psf/black)
+[![GitHub Stars](https://img.shields.io/github/stars/agentscope-ai/agentscope-runtime?style=flat&logo=github&color=yellow&label=Stars)](https://github.com/agentscope-ai/agentscope-runtime/stargazers)
+[![GitHub Forks](https://img.shields.io/github/forks/agentscope-ai/agentscope-runtime?style=flat&logo=github&color=purple&label=Forks)](https://github.com/agentscope-ai/agentscope-runtime/network)
+[![Build Status](https://img.shields.io/badge/build-passing-brightgreen.svg?logo=githubactions&label=Build)](https://github.com/agentscope-ai/agentscope-runtime/actions)
+[![Cookbook](https://img.shields.io/badge/📚_Cookbook-English|中文-teal.svg)](https://runtime.agentscope.io)
+[![DeepWiki](https://img.shields.io/badge/DeepWiki-agentscope--runtime-navy.svg?logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAACwAAAAyCAYAAAAnWDnqAAAAAXNSR0IArs4c6QAAA05JREFUaEPtmUtyEzEQhtWTQyQLHNak2AB7ZnyXZMEjXMGeK/AIi+QuHrMnbChYY7MIh8g01fJoopFb0uhhEqqcbWTp06/uv1saEDv4O3n3dV60RfP947Mm9/SQc0ICFQgzfc4CYZoTPAswgSJCCUJUnAAoRHOAUOcATwbmVLWdGoH//PB8mnKqScAhsD0kYP3j/Yt5LPQe2KvcXmGvRHcDnpxfL2zOYJ1mFwrryWTz0advv1Ut4CJgf5uhDuDj5eUcAUoahrdY/56ebRWeraTjMt/00Sh3UDtjgHtQNHwcRGOC98BJEAEymycmYcWwOprTgcB6VZ5JK5TAJ+fXGLBm3FDAmn6oPPjR4rKCAoJCal2eAiQp2x0vxTPB3ALO2CRkwmDy5WohzBDwSEFKRwPbknEggCPB/imwrycgxX2NzoMCHhPkDwqYMr9tRcP5qNrMZHkVnOjRMWwLCcr8ohBVb1OMjxLwGCvjTikrsBOiA6fNyCrm8V1rP93iVPpwaE+gO0SsWmPiXB+jikdf6SizrT5qKasx5j8ABbHpFTx+vFXp9EnYQmLx02h1QTTrl6eDqxLnGjporxl3NL3agEvXdT0WmEost648sQOYAeJS9Q7bfUVoMGnjo4AZdUMQku50McDcMWcBPvr0SzbTAFDfvJqwLzgxwATnCgnp4wDl6Aa+Ax283gghmj+vj7feE2KBBRMW3FzOpLOADl0Isb5587h/U4gGvkt5v60Z1VLG8BhYjbzRwyQZemwAd6cCR5/XFWLYZRIMpX39AR0tjaGGiGzLVyhse5C9RKC6ai42ppWPKiBagOvaYk8lO7DajerabOZP46Lby5wKjw1HCRx7p9sVMOWGzb/vA1hwiWc6jm3MvQDTogQkiqIhJV0nBQBTU+3okKCFDy9WwferkHjtxib7t3xIUQtHxnIwtx4mpg26/HfwVNVDb4oI9RHmx5WGelRVlrtiw43zboCLaxv46AZeB3IlTkwouebTr1y2NjSpHz68WNFjHvupy3q8TFn3Hos2IAk4Ju5dCo8B3wP7VPr/FGaKiG+T+v+TQqIrOqMTL1VdWV1DdmcbO8KXBz6esmYWYKPwDL5b5FA1a0hwapHiom0r/cKaoqr+27/XcrS5UwSMbQAAAABJRU5ErkJggg==)](https://deepwiki.com/agentscope-ai/agentscope-runtime)
+[![A2A](https://img.shields.io/badge/A2A-Agent_to_Agent-blue.svg?label=A2A)](https://a2a-protocol.org/)
+[![MCP](https://img.shields.io/badge/MCP-Model_Context_Protocol-purple.svg?logo=plug&label=MCP)](https://modelcontextprotocol.io/)
+[![Discord](https://img.shields.io/badge/Discord-Join_Us-blueviolet.svg?logo=discord)](https://discord.gg/eYMpfnkG8h)
+[![DingTalk](https://img.shields.io/badge/DingTalk-Join_Us-orange.svg)](https://qr.dingtalk.com/action/joingroup?code=v1,k1,OmDlBXpjW+I2vWjKDsjvI9dhcXjGZi3bQiojOq3dlDw=&_dt_no_comment=1&origin=11)
+
+[[Cookbook]](https://runtime.agentscope.io/)
 [[中文README]](README_zh.md)
 
 **A Production-Ready Runtime Framework for Intelligent Agent Applications**
@@ -44,7 +54,7 @@ Welcome to join our community on
 
 - [🚀 Quick Start](#-quick-start)
 - [📚 Cookbook](#-cookbook)
-- [🔌Agent Framework Integration](#-agent-framework-integration)
+- [🔌 Agent Framework Integration](#-agent-framework-integration)
 - [🏗️ Deployment](#️-deployment)
 - [🤝 Contributing](#-contributing)
 - [📄 License](#-license)
@@ -59,14 +69,30 @@ Welcome to join our community on
 
 ### Installation
 
+From PyPI:
+
 ```bash
-# Install dependencies
+# Install core dependencies
 pip install agentscope-runtime
 
 # Install sandbox dependencies
 pip install "agentscope-runtime[sandbox]"
 ```
 
+(Optional) From source:
+
+```bash
+# Pull the source code from GitHub
+git clone -b main https://github.com/agentscope-ai/agentscope-runtime.git
+cd agentscope-runtime
+
+# Install core dependencies
+pip install -e .
+
+# Install sandbox dependencies
+pip install -e ".[sandbox]"
+```
+
 ### Basic Agent Usage Example
 
 This example demonstrates how to create a simple LLM agent using AgentScope Runtime and stream responses from the Qwen model.
@@ -127,14 +153,19 @@ with BaseSandbox() as box:
     print(box.run_shell_command(command="echo hello"))
 ```
 
+> [!NOTE]
+>
+> Current version requires Docker or Kubernetes to be installed and running on your system. Please refer to [this tutorial](https://runtime.agentscope.io/en/sandbox.html) for more details.
+
 ---
 
 ## 📚 Cookbook
 
-- **[📖 Cookbook](/cookbook/en/intro.md)**: Comprehensive tutorials
-- **[💡 Concept](/cookbook/en/concept.md)**: Core concepts and architecture overview
-- **[🚀 Quick Start](/cookbook/en/quickstart.md)**: Quick start tutorial
-- **[🏠 Demo House](/cookbook/en/demohouse.md)**: Rich example projects
+- **[📖 Cookbook](https://runtime.agentscope.io/en/intro.html)**: Comprehensive tutorials
+- **[💡 Concept](https://runtime.agentscope.io/en/concept.html)**: Core concepts and architecture overview
+- **[🚀 Quick Start](https://runtime.agentscope.io/en/quickstart.html)**: Quick start tutorial
+- **[🏠 Demo House](https://runtime.agentscope.io/en/demohouse.html)**: Rich example projects
+- **[📋 API Reference](https://runtime.agentscope.io/en/api/index.html)**: Complete API documentation
 
 ---
 
@@ -143,6 +174,7 @@ with BaseSandbox() as box:
 ### AgentScope Integration
 
 ```python
+# pip install "agentscope-runtime[agentscope]"
 import os
 
 from agentscope.agent import ReActAgent
@@ -165,6 +197,7 @@ agent = AgentScopeAgent(
 ### Agno Integration
 
 ```python
+# pip install "agentscope-runtime[agno]"
 from agno.agent import Agent
 from agno.models.openai import OpenAIChat
 from agentscope_runtime.engine.agents.agno_agent import AgnoAgent
@@ -181,9 +214,30 @@ agent = AgnoAgent(
 )
 ```
 
+### AutoGen Integration
+
+```python
+# pip install "agentscope-runtime[autogen]"
+from autogen_agentchat.agents import AssistantAgent
+from autogen_ext.models.openai import OpenAIChatCompletionClient
+from agentscope_runtime.engine.agents.autogen_agent import AutogenAgent
+
+agent = AutogenAgent(
+    name="Friday",
+    model=OpenAIChatCompletionClient(
+        model="gpt-4",
+    ),
+    agent_config={
+        "system_message": "You're a helpful assistant",
+    },
+    agent_builder=AssistantAgent,
+)
+```
+
 ### LangGraph Integration
 
 ```python
+# pip install "agentscope-runtime[langgraph]"
 from typing import TypedDict
 from langgraph import graph, types
 from agentscope_runtime.engine.agents.langgraph_agent import LangGraphAgent

{agentscope_runtime-0.1.0 → agentscope_runtime-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentscope-runtime"
-version = "0.1.0"
+version = "0.1.2"
 description = "A production-ready runtime framework for agent applications, providing secure sandboxed execution environments and scalable deployment solutions with multi-framework support."
 readme = "README.md"
 requires-python = ">=3.10"
@@ -32,6 +32,9 @@ dev = [
     "pre-commit>=4.2.0",
     "jupyter-book>=1.0.4.post1",
     "furo>=2025.7.19",
+    "pytest-cov>=6.2.1",
+    "fakeredis>=2.31.0",
+    "sphinx-autoapi>=3.6.0",
 ]
 
 sandbox = [
@@ -39,9 +42,10 @@ sandbox = [
     "steel-sdk>=0.1.0",
     "redis>=6.0.0b2",
     "oss2>=2.19.1",
-    "requests>=2.32.4",
     "pydantic-settings>=2.9.1",
     "dotenv>=0.9.9",
+    "kubernetes>=33.1.0",
+    "shortuuid>=1.0.13",
 ]
 
 agentscope = [
@@ -58,3 +62,14 @@ agno = [
 a2a = [
     "a2a-sdk>=0.3.0",
 ]
+autogen = [
+    "autogen-agentchat>=0.7.4",
+    "autogen-ext[openai]>=0.7.4",
+]
+
+langchain_rag=[
+    "langchain>=0.3.25",
+    "pymilvus>=2.6.0",
+    "langchain-community>=0.3.27",
+    "langchain-milvus>=0.2.1",
+]

{agentscope_runtime-0.1.0 → agentscope_runtime-0.1.2}/src/agentscope_runtime/engine/agents/agentscope_agent/agent.py

@@ -85,6 +85,7 @@ class AgentScopeContextAdapter:
 
     @staticmethod
     def converter(message: Message):
+        # TODO: support more message type
         if message.role not in ["user", "system", "assistant"]:
             role_label = "user"
         else:

{agentscope_runtime-0.1.0 → agentscope_runtime-0.1.2}/src/agentscope_runtime/engine/agents/agno_agent.py

@@ -53,6 +53,7 @@ class AgnoContextAdapter:
 
     @staticmethod
     def converter(message: Message):
+        # TODO: support more message type
         return dict(message)
 
     async def adapt_new_message(self):

agentscope_runtime-0.1.2/src/agentscope_runtime/engine/agents/autogen_agent.py

@@ -0,0 +1,245 @@
+# -*- coding: utf-8 -*-
+from typing import Optional, Type
+
+from autogen_core.models import ChatCompletionClient
+from autogen_core.tools import FunctionTool
+from autogen_agentchat.agents import AssistantAgent
+from autogen_agentchat.messages import (
+    TextMessage,
+    ToolCallExecutionEvent,
+    ToolCallRequestEvent,
+    ModelClientStreamingChunkEvent,
+)
+
+from ..agents import Agent
+from ..schemas.context import Context
+from ..schemas.agent_schemas import (
+    Message,
+    TextContent,
+    DataContent,
+    FunctionCall,
+    FunctionCallOutput,
+    MessageType,
+    RunStatus,
+)
+
+
+class AutogenContextAdapter:
+    def __init__(self, context: Context, attr: dict):
+        self.context = context
+        self.attr = attr
+
+        # Adapted attribute
+        self.toolkit = None
+        self.model = None
+        self.memory = None
+        self.new_message = None
+
+    async def initialize(self):
+        self.model = await self.adapt_model()
+        self.memory = await self.adapt_memory()
+        self.new_message = await self.adapt_new_message()
+        self.toolkit = await self.adapt_tools()
+
+    async def adapt_memory(self):
+        messages = []
+
+        # Build context
+        for msg in self.context.session.messages[:-1]:  # Exclude the last one
+            messages.append(AutogenContextAdapter.converter(msg))
+
+        return messages
+
+    @staticmethod
+    def converter(message: Message):
+        # TODO: support more message type
+        return TextMessage.load(
+            {
+                "id": message.id,
+                "source": message.role,
+                "content": message.content[0].text if message.content else "",
+            },
+        )
+
+    async def adapt_new_message(self):
+        last_message = self.context.session.messages[-1]
+
+        return AutogenContextAdapter.converter(last_message)
+
+    async def adapt_model(self):
+        return self.attr["model"]
+
+    async def adapt_tools(self):
+        toolkit = self.attr["agent_config"].get("toolkit", [])
+        tools = self.attr["tools"]
+
+        # in case, tools is None and tools == []
+        if not tools:
+            return toolkit
+
+        if self.context.activate_tools:
+            # Only add activated tool
+            activated_tools = self.context.activate_tools
+        else:
+            from ...sandbox.tools.utils import setup_tools
+
+            activated_tools = setup_tools(
+                tools=self.attr["tools"],
+                environment_manager=self.context.environment_manager,
+                session_id=self.context.session.id,
+                user_id=self.context.session.user_id,
+                include_schemas=False,
+            )
+
+        for tool in activated_tools:
+            func = FunctionTool(
+                func=tool.make_function(),
+                description=tool.schema["function"]["description"],
+                name=tool.name,
+            )
+            toolkit.append(func)
+
+        return toolkit
+
+
+class AutogenAgent(Agent):
+    def __init__(
+        self,
+        name: str,
+        model: ChatCompletionClient,
+        tools=None,
+        agent_config=None,
+        agent_builder: Optional[Type[AssistantAgent]] = AssistantAgent,
+    ):
+        super().__init__(name=name, agent_config=agent_config)
+
+        assert isinstance(
+            model,
+            ChatCompletionClient,
+        ), "model must be a subclass of ChatCompletionClient in autogen"
+
+        # Set default agent_builder
+        if agent_builder is None:
+            agent_builder = AssistantAgent
+
+        assert issubclass(
+            agent_builder,
+            AssistantAgent,
+        ), "agent_builder must be a subclass of AssistantAgent in autogen"
+
+        # Replace name if not exists
+        self.agent_config["name"] = self.agent_config.get("name") or name
+
+        self._attr = {
+            "model": model,
+            "tools": tools,
+            "agent_config": self.agent_config,
+            "agent_builder": agent_builder,
+        }
+        self._agent = None
+        self.tools = tools
+
+    def copy(self) -> "AutogenAgent":
+        return AutogenAgent(**self._attr)
+
+    def build(self, as_context):
+        self._agent = self._attr["agent_builder"](
+            **self._attr["agent_config"],
+            model_client=as_context.model,
+            tools=as_context.toolkit,
+        )
+
+        return self._agent
+
+    async def run(self, context):
+        ag_context = AutogenContextAdapter(context=context, attr=self._attr)
+        await ag_context.initialize()
+
+        # We should always build a new agent since the state is manage outside
+        # the agent
+        self._agent = self.build(ag_context)
+
+        resp = self._agent.run_stream(
+            task=ag_context.memory + [ag_context.new_message],
+        )
+
+        text_message = Message(
+            type=MessageType.MESSAGE,
+            role="assistant",
+            status=RunStatus.InProgress,
+        )
+        yield text_message
+
+        text_delta_content = TextContent(delta=True)
+        is_text_delta = False
+        stream_mode = False
+        async for event in resp:
+            if getattr(event, "source", "user") == "user":
+                continue
+
+            if isinstance(event, TextMessage):
+                if stream_mode:
+                    continue
+                is_text_delta = True
+                text_delta_content.text = event.content
+                text_delta_content = text_message.add_delta_content(
+                    new_content=text_delta_content,
+                )
+                yield text_delta_content
+            elif isinstance(event, ModelClientStreamingChunkEvent):
+                stream_mode = True
+                is_text_delta = True
+                text_delta_content.text = event.content
+                text_delta_content = text_message.add_delta_content(
+                    new_content=text_delta_content,
+                )
+                yield text_delta_content
+            elif isinstance(event, ToolCallRequestEvent):
+                data = DataContent(
+                    data=FunctionCall(
+                        call_id=event.id,
+                        name=event.content[0].name,
+                        arguments=event.content[0].arguments,
+                    ).model_dump(),
+                )
+                message = Message(
+                    type=MessageType.PLUGIN_CALL,
+                    role="assistant",
+                    status=RunStatus.Completed,
+                    content=[data],
+                )
+                yield message
+            elif isinstance(event, ToolCallExecutionEvent):
+                data = DataContent(
+                    data=FunctionCallOutput(
+                        call_id=event.id,
+                        output=event.content[0].content,
+                    ).model_dump(),
+                )
+                message = Message(
+                    type=MessageType.PLUGIN_CALL_OUTPUT,
+                    role="assistant",
+                    status=RunStatus.Completed,
+                    content=[data],
+                )
+                yield message
+
+                # Add to message
+                is_text_delta = True
+                text_delta_content.text = event.content[0].content
+                text_delta_content = text_message.add_delta_content(
+                    new_content=text_delta_content,
+                )
+                yield text_delta_content
+
+        if is_text_delta:
+            yield text_message.content_completed(text_delta_content.index)
+            yield text_message.completed()
+
+    async def run_async(
+        self,
+        context,
+        **kwargs,
+    ):
+        async for event in self.run(context):
+            yield event

{agentscope_runtime-0.1.0 → agentscope_runtime-0.1.2}/src/agentscope_runtime/engine/schemas/agent_schemas.py

@@ -210,7 +210,7 @@ class Content(Event):
     delta: Optional[bool] = False
     """Whether this content is a delta."""
 
-    msg_id: str = None
+    msg_id: Optional[str] = None
     """message unique id"""