highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-04-02 08:58:11 +02:00
Code Issues Packages Projects Releases Wiki Activity

Model Touchups (#1887)

Browse Source
This commit is contained in:
Yuhong Sun 2024-07-21 12:31:00 -07:00 committed by GitHub
parent 2ba24b1734
commit 0e8ba111c8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
backend/model_server/encoders.py
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,

4
backend/requirements/model_server.txt
View File

@ -10,5 +10,5 @@ transformers==4.39.2
uvicorn==0.21.1
voyageai==0.2.3
openai==1.14.3
cohere==5.5.8
google-cloud-aiplatform==1.58.0
cohere==5.6.1
google-cloud-aiplatform==1.58.0