Class LobeCloudflareAI

packages/model-runtime/src/providers/cloudflare/index.ts:55–205

}

export class LobeCloudflareAI implements LobeRuntimeAI {
  baseURL: string;
  accountID: string;
  apiKey?: string;

  constructor({ apiKey, baseURLOrAccountID }: LobeCloudflareParams = {}) {
    if (!baseURLOrAccountID) {
      throw AgentRuntimeError.createError(AgentRuntimeErrorType.InvalidProviderAPIKey);
    }
    if (baseURLOrAccountID.startsWith('http')) {
      this.baseURL = baseURLOrAccountID.endsWith('/')
        ? baseURLOrAccountID
        : baseURLOrAccountID + '/';
      // Try get accountID from baseURL
      this.accountID = baseURLOrAccountID.replaceAll(/^.*\/([\da-f]{32})\/.*$/gi, '$1');
    } else {
      if (!apiKey) {
        throw AgentRuntimeError.createError(AgentRuntimeErrorType.InvalidProviderAPIKey);
      }
      this.accountID = baseURLOrAccountID;
      this.baseURL = fillUrl(baseURLOrAccountID);
    }
    this.apiKey = apiKey;
  }

  async chat(payload: ChatStreamPayload, options?: ChatMethodOptions): Promise<Response> {
    // Remove internal apiMode parameter to prevent sending to Cloudflare API
    const { model, tools, apiMode: _, ...restPayload } = payload;
    const functions = tools?.map((tool) => tool.function);
    const headers = options?.headers || {};
    if (this.apiKey) {
      headers['Authorization'] = `Bearer ${this.apiKey}`;
    }
    const url = new URL(model, this.baseURL);
    const desensitizedEndpoint = desensitizeCloudflareUrl(url.toString());

    let response: Response;
    try {
      response = await fetch(url, {
        body: JSON.stringify({ tools: functions, ...restPayload }),
        headers: { 'Content-Type': 'application/json', ...headers },
        method: 'POST',
        signal: options?.signal,
      });
    } catch (error) {
      throw AgentRuntimeError.chat({
        endpoint: desensitizeCloudflareUrl(this.baseURL),
        error: error as any,
        errorType: AgentRuntimeErrorType.ProviderBizError,
        message: extractProviderErrorMessage(error) ?? 'Cloudflare API request failed',
        provider: ModelProvider.Cloudflare,
      });
    }

    if (response.status === 400) {
      const bodyText = await response.text().catch(() => '');
      let parsedBody: unknown = bodyText;
      if (bodyText) {