feat: 添加 Google ADK、CrewAI、LangChain 和 PydanticAI 模型适配器的流选项支持，更新数据 API 以包含使用情况，优化测试用例

Change-Id: If568ea68a8f888784070096876f3b61535ea87d7

Author: OhYee

.github/workflows/release-test.yml

@@ -0,0 +1,67 @@
+name: Release Test Package
+
+on:
+  push:
+    tags:
+      - 'test-v*'
+
+jobs:
+  release-test:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      id-token: write
+    steps:
+    - uses: actions/checkout@v4
+
+    - name: Set up Python
+      uses: actions/setup-python@v5
+      with:
+        python-version: '3.10'
+
+    - name: Extract version from tag
+      run: |
+        TAG="${{ github.ref_name }}"
+        # 从 test-v0.0.1 中提取 0.0.1
+        VERSION="${TAG#test-v}"
+        echo "VERSION=${VERSION}" >> $GITHUB_ENV
+        echo "Extracted version: ${VERSION}"
+
+    - name: Update package name and version in pyproject.toml
+      run: |
+        # 修改包名为 agentrun-inner-test
+        sed -i 's/name = "agentrun-sdk"/name = "agentrun-inner-test"/' pyproject.toml
+        # 修改版本号
+        sed -i 's/version = "[^"]*"/version = "'${VERSION}'"/' pyproject.toml
+        echo "Updated pyproject.toml:"
+        head -10 pyproject.toml
+
+    - name: Update __version__ in __init__.py
+      run: |
+        if grep -q "__version__" agentrun/__init__.py; then
+          sed -i 's/__version__ = "[^"]*"/__version__ = "'${VERSION}'"/' agentrun/__init__.py
+        else
+          sed -i '1a __version__ = "'${VERSION}'"' agentrun/__init__.py
+        fi
+        echo "Updated __init__.py version to ${VERSION}"
+        grep "__version__" agentrun/__init__.py
+
+    - name: Build package
+      run: |
+        python -m pip install --upgrade pip
+        pip install build twine
+        python -m build
+        echo "Package built successfully"
+        ls -la dist/
+
+    - name: Verify package
+      run: |
+        python -m twine check dist/*
+        echo "Package verification completed"
+
+    - name: Publish to PyPI
+      uses: pypa/gh-action-pypi-publish@release/v1
+      with:
+        password: ${{ secrets.PYPI_API_TOKEN }}
+        verify-metadata: false
+

agentrun/integration/agentscope/model_adapter.py

@@ -56,4 +56,5 @@ def wrap_model(self, common_model: CommonModel) -> Any:
                 "base_url": info.base_url,
                 "http_client": AsyncClient(headers=info.headers),
             },
+            generate_kwargs={"stream_options": {"include_usage": True}},
         )

agentrun/integration/crewai/model_adapter.py

@@ -22,5 +22,6 @@ def wrap_model(self, common_model: Any) -> Any:
             model=f"{info.provider or 'openai'}/{info.model}",
             base_url=info.base_url,
             default_headers=info.headers,
+            stream_options={"include_usage": True},
             # async_client=AsyncClient(headers=info.headers),
         )

agentrun/integration/google_adk/model_adapter.py

@@ -39,4 +39,5 @@ def wrap_model(self, common_model: CommonModel) -> Any:
             api_base=info.base_url,
             api_key=info.api_key,
             extra_headers=info.headers,
+            stream_options={"include_usage": True},
         )

agentrun/integration/langchain/model_adapter.py

@@ -33,4 +33,5 @@ def wrap_model(self, common_model: Any) -> Any:
             model=info.model,
             base_url=info.base_url,
             async_client=AsyncClient(headers=info.headers),
+            stream_usage=True,
         )

agentrun/integration/pydantic_ai/model_adapter.py

@@ -1,8 +1,6 @@
 """PydanticAI 模型适配器 / PydanticAI Model Adapter"""
 
-from contextlib import asynccontextmanager
-import json
-from typing import Any, AsyncIterator
+from typing import Any
 
 from agentrun.integration.utils.adapter import ModelAdapter
 from agentrun.integration.utils.model import CommonModel
@@ -19,6 +17,7 @@ def wrap_model(self, common_model: CommonModel) -> Any:
         try:
             from pydantic_ai.models.openai import OpenAIChatModel
             from pydantic_ai.providers.openai import OpenAIProvider
+            from pydantic_ai.settings import ModelSettings
         except Exception as e:
             raise ImportError(
                 "PydanticAI is not installed. "
@@ -36,6 +35,9 @@ def wrap_model(self, common_model: CommonModel) -> Any:
                 api_key=info.api_key,
                 http_client=AsyncClient(headers=info.headers),
             ),
+            settings=ModelSettings(
+                extra_body={"stream_options": {"include_usage": True}}
+            ),
         )
 
 

agentrun/model/api/data.py

@@ -54,6 +54,10 @@ def completions(
             **self.headers,
             **kwargs.get("headers", {}),
         }
+        if kwargs["stream_options"] is None:
+            kwargs["stream_options"] = {}
+        kwargs["stream_options"]["include_usage"] = True
+
         from litellm import completion
 
         return completion(
@@ -82,6 +86,9 @@ def responses(
             **self.headers,
             **kwargs.get("headers", {}),
         }
+        if kwargs["stream_options"] is None:
+            kwargs["stream_options"] = {}
+        kwargs["stream_options"]["include_usage"] = True
         from litellm import responses
 
         return responses(

tests/unittests/integration/test_integration.py

@@ -53,12 +53,16 @@ def fake_completion(*args, **kwargs):
             messages = kwargs.get("messages") or []
             tools_payload = kwargs.get("tools")
             assert kwargs.get("stream") in (None, False)
+            assert pydash.get(kwargs, "stream_options.include_usage") is True
+
             return self._build_model_response(messages, tools_payload)
 
         async def fake_acompletion(*args, **kwargs):
             messages = kwargs.get("messages") or []
             tools_payload = kwargs.get("tools")
             assert kwargs.get("stream") in (None, False)
+            assert pydash.get(kwargs, "stream_options.include_usage") is True
+
             return self._build_model_response(messages, tools_payload)
 
         monkeypatch.setattr("litellm.completion", fake_completion)