#!/usr/bin/env bash
set -euo pipefail

ROOT=/root/projects/website-builder-backend-clean
cd "$ROOT"

python3 <<'PY'
from pathlib import Path

p=Path("backend/llm/providers/openai_compatible.py")
src=p.read_text()

if "async def generate(" not in src:

    marker="    async def complete("

    wrapper='''
    async def generate(
        self,
        model: str,
        prompt: str,
        **kwargs,
    ):
        return await self.complete(
            model=model,
            prompt=prompt,
            **kwargs,
        )

'''

    pos=src.find(marker)

    if pos==-1:
        raise RuntimeError("complete() not found")

    src=src[:pos]+wrapper+src[pos:]

    p.write_text(src)

print("patched")
PY

python3 -m py_compile \
backend/llm/providers/openai_compatible.py \
backend/llm/providers/huggingface.py

python3 <<'PY'
from backend.llm.providers.huggingface import HuggingFaceProvider

provider=HuggingFaceProvider()

print(type(provider).__name__)
print(hasattr(provider,"generate"))
print(hasattr(provider,"complete"))
PY

echo
echo "========================================"
echo "OpenAICompatibleProvider repaired"
echo "========================================"