Skip to content

fix(llmcore): support opencode.ai Zen/Go endpoints (x-opencode-session header) #831

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
tuxdio wants to merge 1 commit into
base: main
Choose a base branch
from
+14 −1
Open

fix(llmcore): support opencode.ai Zen/Go endpoints (x-opencode-session header) #831

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 14 additions & 1 deletion llmcore.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,11 @@ def _conn_request_hook(self, *a, **k): # after request() the socket is connecte
r = _orig_conn_request(self, *a, **k); _INFLIGHT[threading.get_ident()] = self.sock; return r
urllib3.connection.HTTPConnection.request = _conn_request_hook
_RESP_CACHE_KEY = str(uuid.uuid4()); _RESP_CODEX_KEY = str(uuid.uuid4())
_OC_SIDS = {}
def _oc_session_id(name):
"""OpenCode Go 要求每个会话上报稳定的 x-opencode-session，进程内按模型名复用同一 ID。"""
if name not in _OC_SIDS: _OC_SIDS[name] = str(uuid.uuid4())
return _OC_SIDS[name]
_ROOT = os.path.dirname(os.path.abspath(__file__))
if _ROOT not in sys.path: sys.path.append(_ROOT)

Expand Down Expand Up @@ -520,6 +525,7 @@ def _openai_stream(sess, messages):
elif 'minimax' in ml: temperature = max(0.01, min(temperature, 1.0)) # MiniMax requires temp in (0, 1]
headers = {"Authorization": f"Bearer {sess.api_key}", "Content-Type": "application/json", "Accept": "text/event-stream", 'originator': 'codex_exec'}
headers["User-Agent"] = sess.user_agent
headers.update(getattr(sess, 'extra_headers', None) or {})
if api_mode == "responses":
url = auto_make_url(sess.api_base, "responses")
payload = {"model": model, "input": _to_responses_input(messages), "stream": sess.stream,
Expand Down Expand Up @@ -676,6 +682,12 @@ def _enum(key, valid):
self.max_tokens = cfg.get('max_tokens')
self.default_ua = "claude-cli/2.1.280 (external, cli)"
self.user_agent = cfg.get("user_agent", self.default_ua)
# ── OpenCode Go (opencode.ai)：必须按会话上报 x-opencode-session，否则 400 MissingSessionID ──
self.extra_headers = dict(cfg.get('extra_headers') or {})
if 'opencode.ai' in self.api_base:
self.session_id = str(cfg.get('session_id') or _oc_session_id(self.name))
self.extra_headers.setdefault('x-opencode-session', self.session_id)
if self.user_agent == self.default_ua: self.user_agent = 'GenericAgent/1.0' # 官方要求客户端自报 UA
def _apply_claude_thinking(self, payload):
if self.thinking_type:
thinking = {"type": self.thinking_type}
Expand Down Expand Up @@ -730,6 +742,7 @@ def raw_ask(self, messages):
messages = _fix_messages(messages)
if self.max_tokens is None: self.max_tokens = 8192
headers = {"x-api-key": self.api_key, "Content-Type": "application/json", "anthropic-version": "2023-06-01", "anthropic-beta": "prompt-caching-2024-07-31"}
headers.update(self.extra_headers)
payload = {"model": self.model, "messages": messages, "max_tokens": self.max_tokens, "stream": self.stream}
if self.temperature != 1: payload["temperature"] = self.temperature
self._apply_claude_thinking(payload)
Expand Down Expand Up @@ -811,7 +824,7 @@ def raw_ask(self, messages):
headers.update({"Accept": "application/json", "X-Claude-Code-Session-Id": self._session_id, "X-Stainless-Arch": "x64", "X-Stainless-Lang": "js", "X-Stainless-OS": "Windows", "X-Stainless-Package-Version": "0.94.0", "X-Stainless-Retry-Count": "0", "X-Stainless-Runtime": "node", "X-Stainless-Runtime-Version": "v24.3.0", "X-Stainless-Timeout": "600"})
if self.api_key_header == 'x-api-key': headers["x-api-key"] = self.api_key
elif self.api_key_header == 'bearer': headers["authorization"] = f"Bearer {self.api_key}"
elif self.api_key.startswith("sk-ant-"): headers["x-api-key"] = self.api_key
elif self.api_key.startswith("sk-ant-") or 'opencode.ai' in self.api_base: headers["x-api-key"] = self.api_key
else: headers["authorization"] = f"Bearer {self.api_key}"
payload = {"model": model, "messages": messages, "max_tokens": self.max_tokens, "stream": self.stream}
#if self.fake_cc_system_prompt: payload["max_tokens"] = 64000
Expand Down