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

Merge pull request #2344 from BerriAI/litellm_fix_cant_import_litellm

Browse source 
(fix) bug importing litellm on python 3.8
This commit is contained in:
Ishaan Jaff 2024-03-05 20:36:08 -08:00 • committed by GitHub
commit a514eae85b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/bedrock.py
View file

@ -376,7 +376,7 @@ class AmazonMistralConfig:
temperature: Optional[float] = None temperature: Optional[float] = None
top_p: Optional[float] = None top_p: Optional[float] = None
top_k: Optional[float] = None top_k: Optional[float] = None
stop: Optional[list[str]] = None stop: Optional[List[str]] = None
def __init__( def __init__(
self, self,
@ -384,7 +384,7 @@ class AmazonMistralConfig:
temperature: Optional[float] = None, temperature: Optional[float] = None,
top_p: Optional[int] = None, top_p: Optional[int] = None,
top_k: Optional[float] = None, top_k: Optional[float] = None,
stop: Optional[list[str]] = None, stop: Optional[List[str]] = None,
) -> None: ) -> None:
locals_ = locals() locals_ = locals()
for key, value in locals_.items(): for key, value in locals_.items():

4
litellm/llms/prompt_templates/factory.py
View file

@ -2,7 +2,7 @@ from enum import Enum
import requests, traceback import requests, traceback
import json, re, xml.etree.ElementTree as ET import json, re, xml.etree.ElementTree as ET
from jinja2 import Template, exceptions, Environment, meta from jinja2 import Template, exceptions, Environment, meta
from typing import Optional, Any from typing import Optional, Any, List
import imghdr, base64 import imghdr, base64
@ -586,7 +586,7 @@ def anthropic_messages_pt(messages: list):
return new_messages return new_messages
def extract_between_tags(tag: str, string: str, strip: bool = False) -> list[str]: def extract_between_tags(tag: str, string: str, strip: bool = False) -> List[str]:
ext_list = re.findall(f"<{tag}>(.+?)</{tag}>", string, re.DOTALL) ext_list = re.findall(f"<{tag}>(.+?)</{tag}>", string, re.DOTALL)
if strip: if strip:
ext_list = [e.strip() for e in ext_list] ext_list = [e.strip() for e in ext_list]