add langfuse host in callback

2023-10-04 09:12:20 -07:00 · 2023-10-04 09:12:20 -07:00 · e0373e05ad
commit e0373e05ad
parent a6e3bcaeb3
1 changed files with 1 additions and 1 deletions
--- a/litellm/integrations/langfuse.py
+++ b/litellm/integrations/langfuse.py
@ -18,11 +18,11 @@ class LangFuseLogger:
        # Instance variables
        self.secret_key = os.getenv("LANGFUSE_SECRET_KEY")
        self.public_key = os.getenv("LANGFUSE_PUBLIC_KEY")
+        self.langfuse_host = os.getenv("LANGFUSE_HOST", "https://cloud.langfuse.com")
        self.Langfuse =  Langfuse(
                        public_key=self.public_key,
                        secret_key=self.secret_key,
                        host="https://cloud.langfuse.com",
-                        # debug=True
                    )

    def log_event(self, kwargs, response_obj, start_time, end_time, print_verbose):