fix

Author: reakaleek

src/api/Elastic.Documentation.Api.Core/AskAi/AskAiUsecase.cs

@@ -14,13 +14,6 @@ public class AskAiUsecase(
 {
 	private static readonly ActivitySource AskAiActivitySource = new("Elastic.Documentation.Api.AskAi");
 
-	private static string GetModelNameFromProvider(string provider) => provider switch
-	{
-		"LlmGateway" => "docs_assistant",
-		"AgentBuilder" => "docs-agent",
-		_ => "elastic-docs-rag"
-	};
-
 	public async Task<Stream> AskAi(AskAiRequest askAiRequest, Cancel ctx)
 	{
 		using var activity = AskAiActivitySource.StartActivity("gen_ai.agent");
@@ -35,7 +28,6 @@ public async Task<Stream> AskAi(AskAiRequest askAiRequest, Cancel ctx)
 			tags:
 			[
 				new KeyValuePair<string, object?>("gen_ai.operation.name", "chat"),
-				new KeyValuePair<string, object?>("gen_ai.request.model", GetModelNameFromProvider("Unknown")), // Will be updated by transformer
 				new KeyValuePair<string, object?>("gen_ai.conversation.id", askAiRequest.ThreadId ?? "pending"), // Will be updated when we receive ConversationStart
 				new KeyValuePair<string, object?>("gen_ai.input.messages", $"[{{\"role\":\"user\",\"content\":\"{askAiRequest.Message}\"}}]"),
 				new KeyValuePair<string, object?>("gen_ai.system_instructions", $"[{{\"type\":\"text\",\"content\":\"{AskAiRequest.SystemPrompt}\"}}]")

src/api/Elastic.Documentation.Api.Infrastructure/Adapters/AskAi/StreamTransformerBase.cs

@@ -163,10 +163,13 @@ protected virtual async Task ProcessStreamAsync(PipeReader reader, PipeWriter wr
 
 			if (transformedEvent != null)
 			{
-				// Update parent activity with conversation ID when we receive ConversationStart events
+				// Update parent activity with conversation ID and model info when we receive ConversationStart events
 				if (transformedEvent is AskAiEvent.ConversationStart conversationStart)
 				{
 					_ = (parentActivity?.SetTag("gen_ai.conversation.id", conversationStart.ConversationId));
+					_ = (parentActivity?.SetTag("gen_ai.request.model", GetAgentId()));
+					_ = (parentActivity?.SetTag("gen_ai.agent.name", GetAgentId()));
+					_ = (parentActivity?.SetTag("gen_ai.provider.name", GetAgentProvider()));
 					_ = (activity?.SetTag("gen_ai.conversation.id", conversationStart.ConversationId));
 				}