scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-10-07 03:00:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

raise error when openai key is missing

Browse source
This commit is contained in:
青轩 2024-08-27 16:33:39 +08:00
parent b5e13b90e7
commit 2c99afaab0
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
memoryscope/core/models/base_model.py
View file

@ -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())