From 2c99afaab00e3dad246385b1179b749dd25fc147 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=9D=92=E8=BD=A9?= Date: Tue, 27 Aug 2024 16:33:39 +0800 Subject: [PATCH] raise error when openai key is missing --- memoryscope/core/models/base_model.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/memoryscope/core/models/base_model.py b/memoryscope/core/models/base_model.py index fa27da2a..05483fee 100644 --- a/memoryscope/core/models/base_model.py +++ b/memoryscope/core/models/base_model.py @@ -1,5 +1,6 @@ import inspect import time +import os from abc import abstractmethod, ABCMeta from typing import Any @@ -46,6 +47,9 @@ class BaseModel(metaclass=ABCMeta): if self.module_name not in MODEL_REGISTRY.module_dict: raise RuntimeError(f"method_type={self.module_name} is not supported!") obj_cls = MODEL_REGISTRY[self.module_name] + if 'openai' in self.module_name: + if os.environ.get('OPENAI_API_KEY', None) is None: + raise ValueError("Missing openai api key!") if self.kwargs_filter: allowed_kwargs = list(inspect.signature(obj_cls.__init__).parameters.keys())