From e0921da38c05b48474dc203150fc713488e67dac Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 21 Nov 2024 22:01:12 -0800
Subject: [PATCH] test_team_logging

---
 tests/test_config.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/test_config.py b/tests/test_config.py
index 03de4653f..888949982 100644
--- a/tests/test_config.py
+++ b/tests/test_config.py
@@ -20,6 +20,7 @@ async def config_update(session):
             "success_callback": ["langfuse"],
         },
         "environment_variables": {
+            "LANGFUSE_HOST": os.environ["LANGFUSE_HOST"],
             "LANGFUSE_PUBLIC_KEY": os.environ["LANGFUSE_PUBLIC_KEY"],
             "LANGFUSE_SECRET_KEY": os.environ["LANGFUSE_SECRET_KEY"],
         },
@@ -98,6 +99,7 @@ async def test_team_logging():
             import langfuse
 
             langfuse_client = langfuse.Langfuse(
+                host=os.getenv("LANGFUSE_HOST"),
                 public_key=os.getenv("LANGFUSE_PUBLIC_KEY"),
                 secret_key=os.getenv("LANGFUSE_SECRET_KEY"),
             )