// Meta plugin module implements stream behavior.
import type { StreamFn } from "openclaw/plugin-sdk/agent-core";
import type { ProviderWrapStreamFnContext } from "openclaw/plugin-sdk/plugin-entry";
import { createPayloadPatchStreamWrapper } from "openclaw/plugin-sdk/provider-stream-shared";
import { filterStringEntries } from "openclaw/plugin-sdk/string-coerce-runtime";

const META_REASONING_ENCRYPTED_CONTENT_INCLUDE = "reasoning.encrypted_content";

export function wrapMetaProviderStream(ctx: ProviderWrapStreamFnContext): StreamFn | undefined {
  if (ctx.provider !== "meta" || (ctx.sourceApi ?? ctx.model?.api) !== "openai-responses") {
    return undefined;
  }
  return createPayloadPatchStreamWrapper(ctx.streamFn, ({ payload, model, options }) => {
    if (model.provider !== "meta") {
      return;
    }
    // Responses treats zero as an unset caller cap. Restore the catalog limit
    // without changing provider-selected behavior when the caller omits the field.
    if (options?.maxTokens === 0 && payload.max_output_tokens === undefined) {
      payload.max_output_tokens = model.maxTokens;
    }
    if (!model.reasoning) {
      return;
    }
    const include = filterStringEntries(payload.include);
    if (!include.includes(META_REASONING_ENCRYPTED_CONTENT_INCLUDE)) {
      include.push(META_REASONING_ENCRYPTED_CONTENT_INCLUDE);
    }
    payload.include = include;
    payload.store = false;
  });
}