modify model_type names

modelscope · Mar 13, 2024 · a5b9648 · a5b9648
1 parent f5cc618
commit a5b9648
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/agentscope/models/dashscope_model.py b/src/agentscope/models/dashscope_model.py
@@ -232,7 +232,7 @@ def __call__(
 class DashScopeImageSynthesisWrapper(DashScopeWrapper):
     """The model wrapper for DashScope Image Synthesis API."""
 
-    model_type: str = "dashscope_imagesynthesis"
+    model_type: str = "dashscope_image_synthesis"
 
     def _register_default_metrics(self) -> None:
         # Set monitor accordingly
@@ -333,7 +333,7 @@ def __call__(
 class DashScopeTextEmbeddingWrapper(DashScopeWrapper):
     """The model wrapper for DashScope Text Embedding API."""
 
-    model_type: str = "dashscope_textembedding"
+    model_type: str = "dashscope_embedding"
 
     def _register_default_metrics(self) -> None:
         # Set monitor accordingly