Merge pull request #194 from flyi/config-api-key

fix exception when using Anthropic_API_KEY rather than OPENAI_API_KEY
This commit is contained in:
stellaHSR 2023-08-17 12:04:07 +08:00 committed by GitHub
commit 625342199a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -44,8 +44,10 @@ class Config(metaclass=Singleton):
logger.info("Config loading done.")
self.global_proxy = self._get("GLOBAL_PROXY")
self.openai_api_key = self._get("OPENAI_API_KEY")
if not self.openai_api_key or "YOUR_API_KEY" == self.openai_api_key:
raise NotConfiguredException("Set OPENAI_API_KEY first")
self.anthropic_api_key = self._get("Anthropic_API_KEY")
if (not self.openai_api_key or "YOUR_API_KEY" == self.openai_api_key) \
and (not self.anthropic_api_key or "YOUR_API_KEY" == self.anthropic_api_key):
raise NotConfiguredException("Set OPENAI_API_KEY or Anthropic_API_KEY first")
self.openai_api_base = self._get("OPENAI_API_BASE")
if not self.openai_api_base or "YOUR_API_BASE" == self.openai_api_base:
openai_proxy = self._get("OPENAI_PROXY") or self.global_proxy