python3Packages.logfire{,-api}: init at 4.6.0

pkgs/development/python-modules/logfire-api/default.nix

@@ -0,0 +1,23 @@
+{
+  buildPythonPackage,
+  logfire,
+
+  # build system
+  hatchling,
+}:
+
+buildPythonPackage (finalAttrs: {
+  pname = "logfire-api";
+  inherit (logfire) version src;
+  pyproject = true;
+
+  sourceRoot = "${finalAttrs.src.name}/logfire-api";
+
+  build-system = [ hatchling ];
+
+  pythonImportsCheck = [ "logfire_api" ];
+
+  meta = logfire.meta // {
+    description = "Shim for the Logfire SDK which does nothing unless Logfire is installed";
+  };
+})

pkgs/development/python-modules/logfire/default.nix

@@ -0,0 +1,200 @@
+{
+  buildPythonPackage,
+  callPackage,
+  fetchFromGitHub,
+  lib,
+
+  # build-system
+  hatchling,
+
+  # dependencies
+  executing,
+  opentelemetry-exporter-otlp-proto-http,
+  opentelemetry-instrumentation,
+  opentelemetry-sdk,
+  protobuf,
+  rich,
+  tomli,
+  typing-extensions,
+
+  # optional dependencies
+  opentelemetry-instrumentation-aiohttp-client,
+  opentelemetry-instrumentation-asgi,
+  opentelemetry-instrumentation-celery,
+  opentelemetry-instrumentation-django,
+  opentelemetry-instrumentation-fastapi,
+  opentelemetry-instrumentation-flask,
+  opentelemetry-instrumentation-httpx,
+  opentelemetry-instrumentation-psycopg2,
+  opentelemetry-instrumentation-redis,
+  opentelemetry-instrumentation-requests,
+  opentelemetry-instrumentation-sqlalchemy,
+  opentelemetry-instrumentation-wsgi,
+  packaging,
+
+  # test dependencies
+  anthropic,
+  anyio,
+  asyncpg,
+  cloudpickle,
+  dirty-equals,
+  google-genai,
+  inline-snapshot,
+  logfire-api,
+  loguru,
+  mysql-connector,
+  openai-agents,
+  pandas,
+  psycopg,
+  pymongo,
+  pymysql,
+  pytest-django,
+  pytest-vcr,
+  pytest-xdist,
+  pytestCheckHook,
+  redis,
+  requests-mock,
+  sqlmodel,
+  structlog,
+  testcontainers,
+}:
+
+buildPythonPackage (finalAttrs: {
+  pname = "logfire";
+  version = "4.6.0";
+  pyproject = true;
+
+  src = fetchFromGitHub {
+    owner = "pydantic";
+    repo = "logfire";
+    tag = "v${finalAttrs.version}";
+    hash = "sha256-dAkT3xh0RsGTnW7Mqml2wV16VHJGUUkjjxiFLg9bUKc=";
+  };
+
+  build-system = [ hatchling ];
+
+  dependencies = [
+    executing
+    opentelemetry-exporter-otlp-proto-http
+    opentelemetry-instrumentation
+    opentelemetry-sdk
+    protobuf
+    rich
+    tomli
+    typing-extensions
+  ];
+
+  # Some optional dependencies are commented out because the deps they require
+  # are not in nixpkgs as of writing
+  optional-dependencies = {
+    aiohttp = [ opentelemetry-instrumentation-aiohttp-client ];
+    aiohttp-client = [ opentelemetry-instrumentation-aiohttp-client ];
+    # aiohttp-server = [ opentelemetry-instrumentation-aiohttp-server ];
+    asgi = [ opentelemetry-instrumentation-asgi ];
+    # asyncpg = [ opentelemetry-instrumentation-asyncpg ];
+    # aws-lambda = [ opentelemetry-instrumentation-aws-lambda ];
+    celery = [ opentelemetry-instrumentation-celery ];
+    django = [
+      opentelemetry-instrumentation-asgi
+      opentelemetry-instrumentation-django
+    ];
+    # dspy = [ opentelemetry-instrumentation-dspy ];
+    fastapi = [ opentelemetry-instrumentation-fastapi ];
+    flask = [ opentelemetry-instrumentation-flask ];
+    # google-genai = [ opentelemetry-instrumentation-google-genai ];
+    httpx = [ opentelemetry-instrumentation-httpx ];
+    # litellm = [ opentelemetry-instrumentation-litellm ];
+    # mysql = [ opentelemetry-instrumentation-mysql ];
+    psycopg = [
+      # opentelemetry-instrumentation-psycopg
+      packaging
+    ];
+    psycopg2 = [
+      opentelemetry-instrumentation-psycopg2
+      packaging
+    ];
+    # pymongo = [ opentelemetry-instrumentation-pymongo ];
+    redis = [ opentelemetry-instrumentation-redis ];
+    requests = [ opentelemetry-instrumentation-requests ];
+    sqlalchemy = [ opentelemetry-instrumentation-sqlalchemy ];
+    # sqlite3 = [ opentelemetry-instrumentation-sqlite3 ];
+    # starlette = [ opentelemetry-instrumentation-starlette ];
+    # system-metrics = [ opentelemetry-instrumentation-system-metrics ];
+    wsgi = [ opentelemetry-instrumentation-wsgi ];
+  };
+
+  pythonImportsCheck = [ "logfire" ];
+
+  # Too many outdated snapshots that fail with inline-snapshot
+  doCheck = false;
+
+  nativeCheckInputs = [
+    anthropic
+    anyio
+    asyncpg
+    cloudpickle
+    dirty-equals
+    google-genai
+    inline-snapshot
+    logfire-api
+    loguru
+    mysql-connector
+    openai-agents
+    pandas
+    psycopg
+    pymongo
+    pymysql
+    pytest-django
+    pytest-vcr
+    pytest-xdist
+    pytestCheckHook
+    redis
+    requests-mock
+    sqlmodel
+    structlog
+    testcontainers
+  ]
+  ++ lib.concatAttrValues finalAttrs.passthru.optional-dependencies;
+
+  disabledTestPaths = [
+    # Tests that require the commented optional dependencies above
+    "tests/otel_integrations/test_aiohttp_server.py"
+    "tests/otel_integrations/test_asyncpg.py"
+    "tests/otel_integrations/test_aws_lambda.py"
+    "tests/otel_integrations/test_google_genai.py"
+    "tests/otel_integrations/test_mysql.py"
+    "tests/otel_integrations/test_psycopg.py"
+    "tests/otel_integrations/test_pymongo.py"
+    "tests/otel_integrations/test_sqlalchemy.py"
+    "tests/otel_integrations/test_sqlite3.py"
+    "tests/otel_integrations/test_starlette.py"
+    "tests/otel_integrations/test_system_metrics.py"
+
+    # No module named 'litellm'
+    "tests/otel_integrations/test_litellm.py::test_litellm_instrumentation"
+
+    # No module named 'pydantic_ai'
+    "tests/otel_integrations/test_pydantic_ai.py"
+
+    # DeprecationWarning: The @wait_container_is_ready decorator is deprecated and will be removed in a future version. Use structured wait strategies instead: container.waiting_for(HttpWaitStrategy(8080).for_status_code(200)) or container.waiting_for(LogMessageWaitStrategy('ready'))
+    "tests/otel_integrations/test_celery.py"
+    "tests/otel_integrations/test_redis.py"
+
+    # Requires network
+    "tests/test_query_client.py"
+    "tests/otel_integrations/test_openai.py"
+    "tests/otel_integrations/test_openai_agents.py"
+  ];
+
+  meta = {
+    changelog = "https://logfire.pydantic.dev/docs/release-notes";
+    description = "Uncomplicated Observability for Python and beyond";
+    downloadPage = "https://github.com/pydantic/logfire/releases/tag/${finalAttrs.src.tag}";
+    homepage = "https://logfire.pydantic.dev";
+    license = lib.licenses.mit;
+    maintainers = with lib.maintainers; [
+      de11n
+      despsyched
+    ];
+  };
+})

pkgs/top-level/python-packages.nix

@@ -9106,6 +9106,10 @@ self: super: with self; {
 
   logboth = callPackage ../development/python-modules/logboth { };
 
+  logfire = callPackage ../development/python-modules/logfire { };
+
+  logfire-api = callPackage ../development/python-modules/logfire-api { };
+
   logfury = callPackage ../development/python-modules/logfury { };
 
   logging-journald = callPackage ../development/python-modules/logging-journald { };