fix: allow manual patch for Gemini and VertexAI (#169)

Scale3-Labs · Aug 29, 2024 · 7202a24 · 7202a24
1 parent dd3bd22
commit 7202a24
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,11 @@
 # @langtrase/typescript-sdk
 
+## 5.3.1
+
+### Patch Changes
+
+- Fix allowing manual patch for Gemini and Vertex AI.
+
 ## 5.3.0
 
 ### Minor Changes

diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@langtrase/typescript-sdk",
-  "version": "5.3.0",
+  "version": "5.3.1",
   "description": "A typescript SDK for Langtrace",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

diff --git a/src/instrumentation/gemini/instrumentation.ts b/src/instrumentation/gemini/instrumentation.ts
@@ -25,6 +25,11 @@ class GeminiInstrumentation extends InstrumentationBase<any> {
     super(name, version)
   }
 
+  public manualPatch (gemini: any): void {
+    diag.debug('Manually instrumenting Gemini SDK')
+    this._patch(gemini)
+  }
+
   init (): Array<InstrumentationNodeModuleDefinition<any>> {
     const module = new InstrumentationNodeModuleDefinition<any>(
       '@google/generative-ai',

diff --git a/src/instrumentation/vertexai/instrumentation.ts b/src/instrumentation/vertexai/instrumentation.ts
@@ -26,6 +26,11 @@ class VertexAIInstrumentation extends InstrumentationBase<any> {
     super(name, version)
   }
 
+  public manualPatch (vertexai: any): void {
+    diag.debug('Manually instrumenting Vertex AI SDK')
+    this._patch(vertexai)
+  }
+
   init (): Array<InstrumentationNodeModuleDefinition<any>> {
     const module = new InstrumentationNodeModuleDefinition<any>(
       '@google-cloud/vertexai',