ti.sil/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-12-15 12:25:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

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
Show stats 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