traceNewStreaming rename

Author: pawbana

intercept_anthropic_messages_streaming.go

@@ -135,7 +135,7 @@ newStream:
 			break
 		}
 
-		stream := i.traceNewStreaming(streamCtx, svc, messages) // traces svc.NewStreaming(streamCtx, messages)
+		stream := i.newStream(streamCtx, svc, messages)
 
 		var message anthropic.Message
 		var lastToolName string
@@ -522,7 +522,8 @@ func (s *AnthropicMessagesStreamingInterception) encodeForStream(payload []byte,
 	return buf.Bytes()
 }
 
-func (s *AnthropicMessagesStreamingInterception) traceNewStreaming(ctx context.Context, svc anthropic.MessageService, messages anthropic.MessageNewParams) *ssestream.Stream[anthropic.MessageStreamEventUnion] {
+// newStream traces svc.NewStreaming(streamCtx, messages)
+func (s *AnthropicMessagesStreamingInterception) newStream(ctx context.Context, svc anthropic.MessageService, messages anthropic.MessageNewParams) *ssestream.Stream[anthropic.MessageStreamEventUnion] {
 	_, span := s.tracer.Start(ctx, "Intercept.ProcessRequest.Upstream", trace.WithAttributes(tracing.InterceptionAttributesFromContext(ctx)...))
 	defer span.End()
 

intercept_openai_chat_streaming.go

@@ -116,7 +116,7 @@ func (i *OpenAIStreamingChatInterception) ProcessRequest(w http.ResponseWriter,
 	)
 	for {
 		// TODO add outer loop span (https://github.com/coder/aibridge/issues/67)
-		stream = i.traceNewStreaming(streamCtx, svc) // traces svc.NewStreaming(streamCtx, i.req.ChatCompletionNewParams) call
+		stream = i.newStream(streamCtx, svc)
 		processor := newStreamProcessor(streamCtx, i.logger.Named("stream-processor"), i.getInjectedToolByName)
 
 		var toolCall *openai.FinishedChatCompletionToolCall
@@ -346,7 +346,8 @@ func (i *OpenAIStreamingChatInterception) encodeForStream(payload []byte) []byte
 	return buf.Bytes()
 }
 
-func (i *OpenAIStreamingChatInterception) traceNewStreaming(ctx context.Context, svc openai.ChatCompletionService) *ssestream.Stream[openai.ChatCompletionChunk] {
+// newStream traces svc.NewStreaming(streamCtx, i.req.ChatCompletionNewParams) call
+func (i *OpenAIStreamingChatInterception) newStream(ctx context.Context, svc openai.ChatCompletionService) *ssestream.Stream[openai.ChatCompletionChunk] {
 	_, span := i.tracer.Start(ctx, "Intercept.ProcessRequest.Upstream", trace.WithAttributes(tracing.InterceptionAttributesFromContext(ctx)...))
 	defer span.End()