estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix: typo (#2318)

Browse Source
This commit is contained in:
Yeuoly
2024-02-01 13:08:31 +08:00
committed by GitHub
parent d8672796b0
commit f0c9bb7c91
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/model_runtime/model_providers/tongyi/llm/llm.py
View File

@@ -168,7 +168,7 @@ class TongyiLargeLanguageModel(LargeLanguageModel):
return result return result
def _handle_generate_stream_response(self, model: str, credentials: dict, responses: list[Generator], def _handle_generate_stream_response(self, model: str, credentials: dict, responses: Generator,
prompt_messages: list[PromptMessage]) -> Generator: prompt_messages: list[PromptMessage]) -> Generator:
""" """
Handle llm stream response Handle llm stream response
@@ -182,7 +182,7 @@ class TongyiLargeLanguageModel(LargeLanguageModel):
for index, response in enumerate(responses): for index, response in enumerate(responses):
resp_finish_reason = response.output.finish_reason resp_finish_reason = response.output.finish_reason
resp_content = response.output.text resp_content = response.output.text
useage = response.usage usage = response.usage
if resp_finish_reason is None and (resp_content is None or resp_content == ''): if resp_finish_reason is None and (resp_content is None or resp_content == ''):
continue continue
@@ -194,7 +194,7 @@ class TongyiLargeLanguageModel(LargeLanguageModel):
if resp_finish_reason is not None: if resp_finish_reason is not None:
# transform usage # transform usage
usage = self._calc_response_usage(model, credentials, useage.input_tokens, useage.output_tokens) usage = self._calc_response_usage(model, credentials, usage.input_tokens, usage.output_tokens)
yield LLMResultChunk( yield LLMResultChunk(
model=model, model=model,