jack 1 month ago
parent
commit
a484b53eb3
1 changed files with 4 additions and 4 deletions
  1. 4 4
      app/services/file/impl/r2r_file.py

+ 4 - 4
app/services/file/impl/r2r_file.py

@@ -51,7 +51,7 @@ class R2RFileService(OSSFileService):
                     await f.write(content)
 
             # storage.save_from_path(filename=file_key, local_file_path=tmp_file_path)
-            # await r2r.init()
+            await r2r.init()
             fileinfo = await r2r.ingest_file(
                 file_path=tmp_file_path,
                 metadata={"file_key": file_key, "title": file.filename},
@@ -117,7 +117,7 @@ class R2RFileService(OSSFileService):
             r2r.search(query, filters={"file_key": {"$in": file_keys}})
         )
         """
-        # await r2r.init()
+        await r2r.init()
         search_results = await r2r.search(query, filters=filters)
         if not search_results:
             return files
@@ -147,7 +147,7 @@ class R2RFileService(OSSFileService):
         asyncio.run(r2r.init())
         list_results = asyncio.run(r2r.list(ids=ids, offset=offset, limit=limit))
         """
-        # await r2r.init()
+        await r2r.init()
         list_results = await r2r.list(ids=ids, offset=offset, limit=limit)
         return list_results
 
@@ -168,7 +168,7 @@ class R2RFileService(OSSFileService):
             r2r.list_documents(id=id, offset=offset, limit=limit)
         )
         """
-        # await r2r.init()
+        await r2r.init()
         list_results = await r2r.list_documents(id=id, offset=offset, limit=limit)
         return list_results