phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix linting errors @krrishdholakia

Browse source
This commit is contained in:
ishaan-jaff 2023-09-06 16:38:42 -07:00
parent c5151aa573
commit 04f8b20651
2 changed files with 12 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/llms/huggingface_restapi.py
View file

@ -52,7 +52,12 @@ def completion(
if model in custom_prompt_dict: if model in custom_prompt_dict:
# check if the model has a registered custom prompt # check if the model has a registered custom prompt
model_prompt_details = custom_prompt_dict[model] model_prompt_details = custom_prompt_dict[model]
prompt = custom_prompt(role_dict=model_prompt_details["roles"], pre_message_sep=model_prompt_details["pre_message_sep"], post_message_sep=model_prompt_details["post_message_sep"], messages=messages) prompt = custom_prompt(
role_dict=model_prompt_details["roles"],
initial_prompt_value=model_prompt_details["pre_message_sep"],
final_prompt_value=model_prompt_details["post_message_sep"],
messages=messages
)
else: else:
prompt = prompt_factory(model=model, messages=messages) prompt = prompt_factory(model=model, messages=messages)
### MAP INPUT PARAMS ### MAP INPUT PARAMS

7
litellm/llms/together_ai.py
View file

@ -45,7 +45,12 @@ def completion(
if model in custom_prompt_dict: if model in custom_prompt_dict:
# check if the model has a registered custom prompt # check if the model has a registered custom prompt
model_prompt_details = custom_prompt_dict[model] model_prompt_details = custom_prompt_dict[model]
prompt = custom_prompt(role_dict=model_prompt_details["roles"], pre_message_sep=model_prompt_details["pre_message_sep"], post_message_sep=model_prompt_details["post_message_sep"], messages=messages) prompt = custom_prompt(
role_dict=model_prompt_details["roles"],
initial_prompt_value=model_prompt_details["pre_message_sep"],
final_prompt_value=model_prompt_details["post_message_sep"],
messages=messages
)
else: else:
prompt = prompt_factory(model=model, messages=messages) prompt = prompt_factory(model=model, messages=messages)
data = { data = {