gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add optional tenant_id parameter to Minio methods

Browse source
This commit is contained in:
Andrea Bugeja 2025-12-04 15:47:45 +01:00
parent 5a50ed5fcc
commit 12316fca3b
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
rag/utils/minio_conn.py
View file

@ -108,7 +108,7 @@ class RAGFlowMinio:
@use_default_bucket
@use_prefix_path
def put(self, bucket, fnm, binary):
def put(self, bucket, fnm, binary, tenant_id=None):
for _ in range(3):
try:
# Note: bucket must already exist - we don't have permission to create buckets
@ -127,7 +127,7 @@ class RAGFlowMinio:
@use_default_bucket
@use_prefix_path
def rm(self, bucket, fnm):
def rm(self, bucket, fnm, tenant_id=None):
try:
self.conn.remove_object(bucket, fnm)
except Exception:
@ -135,7 +135,7 @@ class RAGFlowMinio:
@use_default_bucket
@use_prefix_path
def get(self, bucket, filename):
def get(self, bucket, filename, tenant_id=None):
for _ in range(1):
try:
r = self.conn.get_object(bucket, filename)
@ -148,7 +148,7 @@ class RAGFlowMinio:
@use_default_bucket
@use_prefix_path
def obj_exist(self, bucket, filename):
def obj_exist(self, bucket, filename, tenant_id=None):
try:
if not self.conn.bucket_exists(bucket):
return False
@ -179,7 +179,7 @@ class RAGFlowMinio:
@use_default_bucket
@use_prefix_path
def get_presigned_url(self, bucket, fnm, expires):
def get_presigned_url(self, bucket, fnm, expires, tenant_id=None):
for _ in range(10):
try:
return self.conn.get_presigned_url("GET", bucket, fnm, expires)