Merge pull request #44 from grafana/add-loki-tools

Add loki tools

Author: ioanarm

README.md

@@ -10,14 +10,20 @@ This provides access to your Grafana instance and the surrounding ecosystem.
 - [x] List and fetch datasource information
 - [ ] Query datasources
   - [x] Prometheus
-  - [ ] Loki (log queries, metric queries)
+  - [x] Loki
+    - [x] Log queries
+    - [x] Metric queries
   - [ ] Tempo
   - [ ] Pyroscope
 - [x] Query Prometheus metadata
   - [x] Metric metadata
   - [x] Metric names
   - [x] Label names
   - [x] Label values
+- [x] Query Loki metadata
+  - [x] Label names
+  - [x] Label values
+  - [x] Stats
 - [x] Search, create, update and close incidents
 - [ ] Start Sift investigations and view the results
 
@@ -41,6 +47,10 @@ This is useful if you don't use certain functionality or if you don't want to ta
 | `create_incident` | Incident | Create an incident in Grafana Incident |
 | `add_activity_to_incident` | Incident | Add an activity item to an incident in Grafana Incident |
 | `resolve_incident` | Incident | Resolve an incident in Grafana Incident |
+| `query_loki_logs` | Loki | Query and retrieve logs using LogQL (either log or metric queries) |
+| `list_loki_label_names` | Loki | List all available label names in logs |
+| `list_loki_label_values` | Loki | List values for a specific log label |
+| `query_loki_stats` | Loki | Get statistics about log streams |
 
 ## Usage
 

cmd/mcp-grafana/main.go

@@ -22,6 +22,7 @@ func newServer() *server.MCPServer {
 	tools.AddDatasourceTools(s)
 	tools.AddIncidentTools(s)
 	tools.AddPrometheusTools(s)
+	tools.AddLokiTools(s)
 	return s
 }
 

docker-compose.yaml

@@ -19,5 +19,20 @@ services:
     command:
       - --config.file=/etc/prometheus/prometheus.yml
 
-volumes:
-  var-lib-grafana:
+  loki:
+    image: grafana/loki:2.6.0
+    ports:
+      - "3100:3100"
+    command: -config.file=/etc/loki/loki-config.yml
+    volumes:
+      - ./tests/loki-config.yml:/etc/loki/loki-config.yml
+
+  promtail:
+    image: grafana/promtail:2.6.0
+    volumes:
+      - ./tests/promtail-config.yml:/etc/promtail/config.yml
+      - /var/log:/var/log
+      - /var/run/docker.sock:/var/run/docker.sock
+    command: -config.file=/etc/promtail/config.yml
+    depends_on:
+      - loki

tests/loki-config.yml

@@ -0,0 +1,40 @@
+auth_enabled: false
+
+server:
+  http_listen_port: 3100
+
+ingester:
+  lifecycler:
+    address: 127.0.0.1
+    ring:
+      kvstore:
+        store: inmemory
+      replication_factor: 1
+    final_sleep: 0s
+  chunk_idle_period: 5m
+  chunk_retain_period: 30s
+  wal:
+    enabled: true
+    dir: /loki/wal
+
+compactor:
+  working_directory: /loki/compactor
+
+schema_config:
+  configs:
+    - from: 2020-10-24
+      store: boltdb-shipper
+      object_store: filesystem
+      schema: v11
+      index:
+        prefix: index_
+        period: 24h
+
+storage_config:
+  boltdb_shipper:
+    active_index_directory: /loki/boltdb-shipper-active
+    cache_location: /loki/boltdb-shipper-cache
+    cache_ttl: 24h
+    shared_store: filesystem
+  filesystem:
+    directory: /loki/chunks

tests/promtail-config.yml

@@ -0,0 +1,18 @@
+server:
+  http_listen_port: 9080
+
+positions:
+  filename: /tmp/positions.yaml
+
+clients:
+  - url: http://loki:3100/loki/api/v1/push
+
+scrape_configs:
+  - job_name: docker
+    docker_sd_configs:
+      - host: unix:///var/run/docker.sock
+        refresh_interval: 5s
+    relabel_configs:
+      - source_labels: ['__meta_docker_container_name']
+        regex: '/(.*)'
+        target_label: 'container'

tests/provisioning/datasources/datasources.yaml

@@ -13,4 +13,11 @@ datasources:
     uid: prometheus-demo
     type: prometheus
     access: proxy
-    url: https://prometheus.demo.prometheus.io
+    url: https://prometheus.demo.prometheus.io
+  - name: Loki
+    id: 3
+    uid: loki
+    type: loki
+    access: proxy
+    url: http://loki:3100
+    isDefault: false

tools/datasources_test.go

@@ -53,7 +53,8 @@ func TestDatasourcesTools(t *testing.T) {
 		ctx := newTestContext()
 		result, err := listDatasources(ctx, ListDatasourcesParams{})
 		require.NoError(t, err)
-		assert.Greater(t, len(result), 0)
+		// Three datasources are provisioned in the test environment (Prometheus and Loki).
+		assert.Len(t, result, 3)
 	})
 
 	t.Run("get datasource by uid", func(t *testing.T) {