upsert working?

Author: ssmith-pc

pinecone/pinecone.py

@@ -16,6 +16,7 @@
 if TYPE_CHECKING:
     from pinecone.config import Config, OpenApiConfiguration
     from pinecone.db_data import _Index as Index, _IndexAsyncio as IndexAsyncio
+    from pinecone.repository.data import _Repository as Repository
     from pinecone.db_control.index_host_store import IndexHostStore
     from pinecone.core.openapi.db_control.api.manage_indexes_api import ManageIndexesApi
     from pinecone.db_control.types import CreateIndexForModelEmbedTypedDict, ConfigureIndexEmbed
@@ -518,6 +519,35 @@ def IndexAsyncio(self, host: str, **kwargs) -> "IndexAsyncio":
             **kwargs,
         )
 
+    def Repository(self, name: str = "", host: str = "", **kwargs) -> "Repository":
+        from pinecone.repository.data import _Repository
+
+        if name == "" and host == "":
+            raise ValueError("Either name or host must be specified")
+
+        pt = kwargs.pop("pool_threads", None) or self._pool_threads
+        api_key = self._config.api_key
+        openapi_config = self._openapi_config
+
+        if host != "":
+            check_realistic_host(host)
+
+            # Use host url if it is provided
+            repository_host = normalize_host(host)
+        else:
+            # TODO, get host url from describe_kb using the index name
+            # index_host = self.db.index._get_host(name)
+            raise ValueError("host lookup not yet supported, specify host parameter")
+
+        return _Repository(
+            host=repository_host,
+            api_key=api_key,
+            pool_threads=pt,
+            openapi_config=openapi_config,
+            source_tag=self.config.source_tag,
+            **kwargs,
+        )
+
 
 def check_realistic_host(host: str) -> None:
     """:meta private:

pinecone/repository/data/__init__.py

@@ -0,0 +1,13 @@
+from .repository import Repository, DocumentForUpsert, UpsertDocumentResponse
+
+
+_Repository = Repository  # alias for backwards compatibility
+
+
+__all__ = ["_Repository", "DocumentForUpsert", "UpsertDocumentResponse"]
+
+
+def __getattr__(name):
+    if name in locals():
+        return locals()[name]
+    raise AttributeError(f"module '{__name__}' has no attribute '{name}'")

pinecone/repository/data/repository.py

@@ -66,18 +66,14 @@ def __init__(
             api_version=API_VERSION,
         )
 
-        self._api_client = self._vector_api.api_client
-
-        self._bulk_import_resource = None
-        """ :meta private: """
-
-        self._namespace_resource = None
-        """ :meta private: """
-
-        # Pass the same api_client to the ImportFeatureMixin
-        super().__init__(api_client=self._api_client)
+        self._api_client = self._repository_api.api_client
 
     def upsert(
         self, namespace: str, document: Union[Dict[str, Any], DocumentForUpsert], **kwargs
     ) -> UpsertDocumentResponse:
-        self._repository_api.upsert_document(namespace=namespace, document=document, **kwargs)
+        if isinstance(document, dict):
+            document = DocumentForUpsert(**document)
+
+        return self._repository_api.upsert_document(
+            namespace=namespace, document_for_upsert=document, **kwargs
+        )