tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-04 09:06:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #826 from rodneyxr/ollama-fixes

Browse Source 
Fix typo for initial_prompt_value and too many values to unpack error
This commit is contained in:
Ishaan Jaff
2023-11-16 07:55:53 -08:00
committed by GitHub
parent 791a3877d9 5d39279ac0
commit 3f95fd2da5
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/main.py
+1 -1
View File
@@ -322,7 +322,7 @@ def completion(
max_retries = kwargs.get("max_retries", None)
context_window_fallback_dict = kwargs.get("context_window_fallback_dict", None)
### CUSTOM PROMPT TEMPLATE ###
initial_prompt_value = kwargs.get("intial_prompt_value", None)
initial_prompt_value = kwargs.get("initial_prompt_value", None)
roles = kwargs.get("roles", None)
final_prompt_value = kwargs.get("final_prompt_value", None)
bos_token = kwargs.get("bos_token", None)
litellm/utils.py
+1 -1
View File
@@ -3159,7 +3159,7 @@ def register_prompt_template(model: str, roles: dict, initial_prompt_value: str
)
```
"""
model, _ = get_llm_provider(model=model)
model = get_llm_provider(model=model)[0]
litellm.custom_prompt_dict[model] = {
"roles": roles,
"initial_prompt_value": initial_prompt_value,