Method map

docetl/frame.py:334–390

(
        self,
        name: str | None = None,
        *,
        prompt: str | None = None,
        output: dict[str, Any] | None = None,
        model: str | None = None,
        optimize: bool | None = None,
        recursively_optimize: bool | None = None,
        sample: int | None = None,
        agent: Any | None = None,
        validate: list[str] | None = None,
        num_retries_on_validate_failure: int | None = None,
        drop_keys: list[str] | None = None,
        timeout: int | None = None,
        enable_observability: bool | None = None,
        max_batch_size: int | None = None,
        clustering_method: str | None = None,
        batch_prompt: str | None = None,
        litellm_completion_kwargs: dict[str, Any] | None = None,
        pdf_url_key: str | None = None,
        flush_partial_results: bool | None = None,
        limit: int | None = None,
        calibrate: bool | None = None,
        num_calibration_docs: int | None = None,
        retriever: Retriever | str | None = None,
        **kwargs: Any,
    )

    # ── LLM operations ─────────────────────────────────────────────

    def map(
        self,
        name: str | None = None,
        *,
        prompt: str | None = None,
        output: dict[str, Any] | None = None,
        model: str | None = None,
        optimize: bool | None = None,
        recursively_optimize: bool | None = None,
        sample: int | None = None,
        agent: Any | None = None,
        validate: list[str] | None = None,
        num_retries_on_validate_failure: int | None = None,
        drop_keys: list[str] | None = None,
        timeout: int | None = None,
        enable_observability: bool | None = None,
        max_batch_size: int | None = None,
        clustering_method: str | None = None,
        batch_prompt: str | None = None,
        litellm_completion_kwargs: dict[str, Any] | None = None,
        pdf_url_key: str | None = None,
        flush_partial_results: bool | None = None,
        limit: int | None = None,
        calibrate: bool | None = None,
        num_calibration_docs: int | None = None,
        retriever: Retriever | str | None = None,
        **kwargs: Any,
    ) -> Frame:
        return self._append_op(
            "map",
            name,
            {
                "prompt": prompt,
                "output": output,
                "model": model,
                "optimize": optimize,
                "recursively_optimize": recursively_optimize,
                "sample": sample,
                "agent": agent,
                "validate": validate,
                "num_retries_on_validate_failure": num_retries_on_validate_failure,
                "drop_keys": drop_keys,
                "timeout": timeout,
                "enable_observability": enable_observability,
                "max_batch_size": max_batch_size,
                "clustering_method": clustering_method,
                "batch_prompt": batch_prompt,
                "litellm_completion_kwargs": litellm_completion_kwargs,
                "pdf_url_key": pdf_url_key,
                "flush_partial_results": flush_partial_results,
                "limit": limit,
                "calibrate": calibrate,
                "num_calibration_docs": num_calibration_docs,
                "retriever": retriever,
                **kwargs,
            },
        )