Skip to content

Commit a8b1ec5

Browse files
committed
address PR comments
1 parent 682b490 commit a8b1ec5

File tree

9 files changed

+15
-14
lines changed

9 files changed

+15
-14
lines changed

dd-java-agent/instrumentation/openai/build.gradle renamed to dd-java-agent/instrumentation/openai-client/build.gradle

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ apply from: "$rootDir/gradle/java.gradle"
1414
addTestSuiteForDir('latestDepTest', 'test')
1515

1616
dependencies {
17-
compileOnly(group: 'com.openai', name: 'openai-java', version: '2.8.1')
17+
compileOnly(group: 'com.openai', name: 'openai-java', version: '2.8.0')
1818

1919
testImplementation(group: 'com.openai', name: 'openai-java') {
2020
version {
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
package datadog.trace.instrumentation.openai;
1+
package datadog.trace.instrumentation.openaiclient;
22

33
import static datadog.trace.agent.tooling.bytebuddy.matcher.HierarchyMatchers.implementsInterface;
44
import static datadog.trace.agent.tooling.bytebuddy.matcher.NameMatchers.named;
@@ -24,7 +24,7 @@
2424
public class ChatCompletionServiceInstrumentation extends InstrumenterModule.Tracing
2525
implements Instrumenter.ForTypeHierarchy, Instrumenter.HasMethodAdvice {
2626
public ChatCompletionServiceInstrumentation() {
27-
super("openai-client");
27+
super("openai-client", "openai-java", "openai-2.8");
2828
}
2929

3030
@Override
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
package datadog.trace.instrumentation.openai;
1+
package datadog.trace.instrumentation.openaiclient;
22

33
import static datadog.trace.agent.tooling.bytebuddy.matcher.HierarchyMatchers.implementsInterface;
44
import static datadog.trace.agent.tooling.bytebuddy.matcher.NameMatchers.named;
@@ -22,7 +22,7 @@
2222
public class CompletionServiceInstrumentation extends InstrumenterModule.Tracing
2323
implements Instrumenter.ForTypeHierarchy, Instrumenter.HasMethodAdvice {
2424
public CompletionServiceInstrumentation() {
25-
super("openai-client");
25+
super("openai-client", "openai-java", "openai-2.8");
2626
}
2727

2828
@Override
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
package datadog.trace.instrumentation.openai;
1+
package datadog.trace.instrumentation.openaiclient;
22

33
import static datadog.trace.agent.tooling.bytebuddy.matcher.HierarchyMatchers.implementsInterface;
44
import static datadog.trace.agent.tooling.bytebuddy.matcher.NameMatchers.named;
@@ -22,7 +22,7 @@ public class EmbeddingServiceInstrumentation extends InstrumenterModule.Tracing
2222
implements Instrumenter.ForTypeHierarchy, Instrumenter.HasMethodAdvice {
2323

2424
public EmbeddingServiceInstrumentation() {
25-
super("openai", "openai-java");
25+
super("openai", "openai-java", "openai-2.8");
2626
}
2727

2828
@Override
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
package datadog.trace.instrumentation.openai;
1+
package datadog.trace.instrumentation.openaiclient;
22

33
import static datadog.trace.bootstrap.instrumentation.api.AgentTracer.activateSpan;
44
import static datadog.trace.bootstrap.instrumentation.api.AgentTracer.startSpan;

dd-java-agent/instrumentation/openai/src/main/java/datadog/trace/instrumentation/openai/OpenAIClientInfo.java renamed to dd-java-agent/instrumentation/openai-client/src/main/java/datadog/trace/instrumentation/openaiclient/OpenAIClientInfo.java

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
package datadog.trace.instrumentation.openai;
1+
package datadog.trace.instrumentation.openaiclient;
22

33
import com.openai.azure.credential.AzureApiKeyCredential;
44
import com.openai.core.ClientOptions;
@@ -33,10 +33,11 @@ public static OpenAIClientInfo fromClientOptions(ClientOptions options) {
3333
info.setApiKey(((BearerTokenCredential) options.credential()).token());
3434
} else if (options.credential() instanceof AzureApiKeyCredential) {
3535
info.setApiKey(((AzureApiKeyCredential) options.credential()).apiKey());
36-
} else info.setApiKey(null);
36+
} else {
37+
info.setApiKey(null);
38+
}
3739

3840
} catch (Exception e) {
39-
// TODO: returns empty info for now
4041
return info;
4142
}
4243

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
package datadog.trace.instrumentation.openai;
1+
package datadog.trace.instrumentation.openaiclient;
22

33
import static net.bytebuddy.matcher.ElementMatchers.isConstructor;
44
import static net.bytebuddy.matcher.ElementMatchers.takesArgument;
@@ -20,7 +20,7 @@
2020
public class OpenAIClientInstrumentation extends InstrumenterModule.Tracing
2121
implements Instrumenter.ForSingleType, Instrumenter.HasMethodAdvice {
2222
public OpenAIClientInstrumentation() {
23-
super("openai-client");
23+
super("openai-client", "openai-java", "openai-2.8");
2424
}
2525

2626
@Override

settings.gradle

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -430,7 +430,7 @@ include ':dd-java-agent:instrumentation:ognl-appsec'
430430
include ':dd-java-agent:instrumentation:opensearch'
431431
include ':dd-java-agent:instrumentation:opensearch:rest'
432432
include ':dd-java-agent:instrumentation:opensearch:transport'
433-
include ':dd-java-agent:instrumentation:openai'
433+
include ':dd-java-agent:instrumentation:openai-client'
434434
include ':dd-java-agent:instrumentation:opentelemetry:opentelemetry-0.3'
435435
include ':dd-java-agent:instrumentation:opentelemetry:opentelemetry-1.4'
436436
include ':dd-java-agent:instrumentation:opentelemetry:opentelemetry-annotations-1.20'

0 commit comments

Comments
 (0)