We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9d24a83 commit 6eb43e1Copy full SHA for 6eb43e1
packages/mistralai_gcp/src/mistralai_gcp/sdk.py
@@ -26,11 +26,12 @@
26
"mistral-nemo-2407": "mistral-nemo@2407",
27
}
28
29
-def get_model_info(model: str) -> Tuple[str,str]:
+def get_model_info(model: str) -> Tuple[str, str]:
30
# if the model requiers the legacy fomat, use it, else do nothing.
31
- model_id = LEGACY_MODEL_ID_FORMAT.get(model, model)
32
- model = "-".join(model.split("-")[:-1])
33
- return model, model_id
+ if model in LEGACY_MODEL_ID_FORMAT:
+ return "-".join(model.split("-")[:-1]), LEGACY_MODEL_ID_FORMAT[model]
+ else:
34
+ return model, model
35
36
37
0 commit comments