diff --git a/ai-hub/app/core/rag_service.py b/ai-hub/app/core/rag_service.py index 77e1cc0..dd0a7fb 100644 --- a/ai-hub/app/core/rag_service.py +++ b/ai-hub/app/core/rag_service.py @@ -5,6 +5,7 @@ from app.core.vector_store import FaissVectorStore # Import the concrete vector store implementation from app.db import models # Import the database models from app.core.retrievers import Retriever # Import the retriever base class +from app.core import llm_providers # Import the retriever base class # --- Placeholder/Helper functions and classes for dependencies --- @@ -22,8 +23,9 @@ A placeholder function to retrieve the correct LLM provider. This resolves the AttributeError from the test suite. """ + print(f"Retrieving LLM provider for model: {model_name}") - return LLMProvider() + return llm_providers.get_llm_provider(model_name) # --- Main RAG Service Class ---