From 9bcdee0f6465c699c03891eb1fe97b1251609a1b Mon Sep 17 00:00:00 2001
From: Cyber MacGeddon <cybermaggedon@gmail.com>
Date: Tue, 23 Jul 2024 23:27:09 +0100
Subject: [PATCH] Fixed dashboard

---
 grafana/dashboard.json    | 6 +++---
 prometheus/prometheus.yml | 4 ++--
 setup.py                  | 2 ++
 3 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/grafana/dashboard.json b/grafana/dashboard.json
index 0485a3bc..dcd793d0 100644
--- a/grafana/dashboard.json
+++ b/grafana/dashboard.json
@@ -106,7 +106,7 @@
             "uid": "f6b18033-5918-4e05-a1ca-4cb30343b129"
           },
           "editorMode": "builder",
-          "expr": "histogram_quantile(0.99, sum by(le) (rate(request_latency_bucket{instance=\"llm:8000\"}[$__rate_interval])))",
+          "expr": "histogram_quantile(0.99, sum by(le) (rate(request_latency_bucket{instance=\"text-completion:8000\"}[$__rate_interval])))",
           "instant": false,
           "legendFormat": "99%",
           "range": true,
@@ -118,7 +118,7 @@
             "uid": "f6b18033-5918-4e05-a1ca-4cb30343b129"
           },
           "editorMode": "builder",
-          "expr": "histogram_quantile(0.95, sum by(le) (rate(request_latency_bucket{instance=\"llm:8000\"}[$__rate_interval])))",
+          "expr": "histogram_quantile(0.95, sum by(le) (rate(request_latency_bucket{instance=\"text-completion:8000\"}[$__rate_interval])))",
           "hide": false,
           "instant": false,
           "legendFormat": "95%",
@@ -131,7 +131,7 @@
             "uid": "f6b18033-5918-4e05-a1ca-4cb30343b129"
           },
           "editorMode": "builder",
-          "expr": "histogram_quantile(0.75, sum by(le) (rate(request_latency_bucket{instance=\"llm:8000\"}[$__rate_interval])))",
+          "expr": "histogram_quantile(0.75, sum by(le) (rate(request_latency_bucket{instance=\"text-completion:8000\"}[$__rate_interval])))",
           "hide": false,
           "instant": false,
           "legendFormat": "75%",
diff --git a/prometheus/prometheus.yml b/prometheus/prometheus.yml
index 1f305f6b..f24a495e 100644
--- a/prometheus/prometheus.yml
+++ b/prometheus/prometheus.yml
@@ -28,8 +28,8 @@ scrape_configs:
         - 'embeddings:8000'
         - 'kg-extract-definitions:8000'
         - 'kg-extract-relationships:8000'
-        - 'vector-write:8000'
-        - 'graph-write:8000'
+        - 'store-graph-embeddings:8000'
+        - 'store-triples:8000'
         - 'text-completion:8000'
         - 'graph-rag:8000'
         - 'pulsar:8080'
diff --git a/setup.py b/setup.py
index 2b95590c..36bbf6d9 100644
--- a/setup.py
+++ b/setup.py
@@ -67,5 +67,7 @@ setuptools.setup(
         "scripts/text-completion-ollama",
         "scripts/text-completion-vertexai",
         "scripts/triples-write-cassandra",
+        "scripts/dump-parquet",
+        "scripts/triples-dump-parquet",
     ]
 )