APM instrumentation improvements

jadevogt · Sep 17, 2023 · 6df7e87 · 6df7e87
1 parent 91f3b14
commit 6df7e87
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 5 deletions.
diff --git a/pom.xml b/pom.xml
@@ -6,7 +6,7 @@
 
     <groupId>best.tigers</groupId>
     <artifactId>tigersbot</artifactId>
-    <version>1.3.1</version>
+    <version>1.3.2</version>
     <properties>
         <maven.compiler.source>19</maven.compiler.source>
         <maven.compiler.target>19</maven.compiler.target>

diff --git a/src/main/java/best/tigers/tigersbot/handlers/GptHandler.java b/src/main/java/best/tigers/tigersbot/handlers/GptHandler.java
@@ -2,6 +2,7 @@
 
 import best.tigers.tigersbot.error.MissingEnvironmentVariableException;
 import best.tigers.tigersbot.services.CompletionService;
+import co.elastic.apm.api.ElasticApm;
 import com.pengrad.telegrambot.TelegramBot;
 import com.pengrad.telegrambot.model.Message;
 
@@ -25,10 +26,18 @@ public void handle(Message message) {
             return;
         }
         if (message.text().startsWith("/gpt4 ")) {
-            showTypingIndicator();
-            var prompt = message.text().split("/gpt4 ")[1].strip();
-            var completion = completionService.getAdvancedCompletion(prompt, message.from().username());
-            sendReplyMessage(message, completion);
+            var span = ElasticApm.currentTransaction().startSpan("external", "openai", "chatcompletion");
+            try {
+                showTypingIndicator();
+                var prompt = message.text().split("/gpt4 ")[1].strip();
+                var completion = completionService.getAdvancedCompletion(prompt, message.from().username());
+                sendReplyMessage(message, completion);
+            } catch (Throwable e) {
+                span.captureException(e);
+                throw e;
+            } finally {
+                span.end();
+            }
             return;
         }
         if (message.text().startsWith("/pissed ")) {