github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-18 09:01:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

🦍

Browse Source
This commit is contained in:
Chester Curme 2024-12-17 15:21:48 -05:00
parent 897fd22915
commit a9816f7545
1 changed files with 18 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
libs/partners/pinecone/langchain_pinecone/embeddings.py
View File

@ -66,22 +66,25 @@ class PineconeEmbeddings(BaseModel, Embeddings):
protected_namespaces=(), protected_namespaces=(),
) )
async def _initialize_async_client(self) -> aiohttp.ClientSession: async def _get_async_client(self) -> aiohttp.ClientSession:
return aiohttp.ClientSession(
headers={
"Api-Key": self.pinecone_api_key.get_secret_value(),
"Content-Type": "application/json",
"X-Pinecone-API-Version": "2024-10",
}
)
@property
def async_client(self) -> aiohttp.ClientSession:
"""Lazily initialize the async client.""" """Lazily initialize the async client."""
if self._async_client is None: if self._async_client is None:
self._async_client = asyncio.run(self._initialize_async_client()) self._async_client = aiohttp.ClientSession(
headers={
"Api-Key": self.pinecone_api_key.get_secret_value(),
"Content-Type": "application/json",
"X-Pinecone-API-Version": "2024-10",
}
)
return self._async_client return self._async_client
# @property
# def async_client(self) -> aiohttp.ClientSession:
# """Lazily initialize the async client."""
# if self._async_client is None:
# self._async_client = asyncio.run(self._get_async_client())
# return self._async_client
@model_validator(mode="before") @model_validator(mode="before")
@classmethod @classmethod
def set_default_config(cls, values: dict) -> Any: def set_default_config(cls, values: dict) -> Any:
@ -110,7 +113,7 @@ class PineconeEmbeddings(BaseModel, Embeddings):
self._client = client self._client = client
# Ensure async_client is lazily initialized # Ensure async_client is lazily initialized
_ = self.async_client # _ = asyncio.run(self._get_async_client())
return self return self
def _get_batch_iterator(self, texts: List[str]) -> Iterable: def _get_batch_iterator(self, texts: List[str]) -> Iterable:
@ -184,8 +187,7 @@ class PineconeEmbeddings(BaseModel, Embeddings):
"inputs": [{"text": text} for text in texts], "inputs": [{"text": text} for text in texts],
"parameters": parameters, "parameters": parameters,
} }
async with self.async_client.post( client = await self._get_async_client()
"https://api.pinecone.io/embed", json=data async with client.post("https://api.pinecone.io/embed", json=data) as response:
) as response:
response_data = await response.json(content_type=None) response_data = await response.json(content_type=None)
return response_data return response_data