fix: prevent extra prompts from being logged (#14)

We create a new stream when a tool call occurs, but the prompt remains the same from the previous request; let's only record it once

Signed-off-by: Danny Kopping <[email protected]>

Author: dannykopping

bridge_integration_test.go

@@ -651,6 +651,9 @@ func TestAnthropicInjectedTools(t *testing.T) {
 			// Ensure tokens used during injected tool invocation are accounted for.
 			assert.EqualValues(t, 15308, calculateTotalInputTokens(recorderClient.tokenUsages))
 			assert.EqualValues(t, 204, calculateTotalOutputTokens(recorderClient.tokenUsages))
+
+			// Ensure we received exactly one prompt.
+			require.Len(t, recorderClient.userPrompts, 1)
 		})
 	}
 }
@@ -744,6 +747,9 @@ func TestOpenAIInjectedTools(t *testing.T) {
 			// Ensure tokens used during injected tool invocation are accounted for.
 			require.EqualValues(t, 5047, calculateTotalInputTokens(recorderClient.tokenUsages))
 			require.EqualValues(t, 105, calculateTotalOutputTokens(recorderClient.tokenUsages))
+
+			// Ensure we received exactly one prompt.
+			require.Len(t, recorderClient.userPrompts, 1)
 		})
 	}
 }

intercept_anthropic_messages_blocking.go

@@ -91,6 +91,7 @@ func (i *AnthropicMessagesBlockingInterception) ProcessRequest(w http.ResponseWr
 				MsgID:          resp.ID,
 				Prompt:         *prompt,
 			})
+			prompt = nil
 		}
 
 		_ = i.recorder.RecordTokenUsage(ctx, &TokenUsageRecord{

intercept_anthropic_messages_streaming.go

@@ -405,6 +405,7 @@ newStream:
 				MsgID:          message.ID,
 				Prompt:         *prompt,
 			})
+			prompt = nil
 		}
 
 		// Check if the stream encountered any errors.

intercept_openai_chat_blocking.go

@@ -73,6 +73,7 @@ func (i *OpenAIBlockingChatInterception) ProcessRequest(w http.ResponseWriter, r
 				MsgID:          completion.ID,
 				Prompt:         *prompt,
 			})
+			prompt = nil
 		}
 
 		lastUsage := completion.Usage

intercept_openai_chat_streaming.go

@@ -157,6 +157,7 @@ func (i *OpenAIStreamingChatInterception) ProcessRequest(w http.ResponseWriter,
 				MsgID:          processor.getMsgID(),
 				Prompt:         *prompt,
 			})
+			prompt = nil
 		}
 
 		if lastUsage := processor.getLastUsage(); lastUsage.CompletionTokens > 0 {