Model Touchups (#1887)

This commit is contained in:
Yuhong Sun
2024-07-21 12:31:00 -07:00
committed by GitHub
parent 2ba24b1734
commit 0e8ba111c8
2 changed files with 4 additions and 3 deletions

View File

@@ -97,6 +97,7 @@ class CloudEmbedding:
texts=texts,
model=model,
input_type=embedding_type,
truncate="NONE",
)
return response.embeddings
@@ -126,6 +127,7 @@ class CloudEmbedding:
)
return [embedding.values for embedding in embeddings]
@retry(tries=_RETRY_TRIES, delay=_RETRY_DELAY)
def embed(
self,
*,
@@ -211,7 +213,6 @@ def warm_up_cross_encoders() -> None:
@simple_log_function_time()
@retry(tries=_RETRY_TRIES, delay=_RETRY_DELAY)
def embed_text(
texts: list[str],
text_type: EmbedTextType,