Add OpenAI metrics collections

instana · May 8, 2024 · b8b59d2 · b8b59d2
1 parent e7a795e
commit b8b59d2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/llm/src/main/java/com/instana/dc/llm/impl/llm/MetricsCollectorService.java b/llm/src/main/java/com/instana/dc/llm/impl/llm/MetricsCollectorService.java
@@ -116,7 +116,8 @@ public void export(
 
                         switch (metric.getDataCase()) {
                             case SUM:
-                                if (metric.getName().compareTo("llm.watsonx.completions.tokens") == 0) {
+                                if (metric.getName().compareTo("llm.watsonx.completions.tokens") == 0 ||
+                                        metric.getName().compareTo("llm.openai.chat_completions.tokens") == 0) {
 
                                     List<NumberDataPoint> sumDataPoints = metric.getSum().getDataPointsList();
                                     for (NumberDataPoint dataPoint : sumDataPoints) {
@@ -157,7 +158,8 @@ public void export(
                                 }
                                 break;
                             case HISTOGRAM:
-                                if (metric.getName().compareTo("llm.watsonx.completions.duration") == 0) {
+                                if (metric.getName().compareTo("llm.watsonx.completions.duration") == 0 ||
+                                        metric.getName().compareTo("llm.openai.chat_completions.duration") == 0) {
 
                                     List<HistogramDataPoint> histDataPoints = metric.getHistogram().getDataPointsList();
                                     for (HistogramDataPoint dataPoint : histDataPoints) {