Rename dir

.github/workflows/publish_python.yml

@@ -14,9 +14,6 @@ jobs:
       url: https://pypi.org/p/starpoint
     permissions:
       id-token: write  # IMPORTANT: this permission is mandatory for trusted publishing
-    defaults:
-      run:
-        working-directory: ./python
     steps:
       - uses: actions/checkout@v3
         with:
@@ -36,4 +33,4 @@ jobs:
       - name: Publish package distributions to PyPI
         uses: pypa/gh-action-pypi-publish@release/v1
         with:
-          packages-dir: ./python/dist/
+          packages-dir: ./dist/

.github/workflows/test_python.yml

@@ -13,9 +13,6 @@ jobs:
   test-python-sdk:
     name: Python SDK Tests
     runs-on: ubuntu-latest
-    defaults:
-      run:
-        working-directory: ./python
     steps:
       - uses: actions/checkout@v3
       - name: Set up Python

python/starpoint/_utils.py → starpoint/_utils.py

@@ -104,11 +104,10 @@ def _check_collection_identifier_collision(
         raise ValueError(MULTI_COLLECTION_VALUE_ERROR)
 
 
-def _ensure_embedding_dict(embeddings: List[float] | Dict[str, List[float] | int] | None):
+def _ensure_embedding_dict(
+    embeddings: List[float] | Dict[str, List[float] | int] | None
+):
     if isinstance(embeddings, list):
-        dict_embeddings = {
-                "values": embeddings,
-                "dimensionality": len(embeddings)
-            }
+        dict_embeddings = {"values": embeddings, "dimensionality": len(embeddings)}
         return dict_embeddings
     return embeddings

python/starpoint/pandas.py → starpoint/pandas.py

@@ -111,4 +111,3 @@ def update_by_dataframe(
             collection_id=collection_id,
             collection_name=collection_name,
         )
-

python/starpoint/reader.py → starpoint/reader.py

@@ -9,7 +9,7 @@
     _build_header,
     _check_collection_identifier_collision,
     _validate_host,
-    _ensure_embedding_dict
+    _ensure_embedding_dict,
 )
 
 
@@ -89,7 +89,7 @@ def query(
         )
         """
 
-        # check if type of query embeddings is list of float, if so convert to a dict 
+        # check if type of query embeddings is list of float, if so convert to a dict
         query_embeddings = _ensure_embedding_dict(query_embeddings)
 
         request_data = dict(

python/starpoint/writer.py → starpoint/writer.py

@@ -317,7 +317,9 @@ def column_update(
                 "embeddings": embedding,
                 "metadata": document_metadata,
             }
-            for embedding, document_metadata, id in zip(embeddings, document_metadatas, ids)
+            for embedding, document_metadata, id in zip(
+                embeddings, document_metadatas, ids
+            )
         ]
 
         return self.update(

python/tests/test_db.py → tests/test_db.py

@@ -46,7 +46,10 @@ def test_client_insert(mock_writer: MagicMock, mock_reader: MagicMock):
 def test_client_column_insert(mock_writer: MagicMock, mock_reader: MagicMock):
     client = db.Client(api_key=uuid4())
 
-    client.column_insert(embeddings=[{"values": [1.1], "dimensionality": 1}], document_metadatas=[{"mock": "value"}])
+    client.column_insert(
+        embeddings=[{"values": [1.1], "dimensionality": 1}],
+        document_metadatas=[{"mock": "value"}],
+    )
 
     mock_reader.assert_called_once()  # Only called during init
     mock_writer().column_insert.assert_called_once()
@@ -90,6 +93,10 @@ def test_client_update(mock_writer: MagicMock, mock_reader: MagicMock):
 def test_client_column_update(mock_writer: MagicMock, mock_reader: MagicMock):
     client = db.Client(api_key=uuid4())
 
-    client.column_update(ids=["a"], embeddings=[{"values": [1.1], "dimensionality": 1}], document_metadatas=[{"mock": "value"}])
+    client.column_update(
+        ids=["a"],
+        embeddings=[{"values": [1.1], "dimensionality": 1}],
+        document_metadatas=[{"mock": "value"}],
+    )
     mock_reader.assert_called_once()  # Only called during init
     mock_writer().column_update.assert_called_once()

python/tests/test_writer.py → tests/test_writer.py

@@ -161,9 +161,9 @@ def test_writer_insert_SSLError(
 @patch("starpoint.writer.Writer.insert")
 def test_writer_column_insert(insert_mock: MagicMock, mock_writer: writer.Writer):
     test_embeddings = [
-            {"values": [0.88], "dimensionality": 1},
-            {"values": [0.71], "dimensionality": 1}
-        ]
+        {"values": [0.88], "dimensionality": 1},
+        {"values": [0.71], "dimensionality": 1},
+    ]
     test_document_metadatas = [{"mock": "metadata"}, {"mock2": "metadata2"}]
     expected_insert_document = [
         {
@@ -192,8 +192,8 @@ def test_writer_column_insert_collection_id_collection_name_passed_through(
     insert_mock: MagicMock, mock_writer: writer.Writer
 ):
     test_embeddings = [
-            {"values": [0.88], "dimensionality": 1},
-        ]
+        {"values": [0.88], "dimensionality": 1},
+    ]
 
     test_document_metadatas = [{"mock": "metadata"}]
     expected_insert_document = [
@@ -224,9 +224,9 @@ def test_writer_column_insert_shorter_metadatas_length(
     insert_mock: MagicMock, mock_writer: writer.Writer, monkeypatch: MonkeyPatch
 ):
     test_embeddings = [
-            {"values": [0.88], "dimensionality": 1},
-            {"values": [0.71], "dimensionality": 1}
-        ]
+        {"values": [0.88], "dimensionality": 1},
+        {"values": [0.71], "dimensionality": 1},
+    ]
     test_document_metadatas = [{"mock": "metadata"}]
     expected_insert_document = [
         {
@@ -257,8 +257,8 @@ def test_writer_column_insert_shorter_embeddings_length(
     insert_mock: MagicMock, mock_writer: writer.Writer, monkeypatch: MonkeyPatch
 ):
     test_embeddings = [
-            {"values": [0.88], "dimensionality": 1},
-        ]
+        {"values": [0.88], "dimensionality": 1},
+    ]
     test_document_metadatas = [{"mock": "metadata"}, {"mock2": "metadata2"}]
     expected_insert_document = [
         {
@@ -350,9 +350,9 @@ def test_writer_update_SSLError(
 def test_writer_column_update(update_mock: MagicMock, mock_writer: writer.Writer):
     ids = ["a", "b"]
     test_embeddings = [
-            {"values": [0.88], "dimensionality": 1},
-            {"values": [0.71], "dimensionality": 1}
-        ]
+        {"values": [0.88], "dimensionality": 1},
+        {"values": [0.71], "dimensionality": 1},
+    ]
     test_document_metadatas = [{"mock": "metadata"}, {"mock2": "metadata2"}]
     expected_update_document = [
         {
@@ -416,9 +416,9 @@ def test_writer_column_insert_shorter_metadatas_length(
 ):
     ids = ["a", "b"]
     test_embeddings = [
-            {"values": [0.88], "dimensionality": 1},
-            {"values": [0.71], "dimensionality": 1}
-        ]
+        {"values": [0.88], "dimensionality": 1},
+        {"values": [0.71], "dimensionality": 1},
+    ]
     test_document_metadatas = [{"mock": "metadata"}]
     expected_update_document = [
         {
@@ -451,8 +451,8 @@ def test_writer_column_update_shorter_embeddings_length(
 ):
     ids = ["a", "b"]
     test_embeddings = [
-            {"values": [0.88], "dimensionality": 1},
-        ]
+        {"values": [0.88], "dimensionality": 1},
+    ]
     test_document_metadatas = [{"mock": "metadata"}, {"mock2": "metadata2"}]
     expected_update_document = [
         {