add env OPENAI_LLM_MODEL

2024-12-06 18:00:20 +08:00
parent 4784bd53a2
commit 890da4f4ac
2 changed files with 4 additions and 2 deletions
--- a/whisper_fastapi.py
+++ b/whisper_fastapi.py
@@ -81,7 +81,7 @@ async def gpt_refine_text(
            os.environ.get("OPENAI_BASE_URL", "https://api.openai.com/v1")
            + "/chat/completions",
            json={
-                "model": "gpt-4o-mini",
+                "model": os.environ.get("OPENAI_LLM_MODEL", "gpt-4o-mini"),
                "temperature": 0.1,
                "stream": False,
                "messages": [