test: Fixed two assertions to work against older versions of openai (…

…#2726)
newrelic · Nov 12, 2024 · 8996fc3 · 8996fc3
1 parent 64fb1dc
commit 8996fc3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/test/versioned/openai/chat-completions.test.js b/test/versioned/openai/chat-completions.test.js
@@ -360,10 +360,11 @@ if (semver.gte(pkgVersion, '4.12.2')) {
       const events = agent.customEventAggregator.events.toArray()
       assert.equal(events.length, 0)
       // we will still record the external segment but not the chat completion
-      assertSegments(tx.trace.root, [
-        'timers.setTimeout',
-        `External/${host}:${port}/chat/completions`
-      ])
+      assertSegments(
+        tx.trace.root,
+        ['timers.setTimeout', `External/${host}:${port}/chat/completions`],
+        { exact: false }
+      )
 
       tx.end()
       end()

diff --git a/test/versioned/openai/embeddings.test.js b/test/versioned/openai/embeddings.test.js
@@ -146,7 +146,7 @@ test('embedding invalid payload errors should be tracked', (t, end) => {
       },
       customAttributes: {
         'http.statusCode': 403,
-        'error.message': '403 You are not allowed to generate embeddings from this model',
+        'error.message': /You are not allowed to generate embeddings from this model/,
         'error.code': null,
         'error.param': null,
         'completion_id': undefined,