free-claude-code/providers/zai/request.py
Siarhei Krotau 972bc1661c
Some checks are pending
CI / checks (push) Waiting to run
feat(providers): add Z.ai Coding Plan provider (#440)
2026-05-13 12:02:15 -07:00

33 lines
1.1 KiB
Python

"""Request builder for Z.ai OpenAI-compatible Coding Plan API."""
from typing import Any
from loguru import logger
from core.anthropic import ReasoningReplayMode, build_base_request_body
from core.anthropic.conversion import OpenAIConversionError
from providers.exceptions import InvalidRequestError
def build_request_body(request_data: Any, *, thinking_enabled: bool) -> dict:
"""Build OpenAI-format request body from Anthropic request."""
logger.debug(
"ZAI_REQUEST: conversion start model={} msgs={}",
getattr(request_data, "model", "?"),
len(getattr(request_data, "messages", [])),
)
try:
body = build_base_request_body(
request_data,
reasoning_replay=ReasoningReplayMode.REASONING_CONTENT,
)
except OpenAIConversionError as exc:
raise InvalidRequestError(str(exc)) from exc
logger.debug(
"ZAI_REQUEST: conversion done model={} msgs={} tools={}",
body.get("model"),
len(body.get("messages", [])),
len(body.get("tools", [])),
)
return body