workbuddy2api 2.0.2__tar.gz → 2.0.3__tar.gz

{workbuddy2api-2.0.2 → workbuddy2api-2.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: workbuddy2api
-Version: 2.0.2
+Version: 2.0.3
 Summary: Local OpenAI/Responses/Anthropic compatible proxy for CodeBuddy
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
@@ -72,8 +72,10 @@ uv run python -m codebuddy_proxy --login --desensitize
 
 ```bash
 # 使用最新版本（推荐）
-uv run --with workbuddy2api python -m codebuddy_proxy --desensitize \
-  --log-file "$HOME/.workbuddy2api/codebuddy-proxy.jsonl"
+uv run --with workbuddy2api python -m codebuddy_proxy --desensitize
+
+# 首次使用：登录并启动
+uv run --with workbuddy2api python -m codebuddy_proxy --login --desensitize
 ```
 
 默认监听 `http://127.0.0.1:8787`

{workbuddy2api-2.0.2 → workbuddy2api-2.0.3}/README.md

@@ -57,8 +57,10 @@ uv run python -m codebuddy_proxy --login --desensitize
 
 ```bash
 # 使用最新版本（推荐）
-uv run --with workbuddy2api python -m codebuddy_proxy --desensitize \
-  --log-file "$HOME/.workbuddy2api/codebuddy-proxy.jsonl"
+uv run --with workbuddy2api python -m codebuddy_proxy --desensitize
+
+# 首次使用：登录并启动
+uv run --with workbuddy2api python -m codebuddy_proxy --login --desensitize
 ```
 
 默认监听 `http://127.0.0.1:8787`

{workbuddy2api-2.0.2 → workbuddy2api-2.0.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "workbuddy2api"
-version = "2.0.2"
+version = "2.0.3"
 description = "Local OpenAI/Responses/Anthropic compatible proxy for CodeBuddy"
 readme = "README.md"
 requires-python = ">=3.10"

{workbuddy2api-2.0.2 → workbuddy2api-2.0.3}/src/codebuddy_proxy/__main__.py

@@ -332,7 +332,7 @@ class RemoteConfigCache:
             
             # 构造请求头（包含认证信息）
             headers = state.client.auth_headers()
-            headers["User-Agent"] = "CodeBuddy-Proxy/2.0"
+            headers["User-Agent"] = "Mozilla/5.0 (compatible; Genie-IDE/1.0)"
             headers["Accept"] = "application/json"
             
             # 发起请求（超时 5 秒，匹配插件）
@@ -897,11 +897,18 @@ async def forward_chat(
     
     url = state.client.endpoint + "/v2/chat/completions"
     headers = {
+        "User-Agent": "Mozilla/5.0 (compatible; Genie-IDE/1.0)",
         **state.client.auth_headers(),
         "Content-Type": "application/json",
         "Accept": "text/event-stream",
     }
     
+    # 🔍 调试：输出实际发送的IDE识别headers
+    if state.logger:
+        ide_headers = {k: v for k, v in headers.items() 
+                      if k.startswith("X-IDE-") or k == "X-Product-Version" or k == "X-Machine-Id"}
+        diagnostic("upstream_ide_headers", **ide_headers)
+    
     if stream:
         # 流式：直接转发
         return StreamingResponse(

{workbuddy2api-2.0.2 → workbuddy2api-2.0.3}/src/codebuddy_proxy/codebuddy_client_demo.py

@@ -81,8 +81,11 @@ class CodeBuddyClient:
         body: Any = None,
         timeout: float = 30,
     ) -> Any:
-        request_headers = {"User-Agent": "CodeBuddyClientDemo/1.0"}
-        request_headers.update(headers or {})
+        # 🔥 所有请求都必须包含这两个基础标识（匹配插件行为）
+        request_headers = {
+            "User-Agent": "Mozilla/5.0 (compatible; Genie-IDE/1.0)",
+            "X-Product-Code": "codebuddy",
+        }
         data = None
         if body is not None:
             data = json.dumps(body).encode("utf-8")
@@ -110,9 +113,20 @@ class CodeBuddyClient:
             raise CodeBuddyError(f"{path} 返回的不是 JSON: {raw[:300]!r}") from exc
 
     def auth_headers(self, *, access: bool = True, refresh: bool = False) -> dict[str, str]:
+        """构造认证 headers，模拟插件行为"""
         account = self.session.get("account") or {}
         auth = self.session.get("auth") or {}
-        headers: dict[str, str] = {}
+        headers: dict[str, str] = {
+            # 🔥 关键标识：所有请求都需要这个 header
+            "X-Product-Code": "codebuddy",
+            # 🔥 IDE 识别信息（服务端用于识别客户端类型）
+            "X-IDE-Type": "vscode",
+            "X-IDE-Name": "Visual Studio Code",
+            "X-IDE-Version": "1.70.2",  # VSCode 最低版本要求
+            "X-Product-Version": "4.10.33259736",  # 插件版本
+            "X-Machine-Id": self._get_machine_id(),
+        }
+        
         if account.get("uid"):
             headers["X-User-Id"] = str(account["uid"])
         if access and auth.get("accessToken"):
@@ -122,12 +136,41 @@ class CodeBuddyClient:
         if account.get("enterpriseId"):
             headers["X-Enterprise-Id"] = str(account["enterpriseId"])
             headers["X-Tenant-Id"] = str(account["enterpriseId"])
+        if account.get("departmentInfo"):
+            # 🔥 新增：部门信息（如果有）
+            headers["X-Department-Info"] = str(account["departmentInfo"])
         if auth.get("domain"):
             # The extension calls this the domain header.  The server accepts
             # X-Domain for the plugin protocol.
             headers["X-Domain"] = str(auth["domain"])
         return headers
 
+    def _get_machine_id(self) -> str:
+        """获取或生成机器ID（模拟VSCode的machineId）"""
+        import hashlib
+        import platform
+        import uuid
+        
+        # 尝试从session读取已保存的machineId
+        saved_machine_id = self.session.get("machineId")
+        if saved_machine_id:
+            return str(saved_machine_id)
+        
+        # 生成一个稳定的machineId（基于主机名和用户）
+        try:
+            hostname = platform.node()
+            username = os.environ.get("USER") or os.environ.get("USERNAME") or "unknown"
+            seed = f"{hostname}-{username}"
+            machine_id = str(uuid.uuid5(uuid.NAMESPACE_DNS, seed))
+        except Exception:
+            # 如果失败，生成随机ID
+            machine_id = str(uuid.uuid4())
+        
+        # 保存到session
+        self.session["machineId"] = machine_id
+        self._save_session(self.session)
+        return machine_id
+
     def login(self, *, open_browser: bool = True, timeout: int = 300) -> None:
         no_auth = {
             "X-No-Authorization": "true",
@@ -250,6 +293,8 @@ class CodeBuddyClient:
             f"{self.endpoint}/v2/chat/completions",
             data=json.dumps(payload).encode("utf-8"),
             headers={
+                "User-Agent": "Mozilla/5.0 (compatible; Genie-IDE/1.0)",
+                "X-Product-Code": "codebuddy",
                 **self.auth_headers(),
                 "Content-Type": "application/json",
                 "Accept": "text/event-stream",

{workbuddy2api-2.0.2 → workbuddy2api-2.0.3}/src/workbuddy2api.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: workbuddy2api
-Version: 2.0.2
+Version: 2.0.3
 Summary: Local OpenAI/Responses/Anthropic compatible proxy for CodeBuddy
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
@@ -72,8 +72,10 @@ uv run python -m codebuddy_proxy --login --desensitize
 
 ```bash
 # 使用最新版本（推荐）
-uv run --with workbuddy2api python -m codebuddy_proxy --desensitize \
-  --log-file "$HOME/.workbuddy2api/codebuddy-proxy.jsonl"
+uv run --with workbuddy2api python -m codebuddy_proxy --desensitize
+
+# 首次使用：登录并启动
+uv run --with workbuddy2api python -m codebuddy_proxy --login --desensitize
 ```
 
 默认监听 `http://127.0.0.1:8787`