fix a typo in file agent_app_runner.py (#1927)

This commit is contained in:
Liu Peng 2024-01-04 20:39:06 +08:00 committed by GitHub
parent 2757494265
commit 01f9feff9f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 6 additions and 6 deletions

View File

@ -75,7 +75,7 @@ class AgentApplicationRunner(AppRunner):
# reorganize all inputs and template to prompt messages # reorganize all inputs and template to prompt messages
# Include: prompt template, inputs, query(optional), files(optional) # Include: prompt template, inputs, query(optional), files(optional)
# memory(optional) # memory(optional)
prompt_messages, stop = self.originze_prompt_messages( prompt_messages, stop = self.organize_prompt_messages(
app_record=app_record, app_record=app_record,
model_config=app_orchestration_config.model_config, model_config=app_orchestration_config.model_config,
prompt_template_entity=app_orchestration_config.prompt_template, prompt_template_entity=app_orchestration_config.prompt_template,
@ -153,7 +153,7 @@ class AgentApplicationRunner(AppRunner):
# reorganize all inputs and template to prompt messages # reorganize all inputs and template to prompt messages
# Include: prompt template, inputs, query(optional), files(optional) # Include: prompt template, inputs, query(optional), files(optional)
# memory(optional), external data, dataset context(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, app_record=app_record,
model_config=app_orchestration_config.model_config, model_config=app_orchestration_config.model_config,
prompt_template_entity=app_orchestration_config.prompt_template, prompt_template_entity=app_orchestration_config.prompt_template,

View File

@ -50,7 +50,7 @@ class AppRunner:
max_tokens = 0 max_tokens = 0
# get prompt messages without memory and context # 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, app_record=app_record,
model_config=model_config, model_config=model_config,
prompt_template_entity=prompt_template_entity, prompt_template_entity=prompt_template_entity,
@ -107,7 +107,7 @@ class AppRunner:
or (parameter_rule.use_template and parameter_rule.use_template == 'max_tokens')): or (parameter_rule.use_template and parameter_rule.use_template == 'max_tokens')):
model_config.parameters[parameter_rule.name] = 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, model_config: ModelConfigEntity,
prompt_template_entity: PromptTemplateEntity, prompt_template_entity: PromptTemplateEntity,
inputs: dict[str, str], inputs: dict[str, str],

View File

@ -79,7 +79,7 @@ class BasicApplicationRunner(AppRunner):
# organize all inputs and template to prompt messages # organize all inputs and template to prompt messages
# Include: prompt template, inputs, query(optional), files(optional) # Include: prompt template, inputs, query(optional), files(optional)
# memory(optional) # memory(optional)
prompt_messages, stop = self.originze_prompt_messages( prompt_messages, stop = self.organize_prompt_messages(
app_record=app_record, app_record=app_record,
model_config=app_orchestration_config.model_config, model_config=app_orchestration_config.model_config,
prompt_template_entity=app_orchestration_config.prompt_template, prompt_template_entity=app_orchestration_config.prompt_template,
@ -164,7 +164,7 @@ class BasicApplicationRunner(AppRunner):
# reorganize all inputs and template to prompt messages # reorganize all inputs and template to prompt messages
# Include: prompt template, inputs, query(optional), files(optional) # Include: prompt template, inputs, query(optional), files(optional)
# memory(optional), external data, dataset context(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, app_record=app_record,
model_config=app_orchestration_config.model_config, model_config=app_orchestration_config.model_config,
prompt_template_entity=app_orchestration_config.prompt_template, prompt_template_entity=app_orchestration_config.prompt_template,