fix: 解耦LLM

IdeaLeap · Aug 30, 2023 · de2820e · de2820e
1 parent 88119a7
commit de2820e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/package/motion/chain/chain.ts b/package/motion/chain/chain.ts
@@ -8,6 +8,7 @@ import {
   TypeScriptChain,
 } from "@idealeap/gwt";
 export interface chainSchema {
+  llm?: LLM;
   llmSchema?: createLLMSchema;
   chainName?: string;
 }
@@ -27,8 +28,8 @@ export class Chain {
   chainName?: string;
   chain: FunctionChain | TypeScriptChain;
   constructor(params: chainSchema) {
-    const { llmSchema, chainName } = params;
-    this.llm = new LLM(llmSchema || {});
+    const { llmSchema, chainName, llm } = params;
+    this.llm = llm || new LLM(llmSchema || {});
     this.chainName = chainName || "";
     switch (this.chainName) {
       case "typeChat":