fix somethings
This commit is contained in:
@@ -19,6 +19,15 @@ from app.infrastructure.parser.local_chunk_builder import LocalRegulationChunkBu
|
||||
from app.infrastructure.parser.local_document_parser import LocalDocumentParser
|
||||
from app.infrastructure.parser.vector_chunk_builder import AliyunVectorChunkBuilder
|
||||
from app.infrastructure.perception.mock_event_store import MockEventStore
|
||||
from app.application.perception.crawl_service import CrawlService
|
||||
from app.infrastructure.perception.base_event_store import BaseEventStore
|
||||
from app.infrastructure.perception.crawlers.catarc_crawler import CatarcCrawler
|
||||
from app.infrastructure.perception.crawlers.guobiao_crawler import (
|
||||
GuobiaoMandatoryCrawler,
|
||||
GuobiaoRecommendedCrawler,
|
||||
)
|
||||
from app.infrastructure.perception.crawlers.eurlex_crawler import EurlexCrawler
|
||||
from app.infrastructure.perception.llm_pipeline import LlmPipeline
|
||||
from app.infrastructure.session.in_memory_conversation_store import InMemoryConversationStore
|
||||
from app.infrastructure.storage.json_document_processing_store import JsonDocumentProcessingStore
|
||||
from app.infrastructure.storage.json_document_repository import JsonDocumentRepository
|
||||
@@ -293,11 +302,35 @@ def get_agent_conversation_service() -> AgentConversationService:
|
||||
)
|
||||
|
||||
|
||||
@lru_cache
|
||||
def get_event_store() -> BaseEventStore:
|
||||
"""Return event store selected by DOCUMENT_REPOSITORY_BACKEND setting."""
|
||||
if settings.document_repository_backend == "postgres":
|
||||
from app.infrastructure.perception.postgres_event_store import PostgresEventStore
|
||||
return PostgresEventStore()
|
||||
return MockEventStore()
|
||||
|
||||
|
||||
@lru_cache
|
||||
def get_perception_service() -> PerceptionService:
|
||||
"""Return perception service for regulatory intelligence."""
|
||||
return PerceptionService(
|
||||
event_store=MockEventStore(),
|
||||
event_store=get_event_store(),
|
||||
retrieval_service=get_retrieval_service(),
|
||||
)
|
||||
|
||||
|
||||
@lru_cache
|
||||
def get_crawl_service() -> CrawlService:
|
||||
crawlers = {
|
||||
"CATARC": CatarcCrawler(),
|
||||
"国标委·强制性": GuobiaoMandatoryCrawler(),
|
||||
"国标委·推荐性": GuobiaoRecommendedCrawler(),
|
||||
"EUR-Lex": EurlexCrawler(),
|
||||
}
|
||||
return CrawlService(
|
||||
crawlers=crawlers,
|
||||
event_store=get_event_store(),
|
||||
llm_pipeline=LlmPipeline(),
|
||||
retrieval_service=get_retrieval_service(),
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user