|
|
@@ -5,10 +5,6 @@ from attrs import frozen
|
|
|
|
|
|
from de_quoi_parle_le_monde.http import HttpClient
|
|
|
from de_quoi_parle_le_monde.storage import Storage
|
|
|
-from de_quoi_parle_le_monde.workers.embeddings import (
|
|
|
- EmbeddingsJob,
|
|
|
- EmbeddingsWorker,
|
|
|
-)
|
|
|
from de_quoi_parle_le_monde.similarity_search import SimilaritySearch
|
|
|
|
|
|
|
|
|
@@ -21,25 +17,9 @@ class Application:
|
|
|
async def run(self):
|
|
|
await asyncio.gather(
|
|
|
self._run_similarity_index(),
|
|
|
- self._run_embeddings_worker(),
|
|
|
)
|
|
|
logger.info("Will quit now..")
|
|
|
|
|
|
- async def _run_embeddings_worker(self):
|
|
|
- logger.info("Starting embeddings service..")
|
|
|
- jobs = await EmbeddingsJob.create(self.storage)
|
|
|
- if jobs:
|
|
|
- loop = asyncio.get_event_loop()
|
|
|
- worker = await loop.run_in_executor(
|
|
|
- None,
|
|
|
- EmbeddingsWorker.create,
|
|
|
- self.storage,
|
|
|
- "dangvantuan/sentence-camembert-large",
|
|
|
- )
|
|
|
- await worker.run(jobs)
|
|
|
-
|
|
|
- logger.info("Embeddings service exiting")
|
|
|
-
|
|
|
async def _run_similarity_index(self):
|
|
|
logger.info("Starting index..")
|
|
|
try:
|