fix: transfrom wrong format in json response from Anthropic (#897)

janhq · Jul 19, 2024 · dc57195 · dc57195
1 parent 10bfad2
commit dc57195
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cortex-js/src/extensions/anthropic.engine.ts b/cortex-js/src/extensions/anthropic.engine.ts
@@ -54,7 +54,7 @@ export default class AnthropicEngineExtension extends OAIEngineExtension {
     return _.pick(data, ['messages', 'model', 'stream', 'max_tokens']);
   };
 
-  transformResponse = (data: any): string => {
+  transformResponse = (data: any) => {
     // handling stream response
     if (typeof data === 'string' && data.trim().length === 0) {
       return '';
@@ -82,15 +82,15 @@ export default class AnthropicEngineExtension extends OAIEngineExtension {
     }
     // non-stream response
     if (data.content && data.content.length > 0 && data.content[0].text) {
-      return JSON.stringify({
+      return {
         choices: [
           {
             delta: {
               content: data.content[0].text,
             },
           },
         ],
-      });
+      };
     }
 
     console.error('Invalid response format:', data);