diff --git a/rag/utils/azure_sas_conn.py b/rag/utils/azure_sas_conn.py index bb0062309f..78edc458c3 100644 --- a/rag/utils/azure_sas_conn.py +++ b/rag/utils/azure_sas_conn.py @@ -51,7 +51,7 @@ class RAGFlowAzureSasBlob: _bucket, fnm, binary = "txtxtxtxt1", "txtxtxtxt1", b"_t@@@1" return self.conn.upload_blob(name=fnm, data=BytesIO(binary), length=len(binary)) - def put(self, bucket, fnm, binary): + def put(self, bucket, fnm, binary, tenant_id=None): for _ in range(3): try: return self.conn.upload_blob(name=fnm, data=BytesIO(binary), length=len(binary)) diff --git a/rag/utils/azure_spn_conn.py b/rag/utils/azure_spn_conn.py index 4cfaa0f3e7..418b3ee6af 100644 --- a/rag/utils/azure_spn_conn.py +++ b/rag/utils/azure_spn_conn.py @@ -68,7 +68,7 @@ class RAGFlowAzureSpnBlob: f.append_data(binary, offset=0, length=len(binary)) return f.flush_data(len(binary)) - def put(self, bucket, fnm, binary): + def put(self, bucket, fnm, binary, tenant_id=None): for _ in range(3): try: f = self.conn.create_file(fnm)