mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-07-28 13:53:28 +02:00
Model Touchups (#1887)
This commit is contained in:
@@ -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,
|
||||
|
Reference in New Issue
Block a user