added retrieval and embedding span methods

Author: nayeem-kamal

dd-java-agent/agent-llmobs/src/main/java/datadog/trace/llmobs/LLMObsSystem.java

@@ -189,6 +189,21 @@ public LLMObsSpan startWorkflowSpan(
           Tags.LLMOBS_WORKFLOW_SPAN_KIND, spanName, getMLApp(mlApp), sessionId, serviceName);
     }
 
+    @Override
+    public LLMObsSpan startEmbeddingSpan(
+        String spanName, @Nullable String mlApp, @Nullable String sessionId) {
+      return new DDLLMObsSpan(
+          Tags.LLMOBS_EMBEDDING_SPAN_KIND, spanName, getMLApp(mlApp), sessionId, serviceName);
+    }
+
+    public LLMObsSpan startRetrievalSpan(
+        String spanName,
+        @javax.annotation.Nullable String mlApp,
+        @javax.annotation.Nullable String sessionId) {
+      return new DDLLMObsSpan(
+          Tags.LLMOBS_RETRIEVAL_SPAN_KIND, spanName, getMLApp(mlApp), sessionId, serviceName);
+    }
+
     private String getMLApp(String mlApp) {
       if (mlApp == null || mlApp.isEmpty()) {
         return defaultMLApp;

dd-trace-api/src/main/java/datadog/trace/api/llmobs/LLMObs.java

@@ -46,6 +46,16 @@ public static LLMObsSpan startWorkflowSpan(
     return SPAN_FACTORY.startWorkflowSpan(spanName, mlApp, sessionId);
   }
 
+  public LLMObsSpan startEmbeddingSpan(
+      String spanName, @Nullable String mlApp, @Nullable String sessionId) {
+    return SPAN_FACTORY.startEmbeddingSpan(spanName, mlApp, sessionId);
+  }
+
+  public LLMObsSpan startRetrievalSpan(
+      String spanName, @Nullable String mlApp, @Nullable String sessionId) {
+    return SPAN_FACTORY.startRetrievalSpan(spanName, mlApp, sessionId);
+  }
+
   public static void SubmitEvaluation(
       LLMObsSpan llmObsSpan, String label, String categoricalValue, Map<String, Object> tags) {
     EVAL_PROCESSOR.SubmitEvaluation(llmObsSpan, label, categoricalValue, tags);
@@ -90,6 +100,12 @@ LLMObsSpan startLLMSpan(
 
     LLMObsSpan startWorkflowSpan(
         String spanName, @Nullable String mlApp, @Nullable String sessionId);
+
+    LLMObsSpan startEmbeddingSpan(
+        String spanName, @Nullable String mlApp, @Nullable String sessionId);
+
+    LLMObsSpan startRetrievalSpan(
+        String spanName, @Nullable String mlApp, @Nullable String sessionId);
   }
 
   public interface LLMObsEvalProcessor {

dd-trace-api/src/main/java/datadog/trace/api/llmobs/noop/NoOpLLMObsSpanFactory.java

@@ -35,4 +35,14 @@ public LLMObsSpan startWorkflowSpan(
       String spanName, @Nullable String mlApp, @Nullable String sessionId) {
     return NoOpLLMObsSpan.INSTANCE;
   }
+
+  public LLMObsSpan startEmbeddingSpan(
+      String spanName, @Nullable String mlApp, @Nullable String sessionId) {
+    return NoOpLLMObsSpan.INSTANCE;
+  }
+
+  public LLMObsSpan startRetrievalSpan(
+      String spanName, @Nullable String mlApp, @Nullable String sessionId) {
+    return NoOpLLMObsSpan.INSTANCE;
+  }
 }

internal-api/src/main/java/datadog/trace/bootstrap/instrumentation/api/Tags.java

@@ -166,4 +166,6 @@ public class Tags {
   public static final String LLMOBS_TASK_SPAN_KIND = "task";
   public static final String LLMOBS_AGENT_SPAN_KIND = "agent";
   public static final String LLMOBS_TOOL_SPAN_KIND = "tool";
+  public static final String LLMOBS_EMBEDDING_SPAN_KIND = "embedding";
+  public static final String LLMOBS_RETRIEVAL_SPAN_KIND = "retrieval";
 }