runtime support get from attribute

Author: JqRrt

backend/modules/observability/lib/otel/consts.go

@@ -35,6 +35,9 @@ const (
 	otelAttributePromptKey      = "cozeloop.prompt_key"
 	otelAttributePromptVersion  = "cozeloop.prompt_version"
 	otelAttributePromptProvider = "cozeloop.prompt_provider"
+
+	// system
+	otelAttributeSystemRuntime = "cozeloop.system_tag_runtime"
 )
 
 // openinference attribute key

backend/modules/observability/lib/otel/open_inference/openinference.go

@@ -67,12 +67,14 @@ func convertModelMsg(msg map[string]interface{}) map[string]interface{} {
 		modelMsg["reasoning_content"] = c
 	}
 
-	// contents
+	// contents or parts
 	var contents []interface{}
-	if c, ok := msg["contents"].([]interface{}); ok && len(c) > 0 {
-		contents = c
-	} else if c, ok := msg["content"].([]interface{}); ok && len(c) > 0 {
-		contents = c
+	partsKey := []string{"contents", "content", "parts"}
+	for _, key := range partsKey {
+		if parts, ok := msg[key].([]interface{}); ok && len(parts) > 0 {
+			contents = parts
+			break
+		}
 	}
 	if len(contents) > 0 {
 		parts := make([]interface{}, 0, len(contents))

backend/modules/observability/lib/otel/otel_convert.go

@@ -398,7 +398,7 @@ func OtelSpanConvertToSendSpan(ctx context.Context, spaceID string, resourceScop
 	// set attributes
 	calOtherAttribute(ctx, span, tagsString, tagsLong, tagsDouble, tagsBool)
 	// set runtime
-	calRuntime(systemTagsString, resourceScopeSpan)
+	calRuntime(systemTagsString, tagsString, resourceScopeSpan)
 
 	result := &LoopSpan{
 		StartTime:        startTimeUnixNanoInt64 / 1000,
@@ -544,12 +544,19 @@ func calOtherAttribute(ctx context.Context, span *Span, tagsString map[string]st
 	}
 }
 
-func calRuntime(systemTagsString map[string]string, resourceScopeSpan *ResourceScopeSpan) {
-	systemTagsString[tracespec.Runtime_] = getRuntime(resourceScopeSpan)
+func calRuntime(systemTagsString map[string]string, tagsString map[string]string, resourceScopeSpan *ResourceScopeSpan) {
+	systemTagsString[tracespec.Runtime_] = getRuntime(tagsString, resourceScopeSpan)
 }
 
-func getRuntime(resourceScopeSpan *ResourceScopeSpan) string {
-	runtime := processRuntime(resourceScopeSpan)
+func getRuntime(tagsString map[string]string, resourceScopeSpan *ResourceScopeSpan) string {
+	if len(tagsString) > 0 {
+		if runtime, ok := tagsString[otelAttributeSystemRuntime]; ok && len(runtime) > 0 {
+			delete(tagsString, otelAttributeSystemRuntime)
+			return runtime
+		}
+	}
+
+	runtime := processRuntimeByScope(resourceScopeSpan)
 	marshalString, err := sonic.MarshalString(runtime)
 	if err != nil {
 		return "" // unexpected
@@ -558,7 +565,7 @@ func getRuntime(resourceScopeSpan *ResourceScopeSpan) string {
 	return marshalString
 }
 
-func processRuntime(resourceScopeSpan *ResourceScopeSpan) *tracespec.Runtime {
+func processRuntimeByScope(resourceScopeSpan *ResourceScopeSpan) *tracespec.Runtime {
 	res := &tracespec.Runtime{
 		Library:      tracespec.VLibOpentelemetry,
 		Scene:        "",

backend/modules/observability/lib/otel/otel_convert_test.go

@@ -1036,7 +1036,7 @@ func TestProcessRuntime(t *testing.T) {
 					panic(r) // Re-panic if unexpected
 				}
 			}()
-			result := processRuntime(tt.resourceScopeSpan)
+			result := processRuntimeByScope(tt.resourceScopeSpan)
 			tt.validate(t, result)
 		})
 	}
@@ -1093,7 +1093,7 @@ func TestGetRuntime(t *testing.T) {
 					panic(r) // Re-panic if unexpected
 				}
 			}()
-			result := getRuntime(tt.resourceScopeSpan)
+			result := getRuntime(nil, tt.resourceScopeSpan)
 			tt.validate(t, result)
 		})
 	}
@@ -1149,7 +1149,7 @@ func TestCalRuntime(t *testing.T) {
 
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			calRuntime(tt.systemTagsString, tt.resourceScopeSpan)
+			calRuntime(tt.systemTagsString, nil, tt.resourceScopeSpan)
 			tt.validate(t, tt.systemTagsString)
 		})
 	}