loki-reader-core 0.2.1__tar.gz → 0.2.2__tar.gz

{loki_reader_core-0.2.1 → loki_reader_core-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: loki-reader-core
-Version: 0.2.1
+Version: 0.2.2
 Summary: Python library for querying Grafana Loki logs via REST API
 Project-URL: Homepage, https://github.com/jmazzahacks/loki-reader-core
 Project-URL: Issues, https://github.com/jmazzahacks/loki-reader-core/issues

{loki_reader_core-0.2.1 → loki_reader_core-0.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "loki-reader-core"
-version = "0.2.1"
+version = "0.2.2"
 authors = [
   { name="Jason Byteforge", email="jason@mzmail.me" },
 ]

{loki_reader_core-0.2.1 → loki_reader_core-0.2.2}/src/loki_reader_core/__init__.py

@@ -6,7 +6,7 @@ from .client import LokiClient
 from .exceptions import LokiAuthError, LokiConnectionError, LokiError, LokiQueryError
 from .models import LogEntry, LogStream, MetricSample, MetricSeries, QueryResult, QueryStats
 
-__version__ = "0.2.1"
+__version__ = "0.2.2"
 
 __all__ = [
     "LokiClient",

{loki_reader_core-0.2.1 → loki_reader_core-0.2.2}/src/loki_reader_core/client.py

@@ -269,6 +269,20 @@ class LokiClient:
 
         return data
 
+    def _discovery_time_range(self) -> tuple[int, int]:
+        """Get a 30-day time range for label discovery queries.
+
+        Without a time range, Loki's label API only returns labels for
+        recently active streams. Using a 30-day window ensures we find
+        apps that haven't logged in a while.
+
+        Returns:
+            Tuple of (start_ns, end_ns) covering the last 30 days.
+        """
+        end = now_ns()
+        start = end - (30 * 24 * NANOSECONDS_PER_HOUR)
+        return (start, end)
+
     def _find_app_label(self, app_value: str) -> str:
         """Discover which label name contains the given app value.
 
@@ -287,8 +301,9 @@ class LokiClient:
         if app_value in self._app_label_cache:
             return self._app_label_cache[app_value]
 
+        start, end = self._discovery_time_range()
         for label_name in APP_LABEL_NAMES:
-            values = self.get_label_values(label_name)
+            values = self.get_label_values(label_name, start=start, end=end)
             if app_value in values:
                 self._app_label_cache[app_value] = label_name
                 return label_name
@@ -310,8 +325,9 @@ class LokiClient:
         if self._severity_label_cache is not None:
             return self._severity_label_cache
 
+        start, end = self._discovery_time_range()
         for label_name in SEVERITY_LABEL_NAMES:
-            values = self.get_label_values(label_name)
+            values = self.get_label_values(label_name, start=start, end=end)
             if values:
                 self._severity_label_cache = label_name
                 return label_name

{loki_reader_core-0.2.1 → loki_reader_core-0.2.2}/tests/test_client.py

@@ -453,12 +453,16 @@ class TestLabelDiscovery:
             mock_glv.return_value = ["myapp", "otherapp"]
             result = client._find_app_label("myapp")
             assert result == "application"
-            mock_glv.assert_called_once_with("application")
+            mock_glv.assert_called_once()
+            args = mock_glv.call_args
+            assert args.args[0] == "application"
+            assert "start" in args.kwargs
+            assert "end" in args.kwargs
 
     def test_find_app_label_job(self) -> None:
         client = LokiClient(base_url="http://localhost:3100")
         with patch.object(client, "get_label_values") as mock_glv:
-            def side_effect(label: str) -> list[str]:
+            def side_effect(label: str, start: int = None, end: int = None) -> list[str]:
                 if label == "job":
                     return ["myapp", "worker"]
                 return []
@@ -480,7 +484,11 @@ class TestLabelDiscovery:
             mock_glv.return_value = ["info", "error", "warn"]
             result = client._find_severity_label()
             assert result == "level"
-            mock_glv.assert_called_once_with("level")
+            mock_glv.assert_called_once()
+            args = mock_glv.call_args
+            assert args.args[0] == "level"
+            assert "start" in args.kwargs
+            assert "end" in args.kwargs
 
     def test_find_severity_label_none(self) -> None:
         client = LokiClient(base_url="http://localhost:3100")
@@ -498,7 +506,7 @@ class TestLabelDiscovery:
             client._find_app_label("myapp")
 
             # Only called once - second call uses cache
-            mock_glv.assert_called_once_with("application")
+            mock_glv.assert_called_once()
 
     def test_severity_label_cached(self) -> None:
         client = LokiClient(base_url="http://localhost:3100")
@@ -508,7 +516,19 @@ class TestLabelDiscovery:
             client._find_severity_label()
             client._find_severity_label()
 
-            mock_glv.assert_called_once_with("level")
+            mock_glv.assert_called_once()
+
+    def test_discovery_uses_30_day_range(self) -> None:
+        client = LokiClient(base_url="http://localhost:3100")
+        with patch.object(client, "get_label_values") as mock_glv:
+            mock_glv.return_value = ["myapp"]
+            client._find_app_label("myapp")
+
+            args = mock_glv.call_args
+            start = args.kwargs["start"]
+            end = args.kwargs["end"]
+            diff_days = (end - start) / (24 * 60 * 60 * 1_000_000_000)
+            assert diff_days == 30
 
 
 class TestMergeStreams: