Create rag-client package

This commit is contained in:
2026-02-20 05:11:24 +00:00
parent d663394106
commit b86dfe7373
16 changed files with 1325 additions and 5 deletions

View File

@@ -2,6 +2,8 @@
from functools import lru_cache
from rag_client import HTTPRAGService, RAGServiceBase
from .config import Settings, settings
from .services import (
ConversationManagerService,
@@ -10,7 +12,6 @@ from .services import (
QuickReplyContentService,
)
from .services.firestore_service import FirestoreService
from .services.rag_service import RAGService
from .services.redis_service import RedisService
@@ -50,9 +51,14 @@ def get_notification_manager() -> NotificationManagerService:
@lru_cache(maxsize=1)
def get_rag_service() -> RAGService:
def get_rag_service() -> RAGServiceBase:
"""Get RAG service instance."""
return RAGService(settings)
return HTTPRAGService(
endpoint_url=settings.rag_endpoint_url,
max_connections=100,
max_keepalive_connections=20,
timeout=30.0,
)
@lru_cache(maxsize=1)