fix: send x-opencode-session on opencode gateway requests

Author: elkaix

.changeset/opencode-session-header.md

@@ -0,0 +1,5 @@
+---
+"@pymodel/pythinker-code": patch
+---
+
+Send OpenCode Go requests with a per-conversation session header.

packages/agent-core-v2/src/agent/profile/profileService.ts

@@ -464,6 +464,7 @@ export class AgentProfileService extends Disposable implements IAgentProfileServ
     };
     return {
       cacheKey: this.sessionContext.sessionId,
+      conversationId: this.sessionContext.sessionId,
       sampling:
         sampling.temperature === undefined && sampling.topP === undefined ? undefined : sampling,
       thinkingEffort: thinking.effective,

packages/agent-core-v2/src/index.ts

@@ -199,6 +199,7 @@ export * from '#/kosong/model/thinking';
 export * from '#/kosong/model/catalog';
 export * from '#/kosong/model/catalogService';
 export * from '#/kosong/model/modelRequester';
+export * from '#/kosong/model/opencodeSession';
 import '#/kosong/model/errors';
 export {
   MODEL_CATALOG_SECTION,

packages/agent-core-v2/src/kosong/contract/provider.ts

@@ -73,6 +73,7 @@ export interface VideoUploadInput {
 export interface GenerateOptions {
   signal?: AbortSignal;
   auth?: ProviderRequestAuth;
+  extraHeaders?: Readonly<Record<string, string>>;
   responseFormat?: ResponseFormat;
   cacheKey?: string;
   sampling?: SamplingOptions;

packages/agent-core-v2/src/kosong/model/catalogService.ts

@@ -1,4 +1,5 @@
 import { parsePythinkerCodeCustomHeaders } from '@pymodel/pythinker-code-oauth';
+import { randomUUID } from 'node:crypto';
 
 import { Disposable } from '#/_base/di/lifecycle';
 import { LifecycleScope } from '#/app/scopes';
@@ -149,7 +150,7 @@ export class ModelCatalog extends Disposable implements IModelCatalog {
           messages: [{ role: 'user', content: [{ type: 'text', text: 'ping' }], toolCalls: [] }],
         },
         undefined,
-        { maxCompletionTokens: 512 },
+        { conversationId: randomUUID(), maxCompletionTokens: 512 },
       )) {
         if (event.type === 'part' && event.part.type === 'text') {
           text += event.part.text;

packages/agent-core-v2/src/kosong/model/modelRequester.ts

@@ -42,6 +42,7 @@ export type ModelRequestEvent =
 
 export interface ModelRequestParams {
   readonly cacheKey?: string;
+  readonly conversationId?: string;
   readonly sampling?: SamplingOptions;
   readonly thinkingEffort?: ThinkingEffort;
   readonly thinkingKeep?: string;

packages/agent-core-v2/src/kosong/model/modelRequesterImpl.ts

@@ -12,6 +12,8 @@ import type {
 import { translateProviderError } from '#/kosong/protocol/errors';
 import type { IProtocolAdapterRegistry } from '#/kosong/protocol/protocol';
 
+import { opencodeSessionHeaders } from './opencodeSession';
+
 import type { AuthProvider, Model } from './catalog';
 import type {
   ModelRequestEvent,
@@ -90,6 +92,7 @@ export class ModelRequesterImpl implements ModelRequester {
     const options: GenerateOptions = {
       signal,
       cacheKey: params?.cacheKey,
+      extraHeaders: opencodeSessionHeaders(this.model.baseUrl, params?.conversationId),
       sampling: params?.sampling,
       thinking:
         params?.thinkingEffort === undefined

packages/agent-core-v2/src/kosong/model/opencodeSession.ts

@@ -0,0 +1,23 @@
+const OPENCODE_HOST = 'opencode.ai';
+
+export const OPENCODE_SESSION_HEADER = 'x-opencode-session';
+
+export function isOpencodeGatewayBaseUrl(baseUrl: string | undefined): boolean {
+  if (baseUrl === undefined || baseUrl.length === 0) return false;
+  let host: string;
+  try {
+    host = new URL(baseUrl).hostname.toLowerCase();
+  } catch {
+    return false;
+  }
+  return host === OPENCODE_HOST || host.endsWith(`.${OPENCODE_HOST}`);
+}
+
+export function opencodeSessionHeaders(
+  baseUrl: string | undefined,
+  conversationId: string | undefined,
+): Record<string, string> | undefined {
+  const id = conversationId?.trim();
+  if (!isOpencodeGatewayBaseUrl(baseUrl) || id === undefined || id.length === 0) return undefined;
+  return { [OPENCODE_SESSION_HEADER]: id };
+}

packages/agent-core-v2/src/kosong/provider/bases/anthropic/anthropic.ts

@@ -979,7 +979,10 @@ export class AnthropicChatProvider implements ChatProvider {
     }
 
     const requestOptions: Record<string, unknown> = {};
-    const headers = mergeRequestHeaders(extraHeaders, options?.auth?.headers);
+    const headers = mergeRequestHeaders(
+      mergeRequestHeaders(extraHeaders, options?.extraHeaders),
+      options?.auth?.headers,
+    );
     if (headers !== undefined) {
       requestOptions['headers'] = headers;
     }

packages/agent-core-v2/src/kosong/provider/bases/google-genai/google-genai.ts

@@ -21,7 +21,11 @@ import type { Tool } from '#/kosong/contract/tool';
 import type { TokenUsage } from '#/kosong/contract/usage';
 
 import { mergeConsecutiveUserMessages } from '../merge-user-messages';
-import { requireProviderApiKey, resolveAuthBackedClient } from '../request-auth';
+import {
+  mergeRequestHeaders,
+  requireProviderApiKey,
+  resolveAuthBackedClient,
+} from '../request-auth';
 
 function normalizeGoogleGenAIFinishReason(raw: unknown): {
   finishReason: FinishReason | null;
@@ -783,6 +787,11 @@ export class GoogleGenAIChatProvider implements ChatProvider {
       ...(tools.length > 0 ? { tools: tools.map((t) => toolToGoogleGenAI(t)) } : {}),
     };
     applyResponseFormat(config, options?.responseFormat);
+    if (options?.extraHeaders !== undefined) {
+      config['httpOptions'] = {
+        headers: mergeRequestHeaders(this._defaultHeaders, options.extraHeaders),
+      };
+    }
 
     try {
       const client = this._createClient(options?.auth);