Merge pull request #197 from Scale3-Labs/ali/add-support-for-preview-vertexai

Ali/add support for preview vertexai

Author: alizenhom

CHANGELOG.md

@@ -1,4 +1,8 @@
 # @langtrase/typescript-sdk
+## 6.3.2
+### Patch Changes
+
+- Add Support for preview module inside VertexAI
 
 ## 5.4.0
 

src/instrumentation/vertexai/instrumentation.ts

@@ -55,6 +55,18 @@ class VertexAIInstrumentation extends InstrumentationBase<any> {
       this._unpatch(vertexai)
     }
 
+    this._wrap(vertexai.GenerativeModelPreview.prototype,
+      'generateContent',
+      (originalMethod: (...args: any[]) => any) =>
+        generateContentPatch(originalMethod, this.tracer, APIS.vertexai.GENERATE_CONTENT.METHOD, this.instrumentationVersion, name, moduleVersion)
+    )
+
+    this._wrap(vertexai.GenerativeModelPreview.prototype,
+      'generateContentStream',
+      (originalMethod: (...args: any[]) => any) =>
+        generateContentPatch(originalMethod, this.tracer, APIS.vertexai.GENERATE_CONTENT_STREAM.METHOD, this.instrumentationVersion, name, moduleVersion)
+    )
+
     this._wrap(vertexai.GenerativeModel.prototype,
       'generateContent',
       (originalMethod: (...args: any[]) => any) =>