mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-09-24 03:41:36 +00:00
6b75a8d270
- Updated project creation to include `CLAUDE.md` and `GEMINI.md` that import `AGENTS.md`, ensuring consistent guidance across coding assistants. - Implemented utility functions to copy assistant instruction files during project setup. - Enhanced documentation in `AGENTS.md` to emphasize the importance of keeping telemetry enabled for optimal performance. - Added tests to verify the correct scaffolding of assistant instruction files and their contents.
70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
"""The crewai Telemetry pipeline must not log when the collector is unreachable.
|
|
|
|
The filter lives in ``crewai_core``; this pins that the ``crewai`` package wires
|
|
the same exporter, since a duplicated exporter here used to log every failure.
|
|
"""
|
|
|
|
import logging
|
|
import time
|
|
from unittest.mock import patch
|
|
|
|
from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
|
|
import pytest
|
|
import requests
|
|
|
|
import crewai_core.telemetry as core_telemetry
|
|
from crewai.telemetry.telemetry import Telemetry
|
|
|
|
|
|
@pytest.fixture
|
|
def live_telemetry(monkeypatch):
|
|
monkeypatch.setattr(Telemetry, "_instance", None)
|
|
monkeypatch.setattr(Telemetry, "_register_shutdown_handlers", lambda self: None)
|
|
for var in ("CREWAI_DISABLE_TELEMETRY", "CREWAI_DISABLE_TRACKING", "OTEL_SDK_DISABLED"):
|
|
monkeypatch.setenv(var, "false")
|
|
telemetry = Telemetry()
|
|
try:
|
|
yield telemetry
|
|
finally:
|
|
if telemetry.ready:
|
|
telemetry.provider.shutdown()
|
|
Telemetry._instance = None
|
|
|
|
|
|
def test_pipeline_is_silent_when_collector_rejects(live_telemetry, caplog):
|
|
with (
|
|
patch(
|
|
"requests.Session.post",
|
|
side_effect=requests.exceptions.HTTPError("400 Client Error"),
|
|
),
|
|
caplog.at_level(logging.DEBUG),
|
|
):
|
|
live_telemetry.provider.get_tracer("test").start_span("probe").end()
|
|
assert live_telemetry.provider.force_flush(timeout_millis=10_000)
|
|
|
|
otlp_records = [r for r in caplog.records if r.name == OTLPSpanExporter.__module__]
|
|
assert otlp_records == []
|
|
|
|
|
|
def test_exit_hook_stops_waiting_at_the_flush_deadline(
|
|
live_telemetry, monkeypatch, caplog
|
|
):
|
|
monkeypatch.setattr(core_telemetry, "FINAL_FLUSH_SECONDS", 1)
|
|
live_telemetry.provider.get_tracer("test").start_span("probe").end()
|
|
|
|
with (
|
|
patch(
|
|
"requests.Session.post",
|
|
side_effect=requests.exceptions.ConnectionError("collector down"),
|
|
),
|
|
caplog.at_level(logging.DEBUG),
|
|
):
|
|
started = time.monotonic()
|
|
live_telemetry._shutdown()
|
|
elapsed = time.monotonic() - started
|
|
|
|
assert elapsed < 5 # the exporter's own retry budget is 30s
|
|
assert live_telemetry.ready is False
|
|
otlp_records = [r for r in caplog.records if r.name == OTLPSpanExporter.__module__]
|
|
assert otlp_records == []
|