erioon 0.0.9__py3-none-any.whl → 0.1.1__py3-none-any.whl

erioon/client.py

@@ -79,14 +79,12 @@ class ErioonClient:
             raise Exception("Invalid account")
 
     def _update_metadata_fields(self):
-        """
-        Updates internal fields like user_id, database name, and cluster info
-        from login metadata.
-        """
         if self.login_metadata:
             self.user_id = self.login_metadata.get("_id")
-            self.database = self.login_metadata.get("database")
             self.cluster = self.login_metadata.get("cluster")
+            self.database = self.login_metadata.get("database")
+            self.sas_tokens = self.login_metadata.get("sas_tokens", {})
+
 
     def _clear_cached_token(self):
         """
@@ -142,10 +140,6 @@ class ErioonClient:
                 raise e
 
     def _get_database_info(self, db_id):
-        """
-        Sends a POST request to fetch metadata for a given database ID.
-        Returns a `Database` instance initialized with SAS URL and metadata.
-        """
         payload = {"user_id": self.user_id, "db_id": db_id}
         headers = {"Content-Type": "application/json"}
 
@@ -154,8 +148,12 @@ class ErioonClient:
         if response.status_code == 200:
             db_info = response.json()
 
-            container_url = self.login_metadata.get("container_url")
-            sas_token = self.login_metadata.get("sas_token")
+            sas_info = self.sas_tokens.get(db_id)
+            if not sas_info:
+                raise Exception(f"No SAS token info found for database id {db_id}")
+
+            container_url = sas_info.get("container_url")
+            sas_token = sas_info.get("sas_token")
 
             if not container_url or not sas_token:
                 raise Exception("Missing SAS URL components for storage access")
@@ -179,7 +177,7 @@ class ErioonClient:
             except Exception:
                 error_msg = response.text
             raise Exception(error_msg)
-
+    
     def __str__(self):
         """
         Returns user_id or error string when printed.

erioon/collection.py

@@ -236,6 +236,7 @@ class Collection:
         """
         if self._is_read_only():
             return self._read_only_response()
+        self._print_loading()
         return handle_vector(
             user_id_cont=self.user_id,
             database=self.db_id,
@@ -257,7 +258,7 @@ class Collection:
         """
         if self._is_read_only():
             return self._read_only_response()
-
+        self._print_loading()
         return handle_insert_many_vectors(
             user_id_cont=self.user_id,
             database=self.db_id,
@@ -278,7 +279,7 @@ class Collection:
         """
         if self._is_read_only():
             return self._read_only_response()
-            
+        self._print_loading()
         return handle_classify_vector(
             user_id=self.user_id,
             db_id=self.db_id,

{erioon-0.0.9.dist-info → erioon-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: erioon
-Version: 0.0.9
+Version: 0.1.1
 Summary: Erioon SDF for Python
 Author: Zyber Pireci
 Author-email: zyber.pireci@erioon.com

{erioon-0.0.9.dist-info → erioon-0.1.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
 erioon/auth.py,sha256=McktLrdxsWVyxhiFuk0qFokx63RI1yB8_qP3Mo7uZCI,724
-erioon/client.py,sha256=XUdktzV35uPq5dNAMXNN_1h3dva60p_qgJa-HOpzEOQ,6928
-erioon/collection.py,sha256=XEhrffW-1hz9y26C533bzzlgXq3M1j36yyPBl13oz1I,9623
+erioon/client.py,sha256=uxnkd86HQqqIznjby-4wEPavnVd0PxbyHILS9TXxQ3Q,6841
+erioon/collection.py,sha256=EyNHpoOEBBMH29gMy51zzR338z2uI79kl0_ZB7laHpM,9700
 erioon/create.py,sha256=ilx0e3urK0lfYiaM3CSUe3Bf3l7GhHuwZyz0Z6ij0ok,10202
 erioon/database.py,sha256=hj5sgaEDXmItg4aeZlO3MP6_hV3jeS-d9qycEnw7xHQ,2449
 erioon/delete.py,sha256=09fbO-0HKjSvTFqyGz5Oy8osz1EtniXVGI71sMCHNSs,10770
@@ -8,8 +8,8 @@ erioon/functions.py,sha256=LWrqslAok-l9QlMEynT-Pvksy5hwkogWoeK5rJf464A,12479
 erioon/ping.py,sha256=BC0vZiane5YgCc07syZA5dKFcqbq1_Z8BUTmvIGRPcs,1829
 erioon/read.py,sha256=kzzqqRuLtYmlniAnkd2xeWNZm4AwMQ2oAMcpKt6JhcQ,10580
 erioon/update.py,sha256=E3d-dYJWh12bcAHx_vT9NFTscF__hViY3K9ZssN7At4,6060
-erioon-0.0.9.dist-info/LICENSE,sha256=xwnq3DNlZpQyteOK9HvtHRhMdYviXTTaCDljEodFRnQ,569
-erioon-0.0.9.dist-info/METADATA,sha256=UtQc63Q51tmnSCfANjY2mALMxGv24aPABP9bDp7EPhU,715
-erioon-0.0.9.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-erioon-0.0.9.dist-info/top_level.txt,sha256=yjKEg85X5Q5ot46IMML_xukvIGG5YfdrLWcemjalItc,7
-erioon-0.0.9.dist-info/RECORD,,
+erioon-0.1.1.dist-info/LICENSE,sha256=xwnq3DNlZpQyteOK9HvtHRhMdYviXTTaCDljEodFRnQ,569
+erioon-0.1.1.dist-info/METADATA,sha256=X5gjWeQ3dVSzio8GjAYWEjPcAerHeokqRoqZaQ-_PX4,715
+erioon-0.1.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+erioon-0.1.1.dist-info/top_level.txt,sha256=yjKEg85X5Q5ot46IMML_xukvIGG5YfdrLWcemjalItc,7
+erioon-0.1.1.dist-info/RECORD,,