🤖 Merge PR DefinitelyTyped#70742 feat(newrelic-node): Add type def ne…

…wrelic node by @saileshtummepalli

Co-authored-by: saileshtummepalli <[email protected]>

types/newrelic/index.d.ts

@@ -552,3 +552,23 @@ export interface TraceMetadata {
      */
     spanId?: string | undefined;
 }
+/**
+ * Run a function with the passed in LLM context as the active context and return its return value.
+ *
+ * See documentation for `withLlmCustomAttributes` for more information on setting custom attributes.
+ */
+
+export function withLlmCustomAttributes<T>(
+    attrs: Record<string, number | string | boolean>,
+    cb: (...args: any[]) => T,
+): T;
+
+/**
+ * Registers a callback which will be used for calculating token counts on Llm events when they are not available.
+ *
+ * See documentation for `setLlmTokenCountCallback` for more information on setting custom attributes.
+ */
+
+export function setLlmTokenCountCallback<T>(
+    cb: (...args: any[]) => T,
+): T;

types/newrelic/newrelic-tests.ts

@@ -144,3 +144,6 @@ newrelic.setLambdaHandler((event: unknown, context: unknown) => ({ statusCode: 2
 newrelic.setLambdaHandler({ some: "object" });
 
 newrelic.obfuscateSql("SELECT * FROM USERS", "postgres"); // $ExpectType string
+newrelic.withLlmCustomAttributes({ task: "question" }, () => void 0);
+
+newrelic.setLlmTokenCountCallback(() => void 0);