reflection for checking methods

metagpt/rag/factory.py

@@ -53,6 +53,7 @@ class BaseFactory:
 
 class RetrieverFactory(BaseFactory):
     def __init__(self):
+        # Dynamically add configuration and corresponding instance implementation.
         creators = {
             FAISSRetrieverConfig: self._create_faiss_retriever,
             BM25RetrieverConfig: self._create_bm25_retriever,
@@ -88,6 +89,7 @@ class RetrieverFactory(BaseFactory):
 
 class RankerFactory(BaseFactory):
     def __init__(self):
+        # Dynamically add configuration and corresponding instance implementation.
         creators = {
             LLMRankerConfig: self._create_llm_ranker,
         }

metagpt/rag/retrievers/base.py

@@ -6,6 +6,8 @@ from abc import abstractmethod
 from llama_index.retrievers import BaseRetriever
 from llama_index.schema import BaseNode, NodeWithScore, QueryType
 
+from metagpt.utils.reflection import check_methods
+
 
 class RAGRetriever(BaseRetriever):
     """Inherit from llama_index"""
@@ -23,8 +25,8 @@ class ModifiableRAGRetriever(RAGRetriever):
 
     @classmethod
     def __subclasshook__(cls, C):
-        if any("add_nodes" in B.__dict__ for B in C.__mro__):
-            return True
+        if cls is ModifiableRAGRetriever:
+            return check_methods(C, "add_nodes")
         return NotImplemented
 
     @abstractmethod

metagpt/rag/schema.py

@@ -25,5 +25,6 @@ class LLMRankerConfig(RankerConfig):
     ...
 
 
+# If add new config, it is necessary to add the corresponding instance implementation in rag.factory
 RetrieverConfigType = Union[FAISSRetrieverConfig, BM25RetrieverConfig]
 RankerConfigType = LLMRankerConfig

metagpt/utils/reflection.py

@@ -0,0 +1,20 @@
+"""
+class tools, including method inspection, class attributes, inheritance relationships, etc.
+"""
+
+
+def check_methods(C, *methods):
+    """
+    Check if the class has methods. borrow from _collections_abc.
+    Useful when implementing implicit interfaces, such as defining an abstract class, isinstance can be used for determination without inheritance.
+    """
+    mro = C.__mro__
+    for method in methods:
+        for B in mro:
+            if method in B.__dict__:
+                if B.__dict__[method] is None:
+                    return NotImplemented
+                break
+        else:
+            return NotImplemented
+    return True