Function main

examples/openai_api/smoke_test.py:83–109

()

def main() -> int:
    parser = argparse.ArgumentParser(description="Smoke test the FunASR OpenAI-compatible API")
    parser.add_argument("audio_path", nargs="?", default="sample.wav", help="Audio file to transcribe; downloads a sample if missing")
    parser.add_argument("--base-url", default=os.getenv("BASE_URL", "http://localhost:8000"), help="FunASR API base URL")
    parser.add_argument("--model", default=os.getenv("MODEL", "sensevoice"), help="Model alias to use")
    parser.add_argument("--response-format", default=os.getenv("RESPONSE_FORMAT", "verbose_json"), choices=["json", "verbose_json"], help="Transcription response format")
    parser.add_argument("--sample-url", default=os.getenv("SAMPLE_URL", SAMPLE_URL), help="Sample audio URL used when audio_path does not exist")
    parser.add_argument("--timeout", type=float, default=float(os.getenv("TIMEOUT", "300")), help="HTTP timeout in seconds")
    args = parser.parse_args()

    base_url = args.base_url.rstrip("/")
    audio_path = Path(args.audio_path)

    try:
        download_if_needed(audio_path, args.sample_url, args.timeout)
        print_json("health", request_json(f"{base_url}/health", args.timeout))
        print_json("models", request_json(f"{base_url}/v1/models", args.timeout))
        print(f"\nTranscribing {audio_path} with model={args.model}, response_format={args.response_format}")
        print_json("transcription", transcribe(base_url, audio_path, args.model, args.response_format, args.timeout))
    except urllib.error.HTTPError as error:
        detail = error.read().decode("utf-8", errors="replace")
        print(f"HTTP {error.code} from {error.url}: {detail}", file=sys.stderr)
        return 1
    except Exception as error:
        print(f"Smoke test failed: {error}", file=sys.stderr)
        return 1
    return 0


if __name__ == "__main__":