From 14b6550aed5a88a629b5f2e5d39b65fb06203305 Mon Sep 17 00:00:00 2001
From: lixin <shinisbigdevil@hotmail.com>
Date: Wed, 22 Jan 2025 22:08:38 +0800
Subject: [PATCH] deepseek.ts bug

---
 src/modules/flow/nodes/llm/models/deepseek.ts | 22 +++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

diff --git a/src/modules/flow/nodes/llm/models/deepseek.ts b/src/modules/flow/nodes/llm/models/deepseek.ts
index c75ef1f..2c53fe2 100644
--- a/src/modules/flow/nodes/llm/models/deepseek.ts
+++ b/src/modules/flow/nodes/llm/models/deepseek.ts
@@ -2,7 +2,26 @@ import { BaseChatModel } from "@langchain/core/language_models/chat_models";
 import { BaseChatModelParams } from "@langchain/core/language_models/chat_models";
 import { BaseMessage, AIMessage } from "@langchain/core/messages";
 import { CallbackManagerForLLMRun } from "@langchain/core/callbacks/manager";
-import { ChatCompletionResponse } from "./types";
+
+interface ChatCompletionResponse {
+  id: string;
+  object: string;
+  created: number;
+  model: string;
+  choices: {
+    message: {
+      role: string;
+      content: string;
+    };
+    finish_reason: string;
+    index: number;
+  }[];
+  usage: {
+    prompt_tokens: number;
+    completion_tokens: number;
+    total_tokens: number;
+  };
+}
 
 /**
  * Deepseek模型参数接口
@@ -80,5 +99,4 @@ export class ChatDeepseek extends BaseChatModel {
   _llmType(): string {
     return "deepseek";
   }
-
 }
\ No newline at end of file