From 01f9feff9fcb72c49428c8d801bd2626152bca45 Mon Sep 17 00:00:00 2001 From: Liu Peng <134903852+liupeng0913@users.noreply.github.com> Date: Thu, 4 Jan 2024 20:39:06 +0800 Subject: [PATCH] fix a typo in file agent_app_runner.py (#1927) --- api/core/app_runner/agent_app_runner.py | 4 ++-- api/core/app_runner/app_runner.py | 4 ++-- api/core/app_runner/basic_app_runner.py | 4 ++-- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/api/core/app_runner/agent_app_runner.py b/api/core/app_runner/agent_app_runner.py index 344a19d1b7..8f951b7855 100644 --- a/api/core/app_runner/agent_app_runner.py +++ b/api/core/app_runner/agent_app_runner.py @@ -75,7 +75,7 @@ class AgentApplicationRunner(AppRunner): # reorganize all inputs and template to prompt messages # Include: prompt template, inputs, query(optional), files(optional) # memory(optional) - prompt_messages, stop = self.originze_prompt_messages( + prompt_messages, stop = self.organize_prompt_messages( app_record=app_record, model_config=app_orchestration_config.model_config, prompt_template_entity=app_orchestration_config.prompt_template, @@ -153,7 +153,7 @@ class AgentApplicationRunner(AppRunner): # reorganize all inputs and template to prompt messages # Include: prompt template, inputs, query(optional), files(optional) # memory(optional), external data, dataset context(optional) - prompt_messages, stop = self.originze_prompt_messages( + prompt_messages, stop = self.organize_prompt_messages( app_record=app_record, model_config=app_orchestration_config.model_config, prompt_template_entity=app_orchestration_config.prompt_template, diff --git a/api/core/app_runner/app_runner.py b/api/core/app_runner/app_runner.py index 4f247aa8da..bc088e9dc9 100644 --- a/api/core/app_runner/app_runner.py +++ b/api/core/app_runner/app_runner.py @@ -50,7 +50,7 @@ class AppRunner: max_tokens = 0 # get prompt messages without memory and context - prompt_messages, stop = self.originze_prompt_messages( + prompt_messages, stop = self.organize_prompt_messages( app_record=app_record, model_config=model_config, prompt_template_entity=prompt_template_entity, @@ -107,7 +107,7 @@ class AppRunner: or (parameter_rule.use_template and parameter_rule.use_template == 'max_tokens')): model_config.parameters[parameter_rule.name] = max_tokens - def originze_prompt_messages(self, app_record: App, + def organize_prompt_messages(self, app_record: App, model_config: ModelConfigEntity, prompt_template_entity: PromptTemplateEntity, inputs: dict[str, str], diff --git a/api/core/app_runner/basic_app_runner.py b/api/core/app_runner/basic_app_runner.py index 517cd8302a..eed7c1f399 100644 --- a/api/core/app_runner/basic_app_runner.py +++ b/api/core/app_runner/basic_app_runner.py @@ -79,7 +79,7 @@ class BasicApplicationRunner(AppRunner): # organize all inputs and template to prompt messages # Include: prompt template, inputs, query(optional), files(optional) # memory(optional) - prompt_messages, stop = self.originze_prompt_messages( + prompt_messages, stop = self.organize_prompt_messages( app_record=app_record, model_config=app_orchestration_config.model_config, prompt_template_entity=app_orchestration_config.prompt_template, @@ -164,7 +164,7 @@ class BasicApplicationRunner(AppRunner): # reorganize all inputs and template to prompt messages # Include: prompt template, inputs, query(optional), files(optional) # memory(optional), external data, dataset context(optional) - prompt_messages, stop = self.originze_prompt_messages( + prompt_messages, stop = self.organize_prompt_messages( app_record=app_record, model_config=app_orchestration_config.model_config, prompt_template_entity=app_orchestration_config.prompt_template,