pinecone[patch]: source tag (#19739)

langchain-ai · Apr 2, 2024 · d5a2ff5 · d5a2ff5
1 parent 8638029
commit d5a2ff5
Show file tree

Hide file tree

Showing 3 changed files with 190 additions and 126 deletions.
diff --git a/libs/partners/pinecone/langchain_pinecone/vectorstores.py b/libs/partners/pinecone/langchain_pinecone/vectorstores.py
@@ -100,7 +100,7 @@ def __init__(
                 )
 
             # needs
-            client = PineconeClient(api_key=_pinecone_api_key)
+            client = PineconeClient(api_key=_pinecone_api_key, source_tag="langchain")
             self._index = client.Index(_index_name)
 
     @property
@@ -370,7 +370,9 @@ def get_pinecone_index(
         Returns:
             Pinecone Index instance."""
         _pinecone_api_key = pinecone_api_key or os.environ.get("PINECONE_API_KEY") or ""
-        client = PineconeClient(api_key=_pinecone_api_key, pool_threads=pool_threads)
+        client = PineconeClient(
+            api_key=_pinecone_api_key, pool_threads=pool_threads, source_tag="langchain"
+        )
         indexes = client.list_indexes()
         index_names = [i.name for i in indexes.index_list["indexes"]]