mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-09-21 18:36:47 +00:00
993a96c4e0
Some checks failed
CodeQL Advanced / Analyze (actions) (push) Has been cancelled
CodeQL Advanced / Analyze (python) (push) Has been cancelled
Check Documentation Broken Links / Check broken links (push) Has been cancelled
Vulnerability Scan / Detect changes (push) Has been cancelled
Vulnerability Scan / pip-audit (push) Has been cancelled
Build uv cache / build-cache (3.10) (push) Has been cancelled
Build uv cache / build-cache (3.11) (push) Has been cancelled
Build uv cache / build-cache (3.12) (push) Has been cancelled
Build uv cache / build-cache (3.13) (push) Has been cancelled
Nightly Canary Release / Check for new commits (push) Has been cancelled
Nightly Canary Release / Build nightly packages (push) Has been cancelled
Nightly Canary Release / Publish nightly to PyPI (push) Has been cancelled
* feat(tracing): port enterprise event sessions to OSS * fix(tracing): address review findings and verify concurrent exports * fix(tracing): keep redactor ownership in enterprise integrations * test(tracing): isolate intentional failures from cleanup assertions
72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
"""Explicit tracing controls gate the execution session and its transport."""
|
|
|
|
from datetime import datetime, timedelta, timezone
|
|
from unittest.mock import Mock
|
|
|
|
from crewai import Agent, Crew, Task
|
|
from crewai.llms.base_llm import BaseLLM
|
|
from crewai.telemetry.tracing.grants import (
|
|
GrantSpanExporter,
|
|
TraceGrant,
|
|
TraceGrantClient,
|
|
)
|
|
from opentelemetry.sdk.trace.export.in_memory_span_exporter import InMemorySpanExporter
|
|
import pytest
|
|
|
|
|
|
class LocalLLM(BaseLLM):
|
|
def __init__(self):
|
|
super().__init__(model="local-test")
|
|
|
|
def call(self, messages, **kwargs):
|
|
return "Final Answer: hello"
|
|
|
|
def supports_function_calling(self):
|
|
return False
|
|
|
|
def supports_stop_words(self):
|
|
return False
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"environment,override,enabled",
|
|
[
|
|
("false", None, False),
|
|
("true", False, False),
|
|
("true", None, True),
|
|
("false", True, True),
|
|
],
|
|
)
|
|
def test_tracing_controls_gate_execution_export(
|
|
monkeypatch, environment, override, enabled
|
|
):
|
|
monkeypatch.setenv("CREWAI_TRACING_ENABLED", environment)
|
|
monkeypatch.setenv("CREWAI_DISABLE_TELEMETRY", "true")
|
|
monkeypatch.setenv("OTEL_SDK_DISABLED", "false")
|
|
monkeypatch.setenv("CREWAI_USER_PAT", "synthetic-pat")
|
|
exporter = InMemorySpanExporter()
|
|
grant = Mock(
|
|
side_effect=lambda execution_uuid: TraceGrant(
|
|
token="synthetic-grant",
|
|
collector_url="https://collector.invalid/v1/traces",
|
|
execution_uuid=execution_uuid,
|
|
expires_at=datetime.now(timezone.utc) + timedelta(minutes=15),
|
|
)
|
|
)
|
|
monkeypatch.setattr(TraceGrantClient, "create", grant)
|
|
monkeypatch.setattr(
|
|
GrantSpanExporter, "_exporter", staticmethod(lambda _: exporter)
|
|
)
|
|
legacy = Mock(side_effect=AssertionError("legacy trace transport used"))
|
|
monkeypatch.setattr(
|
|
"crewai.events.listeners.tracing.trace_batch_manager.TraceBatchManager.initialize_batch",
|
|
legacy,
|
|
)
|
|
agent = Agent(role="tester", goal="greet", backstory="tester", llm=LocalLLM())
|
|
task = Task(description="say hello", expected_output="hello", agent=agent)
|
|
result = Crew(agents=[agent], tasks=[task], tracing=override).kickoff()
|
|
assert result.raw == "hello"
|
|
assert grant.call_count == int(enabled)
|
|
assert bool(exporter.get_finished_spans()) is enabled
|
|
legacy.assert_not_called()
|