Skip to content

Commit 4587a22

Browse files
Adds skipping non-matching URLs in OpenAITelemetryPlugin (#1206)
1 parent 9c7ad5a commit 4587a22

File tree

2 files changed

+12
-5
lines changed

2 files changed

+12
-5
lines changed

dev-proxy-plugins/Inspection/OpenAITelemetryPlugin.cs

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -136,6 +136,12 @@ private async Task OnRequestAsync(object sender, ProxyRequestArgs e)
136136
{
137137
Logger.LogTrace("OnRequestAsync() called");
138138

139+
if (UrlsToWatch is null || !e.HasRequestUrlMatch(UrlsToWatch))
140+
{
141+
Logger.LogRequest("URL not matched", MessageType.Skipped, new LoggingContext(e.Session));
142+
return;
143+
}
144+
139145
var request = e.Session.HttpClient.Request;
140146
if (request.Method is null ||
141147
!request.Method.Equals("POST", StringComparison.OrdinalIgnoreCase) ||

dev-proxy-plugins/Mocks/OpenAIMockResponsePlugin.cs

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -34,15 +34,16 @@ public override async Task RegisterAsync()
3434

3535
private async Task OnRequestAsync(object sender, ProxyRequestArgs e)
3636
{
37-
if (e.ResponseState.HasBeenSet)
37+
if (UrlsToWatch is null ||
38+
!e.HasRequestUrlMatch(UrlsToWatch))
3839
{
39-
Logger.LogRequest("Response already set", MessageType.Skipped, new LoggingContext(e.Session));
40+
Logger.LogRequest("URL not matched", MessageType.Skipped, new LoggingContext(e.Session));
4041
return;
4142
}
42-
if (UrlsToWatch is null ||
43-
!e.ShouldExecute(UrlsToWatch))
43+
44+
if (e.ResponseState.HasBeenSet)
4445
{
45-
Logger.LogRequest("URL not matched", MessageType.Skipped, new LoggingContext(e.Session));
46+
Logger.LogRequest("Response already set", MessageType.Skipped, new LoggingContext(e.Session));
4647
return;
4748
}
4849

0 commit comments

Comments
 (0)