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

fix(litellm_logging.py): initialize global variables

Browse source 
Fixes https://github.com/BerriAI/litellm/issues/4281
This commit is contained in:
Krrish Dholakia 2024-06-19 18:39:45 -07:00
parent 05a4b6188e
commit 682ec33aa0
5 changed files with 99 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/integrations/langfuse.py
View file

@ -13,7 +13,11 @@ from litellm._logging import verbose_logger
class LangFuseLogger:
# Class variables or attributes
def __init__(
self, langfuse_public_key=None, langfuse_secret=None, flush_interval=1
self,
langfuse_public_key=None,
langfuse_secret=None,
langfuse_host=None,
flush_interval=1,
):
try:
import langfuse
@ -25,7 +29,9 @@ class LangFuseLogger:
# Instance variables
self.secret_key = langfuse_secret or os.getenv("LANGFUSE_SECRET_KEY")
self.public_key = langfuse_public_key or os.getenv("LANGFUSE_PUBLIC_KEY")
self.langfuse_host = os.getenv("LANGFUSE_HOST", "https://cloud.langfuse.com")
self.langfuse_host = langfuse_host or os.getenv(
"LANGFUSE_HOST", "https://cloud.langfuse.com"
)
self.langfuse_release = os.getenv("LANGFUSE_RELEASE")
self.langfuse_debug = os.getenv("LANGFUSE_DEBUG")