feat: add DeepSeek extreme reasoning tier

This commit is contained in:
wuyang
2026-07-26 17:40:51 +08:00
parent ade5ea1210
commit 2e8e81c790
25 changed files with 567 additions and 30 deletions
+19 -2
View File
@@ -599,6 +599,10 @@ class AgentLoop:
model=spec.provider_model,
messages=messages,
tools=available_specs,
provider=spec.provider,
thinking_enabled=spec.thinking_enabled,
reasoning_effort=spec.reasoning_effort,
max_output_tokens=spec.max_output_tokens,
)
choice = response["choices"][0]
message = choice.get("message") or {}
@@ -648,9 +652,12 @@ class AgentLoop:
"completion.rejected",
{"reason": failure, "iteration": iteration + 1, "depth": depth},
)
assistant_checkpoint = {"role": "assistant", "content": candidate}
if isinstance(message.get("reasoning_content"), str):
assistant_checkpoint["reasoning_content"] = message["reasoning_content"]
messages.extend(
[
{"role": "assistant", "content": candidate},
assistant_checkpoint,
{"role": "user", "content": recovery_message(failure)},
]
)
@@ -683,6 +690,8 @@ class AgentLoop:
"content": message.get("content"),
"tool_calls": tool_calls,
}
if isinstance(message.get("reasoning_content"), str):
assistant_message["reasoning_content"] = message["reasoning_content"]
messages.append(assistant_message)
blocked_call_reasons: dict[str, str] = {}
batch_mutation_epoch = 0
@@ -801,7 +810,15 @@ class AgentLoop:
),
}
)
response = await self.provider.complete(model=spec.provider_model, messages=messages, tools=None)
response = await self.provider.complete(
model=spec.provider_model,
messages=messages,
tools=None,
provider=spec.provider,
thinking_enabled=spec.thinking_enabled,
reasoning_effort=spec.reasoning_effort,
max_output_tokens=spec.max_output_tokens,
)
answer = str(response["choices"][0].get("message", {}).get("content") or "").strip()
if completion_failure(
artifact_required,