Skip to content

Commit e70ea91

Browse files
committed
FEAT: add engine ability display
1 parent 3f690f1 commit e70ea91

File tree

1 file changed

+11
-11
lines changed

1 file changed

+11
-11
lines changed

xinference/model/utils.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -462,7 +462,7 @@ def get_engine_params_by_name(
462462
for engine_class in llm_engine_classes:
463463
try:
464464
if hasattr(engine_class, "check_lib"):
465-
lib_available = engine_class.check_lib()
465+
lib_available: bool = engine_class.check_lib() # type: ignore[assignment]
466466
if not lib_available:
467467
error_msg = (
468468
f"Engine {engine_name} library is not available"
@@ -497,11 +497,11 @@ def get_engine_params_by_name(
497497
error_msg = f"Engine {engine_name} is not compatible with current model or environment"
498498

499499
# For unavailable engines, directly return error message string
500-
engine_params[engine_name] = error_msg
500+
engine_params[engine_name] = error_msg # type: ignore[arg-type]
501501

502502
except Exception as e:
503503
# If exception occurs during checking, return error message string
504-
engine_params[engine_name] = (
504+
engine_params[engine_name] = ( # type: ignore[arg-type]
505505
f"Error checking engine {engine_name}: {str(e)}"
506506
)
507507

@@ -549,8 +549,8 @@ def get_engine_params_by_name(
549549
for embedding_engine_class in embedding_engine_classes:
550550
try:
551551
if hasattr(embedding_engine_class, "check_lib"):
552-
lib_available = embedding_engine_class.check_lib()
553-
if not lib_available:
552+
embedding_lib_available: bool = embedding_engine_class.check_lib() # type: ignore[assignment]
553+
if not embedding_lib_available:
554554
error_msg = (
555555
f"Engine {engine_name} library is not available"
556556
)
@@ -586,11 +586,11 @@ def get_engine_params_by_name(
586586
error_msg = f"Engine {engine_name} is not compatible with current model or environment"
587587

588588
# For unavailable engines, directly return error message string
589-
engine_params[engine_name] = error_msg
589+
engine_params[engine_name] = error_msg # type: ignore[arg-type]
590590

591591
except Exception as e:
592592
# If exception occurs during checking, return error message string
593-
engine_params[engine_name] = (
593+
engine_params[engine_name] = ( # type: ignore[arg-type]
594594
f"Error checking engine {engine_name}: {str(e)}"
595595
)
596596

@@ -636,8 +636,8 @@ def get_engine_params_by_name(
636636
for rerank_engine_class in rerank_engine_classes:
637637
try:
638638
if hasattr(rerank_engine_class, "check_lib"):
639-
lib_available = rerank_engine_class.check_lib()
640-
if not lib_available:
639+
rerank_lib_available: bool = rerank_engine_class.check_lib() # type: ignore[assignment]
640+
if not rerank_lib_available:
641641
error_msg = (
642642
f"Engine {engine_name} library is not available"
643643
)
@@ -673,11 +673,11 @@ def get_engine_params_by_name(
673673
error_msg = f"Engine {engine_name} is not compatible with current model or environment"
674674

675675
# For unavailable engines, directly return error message string
676-
engine_params[engine_name] = error_msg
676+
engine_params[engine_name] = error_msg # type: ignore[arg-type]
677677

678678
except Exception as e:
679679
# If exception occurs during checking, return error message string
680-
engine_params[engine_name] = (
680+
engine_params[engine_name] = ( # type: ignore[arg-type]
681681
f"Error checking engine {engine_name}: {str(e)}"
682682
)
683683

0 commit comments

Comments
 (0)