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 f568dd1 commit 972a2eeCopy full SHA for 972a2ee
llm-service/app/services/caii/caii.py
@@ -302,9 +302,11 @@ def get_embedding_model(model_name: str) -> BaseEmbedding:
302
303
304
def get_caii_llm_models() -> List[ModelResponse]:
305
- potential_models = get_models_with_task("TEXT_GENERATION")
+ potential_text_models = get_models_with_task("TEXT_GENERATION")
306
+ potential_text_to_text_models = get_models_with_task("TEXT_TO_TEXT_GENERATION")
307
+
308
results: list[Endpoint] = []
- for potential in potential_models:
309
+ for potential in [*potential_text_models, *potential_text_to_text_models]:
310
try:
311
model = get_llm(
312
endpoint=potential,
0 commit comments