Skip to content

Commit

Permalink
refactor: rename task_type to task for jina embeddings v3 (langgenius…
Browse files Browse the repository at this point in the history
  • Loading branch information
bwanglzu authored and lau-td committed Oct 23, 2024
1 parent 037bac3 commit 1ec1d79
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ def transform_jina_input_text(model, text):
data = {"model": model, "input": [transform_jina_input_text(model, text) for text in texts]}

if model == "jina-embeddings-v3":
data["task_type"] = "retrieval.passage"
data["task"] = "text-matching"

try:
response = post(url, headers=headers, data=dumps(data))
Expand Down

0 comments on commit 1ec1d79

Please sign in to comment.