diff --git a/Achievement_Inputing/settings.py b/Achievement_Inputing/settings.py index 6a17feb..d60ffd7 100644 --- a/Achievement_Inputing/settings.py +++ b/Achievement_Inputing/settings.py @@ -166,3 +166,4 @@ ELASTICSEARCH_INDEX_NAMES = { # AI Studio/OpenAI client settings AISTUDIO_API_KEY = os.environ.get('AISTUDIO_API_KEY', '') OPENAI_BASE_URL = os.environ.get('OPENAI_BASE_URL', 'https://aistudio.baidu.com/llm/lmapi/v3') +OPENAI_MODEL_NAME = os.environ.get('OPENAI_MODEL_NAME', 'ernie-4.5-turbo-vl-32k') diff --git a/elastic/views.py b/elastic/views.py index cfb06af..871687f 100644 --- a/elastic/views.py +++ b/elastic/views.py @@ -357,7 +357,7 @@ def ocr_and_extract_info(image_path: str): ], }, ], - model="ernie-4.5-turbo-vl-32k", + model=getattr(settings, "OPENAI_MODEL_NAME", "ernie-4.5-turbo-vl-32k"), ) response_text = chat_completion.choices[0].message.content