Skip to content

Commit 0bcf9d7

Browse files
committed
refactor
Signed-off-by: Nayeem Kamal <[email protected]>
1 parent 7170ba0 commit 0bcf9d7

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

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

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -202,8 +202,15 @@ public LLMObsSpan startEmbeddingSpan(
202202
DDLLMObsSpan embeddingSpan =
203203
new DDLLMObsSpan(
204204
Tags.LLMOBS_EMBEDDING_SPAN_KIND, spanName, getMLApp(mlApp), sessionId, serviceName);
205-
embeddingSpan.setTag(LLMObsTags.MODEL_PROVIDER, modelProvider);
205+
if (modelName == null || modelName.isEmpty()) {
206+
modelName = CUSTOM_MODEL_VAL;
207+
}
206208
embeddingSpan.setTag(LLMObsTags.MODEL_NAME, modelName);
209+
210+
if (modelProvider == null || modelProvider.isEmpty()) {
211+
modelProvider = CUSTOM_MODEL_VAL;
212+
}
213+
embeddingSpan.setTag(LLMObsTags.MODEL_PROVIDER, modelProvider);
207214
return embeddingSpan;
208215
}
209216

0 commit comments

Comments
 (0)