Browse Source
Change the api_key priority
Signed-off-by: Kaiyuan Hu <kaiyuan.hu@zilliz.com>
main
1 changed files with
1 additions and
1 deletions
-
openai_chat.py
|
|
@ -26,7 +26,7 @@ class OpenaiChat(PyOperator): |
|
|
api_key: str = None, |
|
|
api_key: str = None, |
|
|
**kwargs |
|
|
**kwargs |
|
|
): |
|
|
): |
|
|
openai.api_key = os.getenv('OPENAI_API_KEY', api_key) |
|
|
|
|
|
|
|
|
openai.api_key = api_key or os.getenv('OPENAI_API_KEY') |
|
|
self._model = model_name |
|
|
self._model = model_name |
|
|
self.stream = kwargs.pop('stream') if 'stream' in kwargs else False |
|
|
self.stream = kwargs.pop('stream') if 'stream' in kwargs else False |
|
|
self.kwargs = kwargs |
|
|
self.kwargs = kwargs |
|
|
|
