Adding get_user_agent in utils

Author: rajat08

pinecone/core/utils/__init__.py

@@ -6,6 +6,9 @@
 from pathlib import Path
 from typing import List
 
+import requests
+import urllib3
+
 try:
     from pinecone.core.grpc.protos import vector_column_service_pb2
     import numpy as np
@@ -57,3 +60,10 @@ def _generate_request_id() -> str:
 def fix_tuple_length(t, n):
     """Extend tuple t to length n by adding None items at the end of the tuple. Return the new tuple."""
     return t + ((None,) * (n - len(t))) if len(t) < n else t
+
+
+def get_user_agent():
+    client_id = f'python-client-{get_version()}'
+    user_agent_details = {'requests': requests.__version__, 'urllib3': urllib3.__version__}
+    user_agent = '{} ({})'.format(client_id, ', '.join([f'{k}:{v}' for k, v in user_agent_details.items()]))
+    return user_agent

pinecone/index.py

@@ -8,11 +8,9 @@
 from pinecone.core.client import ApiClient, Configuration
 from pinecone.core.utils.sentry import sentry_decorator as sentry
 from .core.client.models import FetchResponse, ProtobufAny, QueryRequest, QueryResponse, QueryVector, RpcStatus, \
-    ScoredVector, SingleQueryResults, DescribeIndexStatsResponse, UpsertRequest, Vector
-from .core.utils.constants import CLIENT_ID
-import requests, urllib3
+    ScoredVector, SingleQueryResults, DescribeIndexStatsResponse, UpsertRequest, UpsertResponse, Vector
 from pinecone.core.client.api.vector_operations_api import VectorOperationsApi
-from pinecone.core.utils import fix_tuple_length
+from pinecone.core.utils import fix_tuple_length, get_user_agent
 
 __all__ = [
     "Index", "FetchResponse", "ProtobufAny", "QueryRequest", "QueryResponse", "QueryVector", "RpcStatus",
@@ -43,8 +41,7 @@ def __init__(self, index_name: str, pool_threads=1):
             **openapi_client_config.server_variables
         }
         super().__init__(configuration=openapi_client_config, pool_threads=pool_threads)
-        user_agent_details = {'requests': requests.__version__, 'urllib3': urllib3.__version__}
-        self.user_agent = '{} ({})'.format(CLIENT_ID, ', '.join([f'{k}:{v}' for k, v in user_agent_details.items()]))
+        self.user_agent = get_user_agent()
         self._vector_api = VectorOperationsApi(self)
 
     @sentry

pinecone/manage.py

@@ -12,9 +12,8 @@
 from pinecone.core.client.configuration import Configuration
 from pinecone.core.client.model.create_request import CreateRequest
 from pinecone.core.client.model.patch_request import PatchRequest
-from pinecone.core.utils.constants import CLIENT_ID
-import requests, urllib3
 from pinecone.core.utils.sentry import sentry_decorator as sentry
+from pinecone.core.utils import get_user_agent
 
 __all__ = [
     "create_index", "delete_index", "describe_index", "list_indexes", "scale_index", "IndexDescription"
@@ -43,8 +42,7 @@ def _get_api_instance():
         **client_config.server_variables
     }
     api_client = ApiClient(configuration=client_config)
-    user_agent_details = {'requests': requests.__version__, 'urllib3': urllib3.__version__}
-    api_client.user_agent = '{} ({})'.format(CLIENT_ID, ', '.join([f'{k}:{v}' for k, v in user_agent_details.items()]))
+    api_client.user_agent = get_user_agent()
     api_instance = IndexOperationsApi(api_client)
     return api_instance