estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

refactor: rename task_type to task for jina embeddings v3 (#8488)

Browse Source
This commit is contained in:
Wang Bo
2024-09-18 08:53:15 +02:00
committed by GitHub
parent 8dfe8c773a
commit 6f222b49f2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/model_runtime/model_providers/jina/text_embedding/text_embedding.py
View File

@@ -57,7 +57,7 @@ class JinaTextEmbeddingModel(TextEmbeddingModel):
data = {"model": model, "input": [transform_jina_input_text(model, text) for text in texts]} data = {"model": model, "input": [transform_jina_input_text(model, text) for text in texts]}
if model == "jina-embeddings-v3": if model == "jina-embeddings-v3":
data["task_type"] = "retrieval.passage" data["task"] = "text-matching"
try: try:
response = post(url, headers=headers, data=dumps(data)) response = post(url, headers=headers, data=dumps(data))