DF input verification

Verify that the input DF is indeed a pandas dataframe

Author: igiloh-pinecone

pinecone/core/grpc/index_grpc.py

@@ -417,22 +417,30 @@ def _upsert_batch(self,
 
     def upsert_dataframe(self,
                          df,
-                         namespase: str = None,
+                         namespace: str = None,
                          batch_size: int = 500,
+                         use_async_requests: bool = True,
                          show_progress: bool = True) -> None:
         """Upserts a dataframe into the index.
 
         Args:
             df: A pandas dataframe with the following columns: id, vector, and metadata.
             namespace: The namespace to upsert into.
             batch_size: The number of rows to upsert in a single batch.
+            use_async_requests: Whether to upsert multiple requests at the same time using asynchronous request mechanism.
+                                Set to `False`
             show_progress: Whether to show a progress bar.
         """
-        if find_spec("pandas") is None:
-            raise ImportError("pandas not found. Please install pandas to use this method.")
+        try:
+            import pandas as pd
+        except ImportError:
+            raise RuntimeError("The `pandas` package is not installed. Please install pandas to use `upsert_from_dataframe()`")
+
+        if not isinstance(df, pd.DataFrame):
+            raise ValueError(f"Only pandas dataframes are supported. Found: {type(df)}")
 
         async_results = [
-            self.upsert(vectors=chunk, namespace=namespase, async_req=True)
+            self.upsert(vectors=chunk, namespace=namespace, async_req=True)
             for chunk in tqdm(self._iter_dataframe(df, batch_size=batch_size),
                               total=len(df) // batch_size, disable=not show_progress)
         ]

pinecone/index.py

@@ -195,7 +195,7 @@ def _vector_transform(item: Union[Vector, Tuple]):
 
     def upsert_dataframe(self,
                          df,
-                         namespase: str = None,
+                         namespace: str = None,
                          batch_size: int = 500,
                          show_progress: bool = True) -> None:
         """Upserts a dataframe into the index.
@@ -206,13 +206,18 @@ def upsert_dataframe(self,
             batch_size: The number of rows to upsert in a single batch.
             show_progress: Whether to show a progress bar.
         """
-        if find_spec("pandas") is None:
-            raise ImportError("pandas not found. Please install pandas to use this method.")
+        try:
+            import pandas as pd
+        except ImportError:
+            raise RuntimeError("The `pandas` package is not installed. Please install pandas to use `upsert_from_dataframe()`")
+
+        if not isinstance(df, pd.DataFrame):
+            raise ValueError(f"Only pandas dataframes are supported. Found: {type(df)}")
 
         pbar = tqdm(total=len(df), disable=not show_progress)
         for i in range(0, len(df), batch_size):
             batch = df.iloc[i:i + batch_size].to_dict(orient="records")
-            self.upsert(batch, namespace=namespase)
+            self.upsert(batch, namespace=namespace)
             pbar.update(len(batch))
 
     @validate_and_convert_errors