|
@@ -61,7 +61,8 @@ class MinimaxTextEmbeddingModel(TextEmbeddingModel):
|
|
|
url = f"{self.api_base}?GroupId={group_id}"
|
|
|
headers = {"Authorization": "Bearer " + api_key, "Content-Type": "application/json"}
|
|
|
|
|
|
- data = {"model": "embo-01", "texts": texts, "type": "db"}
|
|
|
+ embedding_type = "db" if input_type == EmbeddingInputType.DOCUMENT else "query"
|
|
|
+ data = {"model": "embo-01", "texts": texts, "type": embedding_type}
|
|
|
|
|
|
try:
|
|
|
response = post(url, headers=headers, data=dumps(data))
|