phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(utils.py): adding support for rules + mythomax/alpaca prompt template

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-20 18:57:58 -08:00
parent 4f46ac4ab5
commit 855964ed45
7 changed files with 186 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/openai.py
View file

@ -6,6 +6,7 @@ from litellm.utils import ModelResponse, Choices, Message, CustomStreamWrapper,
from typing import Callable, Optional
import aiohttp, requests
import litellm
from .prompt_templates.factory import prompt_factory, custom_prompt
from openai import OpenAI, AsyncOpenAI
class OpenAIError(Exception):
@ -172,7 +173,8 @@ class OpenAIChatCompletion(BaseLLM):
optional_params=None,
litellm_params=None,
logger_fn=None,
headers: Optional[dict]=None):
headers: Optional[dict]=None,
custom_prompt_dict: dict={}):
super().completion()
exception_mapping_worked = False
try: