fix: replace 'retrival' with 'retrieval' (#657)

Author: Mini256

backend/app/api/admin_routes/knowledge_base/graph/models.py

@@ -50,7 +50,7 @@ class KBKnowledgeGraphRetrievalConfig(BaseModel):
 class KBRetrieveKnowledgeGraphRequest(BaseModel):
     query: str
     llm_id: int
-    retrival_config: KBKnowledgeGraphRetrievalConfig
+    retrieval_config: KBKnowledgeGraphRetrievalConfig
 
 
 ### Experimental

backend/app/api/admin_routes/knowledge_base/graph/routes.py

@@ -207,7 +207,7 @@ def retrieve_kb_knowledge_graph(
         retriever = KnowledgeGraphSimpleRetriever(
             db_session=db_session,
             knowledge_base_id=kb_id,
-            config=request.retrival_config.knowledge_graph,
+            config=request.retrieval_config.knowledge_graph,
         )
         knowledge_graph = retriever.retrieve_knowledge_graph(request.query)
         return KnowledgeGraphRetrievalResult(

backend/app/api/routes/retrieve/models.py

@@ -6,28 +6,28 @@
     KnowledgeGraphRetrieverConfig,
 )
 from app.rag.retrievers.chunk.schema import VectorSearchRetrieverConfig
-from app.rag.retrievers.multiple_knowledge_base import FusionRetrivalBaseConfig
+from app.rag.retrievers.multiple_knowledge_base import FusionRetrievalBaseConfig
 
-# Chunks Retrival
+# Chunks retrieval
 
 
-class ChunkRetrievalConfig(FusionRetrivalBaseConfig):
+class ChunkRetrievalConfig(FusionRetrievalBaseConfig):
     full_documents: Optional[bool] = False
     vector_search: VectorSearchRetrieverConfig
 
 
-class ChunksRetrivalRequest(BaseModel):
+class ChunksRetrievalRequest(BaseModel):
     query: str
     retrieval_config: ChunkRetrievalConfig
 
 
-## Knowledge Graph Retrival
+## Knowledge Graph retrieval
 
 
-class KnowledgeGraphRetrievalConfig(FusionRetrivalBaseConfig):
+class KnowledgeGraphRetrievalConfig(FusionRetrievalBaseConfig):
     knowledge_graph: KnowledgeGraphRetrieverConfig
 
 
-class KnowledgeGraphRetrivalRequest(BaseModel):
+class KnowledgeGraphRetrievalRequest(BaseModel):
     query: str
     retrieval_config: KnowledgeGraphRetrievalConfig

backend/app/api/routes/retrieve/routes.py

@@ -14,7 +14,7 @@
 from app.exceptions import KBNotFound
 from app.rag.retrievers.chunk.schema import ChunksRetrievalResult
 from app.rag.llms.resolver import get_llm_or_default
-from .models import ChunksRetrivalRequest, KnowledgeGraphRetrivalRequest
+from .models import ChunksRetrievalRequest, KnowledgeGraphRetrievalRequest
 
 router = APIRouter()
 logger = logging.getLogger(__name__)
@@ -24,7 +24,7 @@
 def retrieve_chunks(
     db_session: SessionDep,
     user: CurrentSuperuserDep,
-    request: ChunksRetrivalRequest,
+    request: ChunksRetrievalRequest,
 ) -> ChunksRetrievalResult:
     try:
         config = request.retrieval_config
@@ -49,7 +49,7 @@ def retrieve_chunks(
 def retrieve_knowledge_graph(
     db_session: SessionDep,
     user: CurrentSuperuserDep,
-    request: KnowledgeGraphRetrivalRequest,
+    request: KnowledgeGraphRetrievalRequest,
 ) -> KnowledgeGraphRetrievalResult:
     try:
         config = request.retrieval_config

backend/app/rag/chat/config.py

@@ -159,7 +159,7 @@ def get_fast_dspy_lm(self, session: Session) -> dspy.LM:
         llama_llm = self.get_fast_llama_llm(session)
         return get_dspy_lm_by_llama_llm(llama_llm)
 
-    # FIXME: Reranker top_n should be config in the retrival config.
+    # FIXME: Reranker top_n should be config in the retrieval config.
     def get_reranker(
         self, session: Session, top_n: int = None
     ) -> Optional[BaseNodePostprocessor]:

backend/app/rag/rerankers/resolver.py

@@ -88,7 +88,7 @@ def resolve_reranker(
             raise ValueError(f"Got unknown reranker provider: {provider}")
 
 
-# FIXME: Reranker top_n should be config in the retrival config.
+# FIXME: Reranker top_n should be config in the retrieval config.
 def get_default_reranker_model(
     session: Session, top_n: int = None
 ) -> Optional[BaseNodePostprocessor]:

backend/app/rag/retrievers/multiple_knowledge_base.py

@@ -21,7 +21,7 @@
 from app.rag.llms.dspy import get_dspy_lm_by_llama_llm
 
 
-class FusionRetrivalBaseConfig(BaseModel):
+class FusionRetrievalBaseConfig(BaseModel):
     llm_id: Optional[int] = None
     knowledge_base_ids: List[int]
     use_query_decompose: Optional[bool] = None