mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-07-29 02:29:31 +00:00
60f405fe54
`Telemetry.set_tracer()` installed crewAI's anonymous SDK
`TracerProvider` into OpenTelemetry's process-global slot, so the first
`Crew` constructed in a test or host application replaced the default
`ProxyTracerProvider` and exfiltrated every host span emitted via
`trace.get_tracer(...)` to crewAI's OTLP endpoint. Keep the provider
local to the `Telemetry` instance and route every anonymous span
through `self.provider.get_tracer("crewai.telemetry")` so the global
slot stays untouched. Mirrors the fix in `crewai_core.telemetry`,
drops the now-dead `set_tracer()` calls in `event_listener.py` and
`crewai_cli.command`, and adds regression coverage that asserts the
provider stays a `ProxyTracerProvider` after constructing a `Crew`.
80 lines
2.5 KiB
Python
80 lines
2.5 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
|
|
from crewai_core.telemetry import Telemetry
|
|
import httpx
|
|
from rich.console import Console
|
|
|
|
from crewai_cli.authentication.token import get_auth_token
|
|
from crewai_cli.plus_api import PlusAPI
|
|
|
|
|
|
console = Console()
|
|
|
|
|
|
class AuthenticationRequiredError(SystemExit):
|
|
"""Raised when a Plus API command needs the user to log in first."""
|
|
|
|
|
|
class BaseCommand:
|
|
def __init__(self) -> None:
|
|
self._telemetry = Telemetry()
|
|
|
|
|
|
class PlusAPIMixin:
|
|
def __init__(self, telemetry: Telemetry) -> None:
|
|
try:
|
|
self.plus_api_client = PlusAPI(api_key=get_auth_token())
|
|
except Exception:
|
|
telemetry.deploy_signup_error_span()
|
|
console.print(
|
|
"Please sign up/login to CrewAI+ before using the CLI.",
|
|
style="bold red",
|
|
)
|
|
console.print("Run 'crewai login' to sign up/login.", style="bold green")
|
|
raise AuthenticationRequiredError from None
|
|
|
|
def _validate_response(self, response: httpx.Response) -> None:
|
|
"""Handle and display error messages from API responses.
|
|
|
|
Args:
|
|
response: The response from the Plus API.
|
|
"""
|
|
try:
|
|
json_response = response.json()
|
|
except (json.JSONDecodeError, ValueError):
|
|
console.print(
|
|
"Failed to parse response from Enterprise API failed. Details:",
|
|
style="bold red",
|
|
)
|
|
console.print(f"Status Code: {response.status_code}")
|
|
console.print(
|
|
f"Response:\n{response.content.decode('utf-8', errors='replace')}"
|
|
)
|
|
raise SystemExit from None
|
|
|
|
if response.status_code == 422:
|
|
console.print(
|
|
"Failed to complete operation. Please fix the following errors:",
|
|
style="bold red",
|
|
)
|
|
for field, messages in json_response.items():
|
|
for message in messages:
|
|
console.print(
|
|
f"* [bold red]{field.capitalize()}[/bold red] {message}"
|
|
)
|
|
raise SystemExit
|
|
|
|
if not response.is_success:
|
|
console.print(
|
|
"Request to Enterprise API failed. Details:", style="bold red"
|
|
)
|
|
details = (
|
|
json_response.get("error")
|
|
or json_response.get("message")
|
|
or response.content.decode("utf-8", errors="replace")
|
|
)
|
|
console.print(f"{details}")
|
|
raise SystemExit
|