feat(sdk): support max output tokens

This commit is contained in:
Yichen Jiang
2026-07-28 17:36:44 +08:00
parent f63d2deecf
commit 5358168787
55 changed files with 336 additions and 90 deletions
+2
View File
@@ -20,6 +20,7 @@ class DeepSeekHarnessConfig:
provider: str = "deepseek"
model: str = "deepseek-v4-flash"
max_tokens: int | None = None
cwd: str | None = None
runtime_cwd: str | None = None
session_root: str | None = None
@@ -100,6 +101,7 @@ class DeepSeekHarness:
cwd=self._cwd,
provider=self.config.provider,
model=self.config.model,
max_tokens=self.config.max_tokens,
)
self._initialized = True
@@ -120,12 +120,15 @@ class HarnessClient:
cwd: str,
provider: str,
model: str,
max_tokens: int | None = None,
) -> InitializeResponse:
payload: JsonObject = {
"cwd": str(Path(cwd).resolve()),
"provider": provider,
"model": model,
}
if max_tokens is not None:
payload["maxTokens"] = max_tokens
try:
return self.request("initialize", payload, response_model=InitializeResponse)
except BaseException: