Add the genkit client header to calls to Vertex Rapid Eval (#10)

firebase · May 2, 2024 · 143b957 · 143b957
1 parent 99cb078
commit 143b957
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/js/plugins/vertexai/src/evaluator_factory.ts b/js/plugins/vertexai/src/evaluator_factory.ts
@@ -15,7 +15,7 @@
  */
 
 import { BaseDataPoint, defineEvaluator } from '@genkit-ai/ai/evaluator';
-import { Action } from '@genkit-ai/core';
+import { Action, GENKIT_CLIENT_HEADER } from '@genkit-ai/core';
 import { runInNewSpan } from '@genkit-ai/core/tracing';
 import { GoogleAuth } from 'google-auth-library';
 import { JSONClient } from 'google-auth-library/build/src/auth/googleauth';
@@ -84,6 +84,9 @@ export class EvaluatorFactory {
           url,
           method: 'POST',
           body: JSON.stringify(request),
+          headers: {
+            'X-Goog-Api-Client': GENKIT_CLIENT_HEADER,
+          },
         });
         metadata.output = response.data;