Ver Fonte

fix: fix typo (#12034)

Signed-off-by: yihong0618 <zouzou0208@gmail.com>
yihong há 4 meses atrás
pai
commit
6a0ff3686c

+ 0 - 1
api/core/model_runtime/model_providers/huggingface_tei/text_embedding/text_embedding.py

@@ -157,7 +157,6 @@ class HuggingfaceTeiTextEmbeddingModel(TextEmbeddingModel):
                 headers["Authorization"] = f"Bearer {api_key}"
 
             extra_args = TeiHelper.get_tei_extra_parameter(server_url, model, headers)
-            print(extra_args)
             if extra_args.model_type != "embedding":
                 raise CredentialsValidateFailedError("Current model is not a embedding model")
 

+ 5 - 5
api/core/workflow/graph_engine/graph_engine.py

@@ -612,8 +612,8 @@ class GraphEngine:
         max_retries = node_instance.node_data.retry_config.max_retries
         retry_interval = node_instance.node_data.retry_config.retry_interval_seconds
         retries = 0
-        shoudl_continue_retry = True
-        while shoudl_continue_retry and retries <= max_retries:
+        should_continue_retry = True
+        while should_continue_retry and retries <= max_retries:
             try:
                 # run node
                 retry_start_at = datetime.now(UTC).replace(tzinfo=None)
@@ -692,7 +692,7 @@ class GraphEngine:
                                         parent_parallel_id=parent_parallel_id,
                                         parent_parallel_start_node_id=parent_parallel_start_node_id,
                                     )
-                                    shoudl_continue_retry = False
+                                    should_continue_retry = False
                                 else:
                                     yield NodeRunFailedEvent(
                                         error=route_node_state.failed_reason or "Unknown error.",
@@ -706,7 +706,7 @@ class GraphEngine:
                                         parent_parallel_id=parent_parallel_id,
                                         parent_parallel_start_node_id=parent_parallel_start_node_id,
                                     )
-                                shoudl_continue_retry = False
+                                should_continue_retry = False
                             elif run_result.status == WorkflowNodeExecutionStatus.SUCCEEDED:
                                 if node_instance.should_continue_on_error and self.graph.edge_mapping.get(
                                     node_instance.node_id
@@ -758,7 +758,7 @@ class GraphEngine:
                                     parent_parallel_id=parent_parallel_id,
                                     parent_parallel_start_node_id=parent_parallel_start_node_id,
                                 )
-                                shoudl_continue_retry = False
+                                should_continue_retry = False
 
                             break
                         elif isinstance(item, RunStreamChunkEvent):