xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-04 12:50:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update pr_agent/algo/ai_handlers/litellm_ai_handler.py

Browse Source 
Co-authored-by: codiumai-pr-agent-pro[bot] <151058649+codiumai-pr-agent-pro[bot]@users.noreply.github.com>
This commit is contained in:
Tal
2024-07-04 12:26:23 +03:00
committed by GitHub
parent f3c80891f8
commit 20d9d8ad07
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
pr_agent/algo/ai_handlers/litellm_ai_handler.py
View File

@ -33,6 +33,7 @@ class LiteLLMAIHandler(BaseAiHandler):
elif 'OPENAI_API_KEY' not in os.environ: elif 'OPENAI_API_KEY' not in os.environ:
litellm.api_key = "dummy_key" litellm.api_key = "dummy_key"
if get_settings().get("aws.AWS_ACCESS_KEY_ID"): if get_settings().get("aws.AWS_ACCESS_KEY_ID"):
assert get_settings().aws.AWS_SECRET_ACCESS_KEY and get_settings().aws.AWS_REGION_NAME, "AWS credentials are incomplete"
os.environ["AWS_ACCESS_KEY_ID"] = get_settings().aws.AWS_ACCESS_KEY_ID os.environ["AWS_ACCESS_KEY_ID"] = get_settings().aws.AWS_ACCESS_KEY_ID
os.environ["AWS_SECRET_ACCESS_KEY"] = get_settings().aws.AWS_SECRET_ACCESS_KEY os.environ["AWS_SECRET_ACCESS_KEY"] = get_settings().aws.AWS_SECRET_ACCESS_KEY
os.environ["AWS_REGION_NAME"] = get_settings().aws.AWS_REGION_NAME os.environ["AWS_REGION_NAME"] = get_settings().aws.AWS_REGION_NAME