Merge pull request #209 from Scale3-Labs/release-6.3.5

 Release 6.3.5

Author: alizenhom

CHANGELOG.md

@@ -1,8 +1,15 @@
 # @langtrase/typescript-sdk
 
+## 6.3.5
+
+### Patch Changes
+
+- add support for `sendMessage` inside `PreviewGenerativeModel` vertexai
+
 ## 6.3.4
 
 ### Patch Changes
+
 - change startChat span name
 
 ## 6.3.3

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langtrase/typescript-sdk",
-  "version": "6.3.4",
+  "version": "6.3.5",
   "description": "A typescript SDK for Langtrace",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

src/examples/vertexai/basic.ts

@@ -3,7 +3,7 @@ import dotenv from 'dotenv'
 import { VertexAI, FunctionDeclarationSchemaType } from '@google-cloud/vertexai'
 
 dotenv.config()
-init({ batch: false, write_spans_to_console: false })
+init({ batch: false, write_spans_to_console: true })
 
 const project = 'vertex-test-454100'
 const location = 'us-central1'
@@ -13,6 +13,8 @@ const vertexAI = new VertexAI({ project, location })
 
 const generativeModel = vertexAI.getGenerativeModel({ model: textModel })
 
+const previewGenerationModel = vertexAI.preview.getGenerativeModel({ model: textModel })
+
 const functionDeclarations = [
   {
     functionDeclarations: [
@@ -83,7 +85,7 @@ export const basicImageVertexAIChat = async (): Promise<void> => {
 
 export const basicVertexAIStartChat = async (): Promise<void> => {
   try {
-    const chat = generativeModel.startChat()
+    const chat = previewGenerationModel.startChat()
     const chatInput = 'capital of France?'
     const result = await chat.sendMessage(chatInput)
     const response = result.response

src/instrumentation/vertexai/instrumentation.ts

@@ -59,7 +59,7 @@ class VertexAIInstrumentation extends InstrumentationBase<any> {
       startChatPatch(
         originalMethod,
         this.tracer,
-        APIS.vertexai.SEND_MESSAGE.METHOD,
+        APIS.vertexai.START_CHAT.METHOD,
         this.instrumentationVersion,
         name,
         moduleVersion,
@@ -70,7 +70,7 @@ class VertexAIInstrumentation extends InstrumentationBase<any> {
       startChatPatch(
         originalMethod,
         this.tracer,
-        APIS.vertexai.SEND_MESSAGE.METHOD,
+        APIS.vertexai.START_CHAT.METHOD,
         this.instrumentationVersion,
         name,
         moduleVersion,
@@ -148,6 +148,28 @@ class VertexAIInstrumentation extends InstrumentationBase<any> {
         moduleVersion,
       ),
     )
+
+    this._wrap(vertexai.ChatSessionPreview.prototype, 'sendMessage', (originalMethod: (...args: any[]) => any) =>
+      generateContentPatch(
+        originalMethod,
+        this.tracer,
+        APIS.vertexai.SEND_MESSAGE.METHOD,
+        this.instrumentationVersion,
+        name,
+        moduleVersion,
+      ),
+    )
+
+    this._wrap(vertexai.ChatSessionPreview.prototype, 'sendMessageStream', (originalMethod: (...args: any[]) => any) =>
+      generateContentPatch(
+        originalMethod,
+        this.tracer,
+        APIS.vertexai.SEND_MESSAGE_STREAM.METHOD,
+        this.instrumentationVersion,
+        name,
+        moduleVersion,
+      ),
+    )
   }
 
   private _unpatch(vertexai: any): void {