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

add config litellm.drop_params

Browse Source
This commit is contained in:
koid
2024-03-13 11:20:02 +09:00
parent d62796ac68
commit 1ed2cd064a
2 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -36,6 +36,8 @@ class LiteLLMAIHandler(BaseAiHandler):
assert litellm_token, "LITELLM_TOKEN is required" assert litellm_token, "LITELLM_TOKEN is required"
os.environ["LITELLM_TOKEN"] = litellm_token os.environ["LITELLM_TOKEN"] = litellm_token
litellm.use_client = True litellm.use_client = True
if get_settings().get("LITELLM.DROP_PARAMS", None):
litellm.drop_params = get_settings().litellm.drop_params
if get_settings().get("OPENAI.ORG", None): if get_settings().get("OPENAI.ORG", None):
litellm.organization = get_settings().openai.org litellm.organization = get_settings().openai.org
if get_settings().get("OPENAI.API_TYPE", None): if get_settings().get("OPENAI.API_TYPE", None):

3
pr_agent/settings/configuration.toml
View File

@ -192,7 +192,8 @@ pr_commands = [
url = "" url = ""
[litellm] [litellm]
#use_client = false # use_client = false
# drop_params = false
[pr_similar_issue] [pr_similar_issue]
skip_comments = false skip_comments = false