fix: 修复 Azure OpenAI 回复失败的问题 (#1147)

GaiZhenbiao · Aug 2, 2024 · de01dbd · de01dbd
1 parent 75bbf01
commit de01dbd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/modules/models/base_model.py b/modules/models/base_model.py
@@ -129,6 +129,7 @@ def on_llm_new_token(
  containing content and other information.
  """
  logging.info(f"### CHUNK ###: {chunk}")
+ self.callback(token)
 
 
 class ModelType(Enum):