Remove unnecessary LLM settings

This commit is contained in:
Weves 2024-08-12 11:31:14 -07:00 committed by Chris Weaver
parent 0a165aae0b
commit 1a4df1d65e

View File

@ -187,11 +187,6 @@ class DefaultMultiLLM(LLM):
"""Uses Litellm library to allow easy configuration to use a multitude of LLMs
See https://python.langchain.com/docs/integrations/chat/litellm"""
DEFAULT_MODEL_PARAMS: dict[str, Any] = {
"frequency_penalty": 0,
"presence_penalty": 0,
}
def __init__(
self,
api_key: str | None,
@ -224,9 +219,7 @@ class DefaultMultiLLM(LLM):
for k, v in custom_config.items():
os.environ[k] = v
model_kwargs = (
DefaultMultiLLM.DEFAULT_MODEL_PARAMS if model_provider == "openai" else {}
)
model_kwargs: dict[str, Any] = {}
if extra_headers:
model_kwargs.update({"extra_headers": extra_headers})