kunthawat/ALwrity
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Alwrity - Bug fixes

Browse Source
This commit is contained in:
AjaySi
2024-04-12 18:56:20 +05:30
parent bdab2e9959
commit 44c50e9ecd
2 changed files with 6 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
lib/gpt_providers/text_generation/main_text_generation.py
View File

@@ -69,17 +69,11 @@ def check_gpt_provider(gpt_provider):
ValueError: If both the specified GPT provider and environment variable GPT_PROVIDER are missing.
"""
env_gpt_provider = os.getenv('GPT_PROVIDER')
if gpt_provider:
os.environ['GPT_PROVIDER'] = gpt_provider
elif env_gpt_provider:
gpt_provider = env_gpt_provider
else:
raise ValueError("Both specified GPT provider and environment variable 'GPT_PROVIDER' are missing.")
if gpt_provider != env_gpt_provider:
if gpt_provider and if gpt_provider != env_gpt_provider:
logger.warning(f"Config: '{gpt_provider}' different to environment variable 'GPT_PROVIDER' '{env_gpt_provider}'")
logger.info(f"Using GPT provider: {gpt_provider}")
logger.info(f"Using Environment GPT provider: {env_gpt_provider}")
gpt_provider = env_gpt_provider
return gpt_provider