vokit 0.1.4__tar.gz

vokit-0.1.4/PKG-INFO

@@ -0,0 +1,109 @@
+Metadata-Version: 2.4
+Name: vokit
+Version: 0.1.4
+Summary: VirtualOffice Agent SDK — 透明拦截转发 + 平台 API 封装
+Author: VirtualOffice Team
+License: MIT
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: wrapt>=1.15
+Requires-Dist: httpx>=0.24
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.21; extra == "dev"
+Requires-Dist: ruff>=0.1.0; extra == "dev"
+Requires-Dist: mypy>=1.0; extra == "dev"
+Requires-Dist: aiohttp>=3.8; extra == "dev"
+Requires-Dist: requests>=2.28; extra == "dev"
+Requires-Dist: urllib3>=1.26; extra == "dev"
+Requires-Dist: respx>=0.21; extra == "dev"
+
+# vokit — VirtualOffice Agent SDK
+
+一行 `import vokit` 完成拦截激活 + API 就绪。
+
+## 安装
+
+```bash
+# 预发布版本
+pip install git+http://oauth2:YOUR_TOKEN@gitlab.deepleaper.com/deepbrain/vokit-sdk.git@prerelease
+```
+
+## 快速开始
+
+### 1. 拦截 LLM 请求
+
+在 Agent 的**入口文件最顶部** import 一次即可：
+
+```python
+import vokit
+```
+
+- **只需 import 一次**：patch 注册在进程级别的 HTTP 库入口，全局生效。
+- **放在最顶部**：确保在所有 LLM SDK 被 import 之前 patch 已注册好。
+- **幂等安全**：多次 import 不会重复注册。
+
+### 2. 使用平台 API
+
+```python
+import vokit
+
+# 提交产出物
+vokit.submit_output(
+    output_type="text",
+    content="分析报告内容..."
+)
+
+# 查询工位资源
+resources = vokit.get_resources()
+
+# 上报任务状态
+vokit.report_status(tasks=[
+    {"name": "task-1", "status": "running", "progress": 0.75},
+])
+
+# 上报指标数据
+vokit.report_metrics(metrics=[
+    {"metricType": "tokens_used", "metricValue": 1500, "snapshotDate": "2026-04-14"},
+])
+```
+
+## 运行模式
+
+### 开发模式
+
+不设置 `VO_GATEWAY_URL` 和 `VO_ACCESS_TOKEN` 时自动进入：
+- LLM 请求**直连** Provider，不被拦截
+- 平台 API 返回 Mock 数据
+
+```bash
+export OPENAI_API_KEY="your-key"
+python your_agent.py
+```
+
+### 生产模式
+
+设置环境变量后自动进入：
+
+```bash
+export VO_GATEWAY_URL=http://localhost:8888
+export VO_ACCESS_TOKEN=<your-jwt-token>
+export OPENAI_API_KEY="your-key"
+python your_agent.py
+```
+
+## 环境变量
+
+| 变量 | 说明 |
+|------|------|
+| `VO_GATEWAY_URL` | 内网网关地址，缺失则进入开发模式 |
+| `VO_ACCESS_TOKEN` | 网关 Bearer Token，缺失则进入开发模式 |
+
+## Python 版本
+
+要求 Python 3.10+
+
+## License
+
+MIT

vokit-0.1.4/README.md

@@ -0,0 +1,88 @@
+# vokit — VirtualOffice Agent SDK
+
+一行 `import vokit` 完成拦截激活 + API 就绪。
+
+## 安装
+
+```bash
+# 预发布版本
+pip install git+http://oauth2:YOUR_TOKEN@gitlab.deepleaper.com/deepbrain/vokit-sdk.git@prerelease
+```
+
+## 快速开始
+
+### 1. 拦截 LLM 请求
+
+在 Agent 的**入口文件最顶部** import 一次即可：
+
+```python
+import vokit
+```
+
+- **只需 import 一次**：patch 注册在进程级别的 HTTP 库入口，全局生效。
+- **放在最顶部**：确保在所有 LLM SDK 被 import 之前 patch 已注册好。
+- **幂等安全**：多次 import 不会重复注册。
+
+### 2. 使用平台 API
+
+```python
+import vokit
+
+# 提交产出物
+vokit.submit_output(
+    output_type="text",
+    content="分析报告内容..."
+)
+
+# 查询工位资源
+resources = vokit.get_resources()
+
+# 上报任务状态
+vokit.report_status(tasks=[
+    {"name": "task-1", "status": "running", "progress": 0.75},
+])
+
+# 上报指标数据
+vokit.report_metrics(metrics=[
+    {"metricType": "tokens_used", "metricValue": 1500, "snapshotDate": "2026-04-14"},
+])
+```
+
+## 运行模式
+
+### 开发模式
+
+不设置 `VO_GATEWAY_URL` 和 `VO_ACCESS_TOKEN` 时自动进入：
+- LLM 请求**直连** Provider，不被拦截
+- 平台 API 返回 Mock 数据
+
+```bash
+export OPENAI_API_KEY="your-key"
+python your_agent.py
+```
+
+### 生产模式
+
+设置环境变量后自动进入：
+
+```bash
+export VO_GATEWAY_URL=http://localhost:8888
+export VO_ACCESS_TOKEN=<your-jwt-token>
+export OPENAI_API_KEY="your-key"
+python your_agent.py
+```
+
+## 环境变量
+
+| 变量 | 说明 |
+|------|------|
+| `VO_GATEWAY_URL` | 内网网关地址，缺失则进入开发模式 |
+| `VO_ACCESS_TOKEN` | 网关 Bearer Token，缺失则进入开发模式 |
+
+## Python 版本
+
+要求 Python 3.10+
+
+## License
+
+MIT

vokit-0.1.4/pyproject.toml

@@ -0,0 +1,51 @@
+[build-system]
+requires = ["setuptools>=68.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "vokit"
+version = "0.1.4"
+description = "VirtualOffice Agent SDK — 透明拦截转发 + 平台 API 封装"
+readme = "README.md"
+license = { text = "MIT" }
+requires-python = ">=3.10"
+authors = [
+    { name = "VirtualOffice Team" },
+]
+dependencies = [
+    "wrapt>=1.15",
+    "httpx>=0.24",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=7.0",
+    "pytest-cov>=4.0",
+    "pytest-asyncio>=0.21",
+    "ruff>=0.1.0",
+    "mypy>=1.0",
+    "aiohttp>=3.8",
+    "requests>=2.28",
+    "urllib3>=1.26",
+    "respx>=0.21",
+]
+
+[tool.setuptools.packages.find]
+include = ["vokit*"]
+
+[tool.ruff]
+target-version = "py310"
+line-length = 120
+
+[tool.ruff.lint]
+select = ["E", "F", "W", "I"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+asyncio_mode = "auto"
+
+[tool.mypy]
+python_version = "3.10"
+strict = true
+warn_return_any = true
+warn_unused_configs = true

vokit-0.1.4/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

vokit-0.1.4/tests/test_aiohttp_handler.py

@@ -0,0 +1,50 @@
+"""aiohttp handler 专项测试。"""
+from unittest.mock import AsyncMock, MagicMock, patch
+
+import pytest
+
+from vokit._patcher import _handle_aiohttp_async
+
+
+@pytest.mark.asyncio
+async def test_handler_reads_headers_from_positional():
+    """headers 通过位置参数 args[2] 传入时，handler 应正确读取并传递给 intercept_request。"""
+    original_headers = {"Authorization": "Bearer sk-xxx", "Content-Type": "application/json"}
+    new_headers = {"X-Vokit-Intercepted": "1", "Authorization": "Bearer gateway-token"}
+
+    with patch("vokit._patcher.intercept_request") as mock_intercept:
+        mock_intercept.return_value = ("http://gateway:8080/v1/chat", new_headers)
+        wrapped = AsyncMock(return_value=MagicMock(status=200))
+
+        # headers 作为第 3 个位置参数传入（method, url, headers）
+        await _handle_aiohttp_async(
+            wrapped, MagicMock(),
+            ("POST", "https://api.openai.com/v1/chat/completions", original_headers),
+            {},
+        )
+
+        # 验证 intercept_request 被调用且收到了原始 headers
+        call_args = mock_intercept.call_args
+        # call_args[0] = (url, body, headers)
+        assert call_args[0][0] == "https://api.openai.com/v1/chat/completions"
+        assert call_args[0][2] == original_headers
+
+
+@pytest.mark.asyncio
+async def test_handler_headers_from_kwargs():
+    """headers 通过 kwargs 传入时，handler 应正常工作。"""
+    original_headers = {"Authorization": "Bearer sk-xxx"}
+    new_headers = {"X-Vokit-Intercepted": "1"}
+
+    with patch("vokit._patcher.intercept_request") as mock_intercept:
+        mock_intercept.return_value = ("http://gateway:8080/v1/chat", new_headers)
+        wrapped = AsyncMock(return_value=MagicMock(status=200))
+
+        await _handle_aiohttp_async(
+            wrapped, MagicMock(),
+            ("POST", "https://api.openai.com/v1/chat/completions"),
+            {"headers": original_headers},
+        )
+
+        call_args = mock_intercept.call_args
+        assert call_args[0][2] == original_headers