Skip to content

Commit 0067b16

Browse files
authored
fix: refactor all 'or []' and 'or {}' logic to make code more clear (#10883)
Signed-off-by: yihong0618 <[email protected]>
1 parent ec9f622 commit 0067b16

File tree

11 files changed

+20
-26
lines changed

11 files changed

+20
-26
lines changed

api/core/agent/base_agent_runner.py

Lines changed: 4 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -114,16 +114,9 @@ def __init__(
114114
# check if model supports stream tool call
115115
llm_model = cast(LargeLanguageModel, model_instance.model_type_instance)
116116
model_schema = llm_model.get_model_schema(model_instance.model, model_instance.credentials)
117-
if model_schema and ModelFeature.STREAM_TOOL_CALL in (model_schema.features or []):
118-
self.stream_tool_call = True
119-
else:
120-
self.stream_tool_call = False
121-
122-
# check if model supports vision
123-
if model_schema and ModelFeature.VISION in (model_schema.features or []):
124-
self.files = application_generate_entity.files
125-
else:
126-
self.files = []
117+
features = model_schema.features if model_schema and model_schema.features else []
118+
self.stream_tool_call = ModelFeature.STREAM_TOOL_CALL in features
119+
self.files = application_generate_entity.files if ModelFeature.VISION in features else []
127120
self.query = None
128121
self._current_thoughts: list[PromptMessage] = []
129122

@@ -250,7 +243,7 @@ def update_prompt_message_tool(self, tool: Tool, prompt_tool: PromptMessageTool)
250243
update prompt message tool
251244
"""
252245
# try to get tool runtime parameters
253-
tool_runtime_parameters = tool.get_runtime_parameters() or []
246+
tool_runtime_parameters = tool.get_runtime_parameters()
254247

255248
for parameter in tool_runtime_parameters:
256249
if parameter.form != ToolParameter.ToolParameterForm.LLM:

api/core/app/task_pipeline/workflow_cycle_manage.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -381,7 +381,7 @@ def _workflow_start_to_stream_response(
381381
id=workflow_run.id,
382382
workflow_id=workflow_run.workflow_id,
383383
sequence_number=workflow_run.sequence_number,
384-
inputs=workflow_run.inputs_dict or {},
384+
inputs=workflow_run.inputs_dict,
385385
created_at=int(workflow_run.created_at.timestamp()),
386386
),
387387
)
@@ -428,7 +428,7 @@ def _workflow_finish_to_stream_response(
428428
created_by=created_by,
429429
created_at=int(workflow_run.created_at.timestamp()),
430430
finished_at=int(workflow_run.finished_at.timestamp()),
431-
files=self._fetch_files_from_node_outputs(workflow_run.outputs_dict or {}),
431+
files=self._fetch_files_from_node_outputs(workflow_run.outputs_dict),
432432
),
433433
)
434434

api/core/model_runtime/model_providers/cohere/llm/llm.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -691,8 +691,8 @@ def get_customizable_model_schema(self, model: str, credentials: dict) -> AIMode
691691
base_model_schema = cast(AIModelEntity, base_model_schema)
692692

693693
base_model_schema_features = base_model_schema.features or []
694-
base_model_schema_model_properties = base_model_schema.model_properties or {}
695-
base_model_schema_parameters_rules = base_model_schema.parameter_rules or []
694+
base_model_schema_model_properties = base_model_schema.model_properties
695+
base_model_schema_parameters_rules = base_model_schema.parameter_rules
696696

697697
entity = AIModelEntity(
698698
model=model,

api/core/model_runtime/model_providers/openai/llm/llm.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1130,8 +1130,8 @@ def get_customizable_model_schema(self, model: str, credentials: dict) -> AIMode
11301130
base_model_schema = model_map[base_model]
11311131

11321132
base_model_schema_features = base_model_schema.features or []
1133-
base_model_schema_model_properties = base_model_schema.model_properties or {}
1134-
base_model_schema_parameters_rules = base_model_schema.parameter_rules or []
1133+
base_model_schema_model_properties = base_model_schema.model_properties
1134+
base_model_schema_parameters_rules = base_model_schema.parameter_rules
11351135

11361136
entity = AIModelEntity(
11371137
model=model,

api/core/model_runtime/model_providers/openllm/llm/openllm_generate.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,13 +37,14 @@ def __init__(self, content: str, role: str = "user") -> None:
3737
class OpenLLMGenerate:
3838
def generate(
3939
self,
40+
*,
4041
server_url: str,
4142
model_name: str,
4243
stream: bool,
4344
model_parameters: dict[str, Any],
44-
stop: list[str],
45+
stop: list[str] | None = None,
4546
prompt_messages: list[OpenLLMGenerateMessage],
46-
user: str,
47+
user: str | None = None,
4748
) -> Union[Generator[OpenLLMGenerateMessage, None, None], OpenLLMGenerateMessage]:
4849
if not server_url:
4950
raise InvalidAuthenticationError("Invalid server URL")

api/core/tools/tool/tool.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -261,7 +261,7 @@ def get_all_runtime_parameters(self) -> list[ToolParameter]:
261261
"""
262262
parameters = self.parameters or []
263263
parameters = parameters.copy()
264-
user_parameters = self.get_runtime_parameters() or []
264+
user_parameters = self.get_runtime_parameters()
265265
user_parameters = user_parameters.copy()
266266

267267
# override parameters

api/core/tools/tool_engine.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ def agent_invoke(
5555
# check if this tool has only one parameter
5656
parameters = [
5757
parameter
58-
for parameter in tool.get_runtime_parameters() or []
58+
for parameter in tool.get_runtime_parameters()
5959
if parameter.form == ToolParameter.ToolParameterForm.LLM
6060
]
6161
if parameters and len(parameters) == 1:

api/core/tools/utils/configuration.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,7 @@ def _merge_parameters(self) -> list[ToolParameter]:
127127
# get tool parameters
128128
tool_parameters = self.tool_runtime.parameters or []
129129
# get tool runtime parameters
130-
runtime_parameters = self.tool_runtime.get_runtime_parameters() or []
130+
runtime_parameters = self.tool_runtime.get_runtime_parameters()
131131
# override parameters
132132
current_parameters = tool_parameters.copy()
133133
for runtime_parameter in runtime_parameters:

api/services/app_service.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -341,7 +341,7 @@ def get_app_meta(self, app_model: App) -> dict:
341341
if not app_model_config:
342342
return meta
343343

344-
agent_config = app_model_config.agent_mode_dict or {}
344+
agent_config = app_model_config.agent_mode_dict
345345

346346
# get all tools
347347
tools = agent_config.get("tools", [])

api/services/tools/tools_transform_service.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -242,7 +242,7 @@ def tool_to_user_tool(
242242
# get tool parameters
243243
parameters = tool.parameters or []
244244
# get tool runtime parameters
245-
runtime_parameters = tool.get_runtime_parameters() or []
245+
runtime_parameters = tool.get_runtime_parameters()
246246
# override parameters
247247
current_parameters = parameters.copy()
248248
for runtime_parameter in runtime_parameters:

api/services/website_service.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,8 +51,8 @@ def crawl_url(cls, args: dict) -> dict:
5151
excludes = options.get("excludes").split(",") if options.get("excludes") else []
5252
params = {
5353
"crawlerOptions": {
54-
"includes": includes or [],
55-
"excludes": excludes or [],
54+
"includes": includes,
55+
"excludes": excludes,
5656
"generateImgAltText": True,
5757
"limit": options.get("limit", 1),
5858
"returnOnlyUrls": False,

0 commit comments

Comments
 (0)