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

fix - allow langfuse init with flush interval

Browse source
This commit is contained in:
Ishaan Jaff 2024-04-27 14:53:18 -07:00
parent 3cae5e17a3
commit de36ffa292
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/integrations/langfuse.py
View file

@ -12,7 +12,9 @@ import litellm
class LangFuseLogger: class LangFuseLogger:
# Class variables or attributes # Class variables or attributes
def __init__(self, langfuse_public_key=None, langfuse_secret=None): def __init__(
self, langfuse_public_key=None, langfuse_secret=None, flush_interval=1
):
try: try:
from langfuse import Langfuse from langfuse import Langfuse
except Exception as e: except Exception as e:
@ -31,7 +33,7 @@ class LangFuseLogger:
host=self.langfuse_host, host=self.langfuse_host,
release=self.langfuse_release, release=self.langfuse_release,
debug=self.langfuse_debug, debug=self.langfuse_debug,
flush_interval=1, # flush interval in seconds flush_interval=flush_interval, # flush interval in seconds
) )
# set the current langfuse project id in the environ # set the current langfuse project id in the environ