Fix/app runner typo (langgenius#2661)

HuberyHuV1 · Mar 4, 2024 · 51618fa · 51618fa
1 parent ee8ae23
commit 51618fa
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/api/core/app_runner/app_runner.py b/api/core/app_runner/app_runner.py
@@ -84,7 +84,7 @@ def get_pre_calculate_rest_tokens(self, app_record: App,
 
         return rest_tokens
 
-    def recale_llm_max_tokens(self, model_config: ModelConfigEntity,
+    def recalc_llm_max_tokens(self, model_config: ModelConfigEntity,
                               prompt_messages: list[PromptMessage]):
         # recalc max_tokens if sum(prompt_token +  max_tokens) over model token limit
         model_type_instance = model_config.provider_model_bundle.model_type_instance

diff --git a/api/core/app_runner/basic_app_runner.py b/api/core/app_runner/basic_app_runner.py
@@ -181,7 +181,7 @@ def run(self, application_generate_entity: ApplicationGenerateEntity,
             return
 
         # Re-calculate the max tokens if sum(prompt_token +  max_tokens) over model token limit
-        self.recale_llm_max_tokens(
+        self.recalc_llm_max_tokens(
             model_config=app_orchestration_config.model_config,
             prompt_messages=prompt_messages
         )

diff --git a/api/core/features/assistant_cot_runner.py b/api/core/features/assistant_cot_runner.py
@@ -131,7 +131,7 @@ def increase_usage(final_llm_usage_dict: dict[str, LLMUsage], usage: LLMUsage):
             )
 
             # recale llm max tokens
-            self.recale_llm_max_tokens(self.model_config, prompt_messages)
+            self.recalc_llm_max_tokens(self.model_config, prompt_messages)
             # invoke model
             chunks: Generator[LLMResultChunk, None, None] = model_instance.invoke_llm(
                 prompt_messages=prompt_messages,

diff --git a/api/core/features/assistant_fc_runner.py b/api/core/features/assistant_fc_runner.py
@@ -106,7 +106,7 @@ def increase_usage(final_llm_usage_dict: dict[str, LLMUsage], usage: LLMUsage):
             )
 
             # recale llm max tokens
-            self.recale_llm_max_tokens(self.model_config, prompt_messages)
+            self.recalc_llm_max_tokens(self.model_config, prompt_messages)
             # invoke model
             chunks: Union[Generator[LLMResultChunk, None, None], LLMResult] = model_instance.invoke_llm(
                 prompt_messages=prompt_messages,