Skip to content

Commit 2a8aa8d

Browse files
committed
fix test fixtures
1 parent f561ec9 commit 2a8aa8d

File tree

2 files changed

+10
-6
lines changed

2 files changed

+10
-6
lines changed

tests/contrib/crewai/conftest.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
from ddtrace.contrib.internal.crewai.patch import patch
1313
from ddtrace.contrib.internal.crewai.patch import unpatch
1414
from ddtrace.llmobs import LLMObs as llmobs_service
15-
from ddtrace.llmobs._constants import AGENTLESS_BASE_URL
15+
from ddtrace.llmobs._constants import AGENTLESS_SPAN_BASE_URL
1616
from ddtrace.llmobs._writer import LLMObsSpanWriter
1717
from ddtrace.trace import Pin
1818
from tests.utils import DummyTracer
@@ -200,8 +200,10 @@ def crewai_llmobs(mock_tracer, llmobs_span_writer):
200200

201201
@pytest.fixture
202202
def llmobs_span_writer():
203-
agentless_url = "{}.{}".format(AGENTLESS_BASE_URL, "datad0g.com")
204-
yield TestLLMObsSpanWriter(is_agentless=True, agentless_url=agentless_url, interval=1.0, timeout=1.0)
203+
dd_site = "datad0g.com"
204+
api_key = "<not-a-real-key>"
205+
agentless_url = "{}.{}".format(AGENTLESS_SPAN_BASE_URL, dd_site)
206+
yield TestLLMObsSpanWriter(dd_site, api_key, 1.0, 5.0, is_agentless=True, _agentless_url=agentless_url)
205207

206208

207209
@pytest.fixture

tests/contrib/openai_agents/conftest.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
from ddtrace.contrib.internal.openai_agents.patch import patch
1111
from ddtrace.contrib.internal.openai_agents.patch import unpatch
1212
from ddtrace.llmobs import LLMObs as llmobs_service
13-
from ddtrace.llmobs._constants import AGENTLESS_BASE_URL
13+
from ddtrace.llmobs._constants import AGENTLESS_SPAN_BASE_URL
1414
from ddtrace.llmobs._writer import LLMObsSpanWriter
1515
from ddtrace.trace import Pin
1616
from tests.utils import DummyTracer
@@ -192,8 +192,10 @@ def mock_tracer_chat_completions(agents, openai, mock_tracer):
192192

193193
@pytest.fixture
194194
def llmobs_span_writer():
195-
agentless_url = "{}.{}".format(AGENTLESS_BASE_URL, "datad0g.com")
196-
yield TestLLMObsSpanWriter(is_agentless=True, agentless_url=agentless_url, interval=1.0, timeout=1.0)
195+
dd_site = "datad0g.com"
196+
api_key = "<not-a-real-key>"
197+
agentless_url = "{}.{}".format(AGENTLESS_SPAN_BASE_URL, dd_site)
198+
yield TestLLMObsSpanWriter(dd_site, api_key, 1.0, 5.0, is_agentless=True, _agentless_url=agentless_url)
197199

198200

199201
@pytest.fixture

0 commit comments

Comments
 (0)