Fixes related to 1.3 SDK upgrade

langchain-ai · Dec 10, 2024 · cd38500 · cd38500
1 parent e8dd081
commit cd38500
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/libs/langchain-community/src/document_compressors/ibm.ts b/libs/langchain-community/src/document_compressors/ibm.ts
@@ -159,7 +159,7 @@ export class WatsonxRerank
         ? {
             index: document.index,
             relevanceScore: document.score,
-            input: document?.input,
+            input: document?.input.text,
           }
         : {
             index: document.index,

diff --git a/libs/langchain-community/src/llms/ibm.ts b/libs/langchain-community/src/llms/ibm.ts
@@ -298,7 +298,7 @@ export class WatsonxLLM<
     const watsonxCallbacks = this.invocationCallbacks(options);
     if (stream) {
       const textStream = idOrName
-        ? await this.service.deploymentGenerateTextStream({
+        ? this.service.deploymentGenerateTextStream({
             idOrName,
             ...requestOptions,
             parameters: {
@@ -309,7 +309,7 @@ export class WatsonxLLM<
             },
             returnObject: true,
           })
-        : await this.service.generateTextStream(
+        : this.service.generateTextStream(
             {
               input,
               parameters,
@@ -319,7 +319,9 @@ export class WatsonxLLM<
             },
             watsonxCallbacks
           );
-      return textStream;
+      return (await textStream) as AsyncIterable<
+        WatsonXAI.ObjectStreamed<WatsonXAI.TextGenResponse>
+      >;
     } else {
       const textGenerationPromise = idOrName
         ? this.service.deploymentGenerateText(