diff --git a/llama_stack/distribution/resolver.py b/llama_stack/distribution/resolver.py index 465b94cb4..9b3812e9e 100644 --- a/llama_stack/distribution/resolver.py +++ b/llama_stack/distribution/resolver.py @@ -345,6 +345,7 @@ def check_protocol_compliance(obj: Any, protocol: Any) -> None: ) missing_methods.append((name, "signature_mismatch")) else: + # Check if the method is actually implemented in the class method_owner = next( (cls for cls in mro if name in cls.__dict__), None ) @@ -352,7 +353,6 @@ def check_protocol_compliance(obj: Any, protocol: Any) -> None: method_owner is None or method_owner.__name__ == protocol.__name__ ): - print(mro) missing_methods.append((name, "not_actually_implemented")) if missing_methods: