add: Async with

QuivrHQ · Nov 25, 2024 · 1b91281 · 1b91281
1 parent 5bc430a
commit 1b91281
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/core/quivr_core/processor/implementations/megaparse_processor.py b/core/quivr_core/processor/implementations/megaparse_processor.py
@@ -75,8 +75,9 @@ def processor_metadata(self):
 
     async def process_file_inner(self, file: QuivrFile) -> list[Document]:
         logger.info(f"Uploading file {file.path} to MegaParse")
-        client = MegaParseNATSClient(ClientNATSConfig())
-        response = await client.parse_file(file=file.path)
+        async with MegaParseNATSClient(ClientNATSConfig()) as client:
+            response = await client.parse_file(file=file.path)
+
         logger.info(f"File :  {response}")
         document = Document(
             page_content=response,