github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-22 02:50:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

core,langchain,community[patch]: allow langsmith 0.2 (#28598)

Browse Source
This commit is contained in:
Bagatur
2024-12-10 10:50:58 -08:00
committed by GitHub
parent bc4dc7f4b1
commit e6a62d8422
16 changed files with 588 additions and 426 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
libs/community/langchain_community/embeddings/laser.py
View File

@@ -1,4 +1,4 @@
from typing import Any, Dict, List, Optional
from typing import Any, Dict, List, Optional, cast
import numpy as np
from langchain_core.embeddings import Embeddings
@@ -73,7 +73,7 @@ class LaserEmbeddings(BaseModel, Embeddings):
embeddings: np.ndarray
embeddings = self._encoder_pipeline.encode_sentences(texts)
return embeddings.tolist()
return cast(List[List[float]], embeddings.tolist())
def embed_query(self, text: str) -> List[float]:
"""Generate single query text embeddings using LASER.
@@ -86,4 +86,4 @@ class LaserEmbeddings(BaseModel, Embeddings):
"""
query_embeddings: np.ndarray
query_embeddings = self._encoder_pipeline.encode_sentences([text])
return query_embeddings.tolist()[0]
return cast(List[List[float]], query_embeddings.tolist())[0]