diff --git a/kairon/shared/llm/processor.py b/kairon/shared/llm/processor.py index 07388c1a2..44a49c3ed 100644 --- a/kairon/shared/llm/processor.py +++ b/kairon/shared/llm/processor.py @@ -417,7 +417,7 @@ def fetch_llm_metadata(bot: str): """ metadata = Utility.llm_metadata llm_types = metadata.keys() - + final_metadata = {} for llm_type in llm_types: secret = LLMSecret.objects(bot=bot, llm_type=llm_type).first() if not secret: @@ -428,9 +428,11 @@ def fetch_llm_metadata(bot: str): else: models = [] - metadata[llm_type]['properties']['model']['enum'] = models + if models: + metadata[llm_type]['properties']['model']['enum'] = models + final_metadata[llm_type] = metadata[llm_type] - return metadata + return final_metadata @staticmethod def modify_user_message_for_perplexity(user_msg: str, llm_type: str, hyperparameters: Dict) -> str: diff --git a/tests/integration_test/services_test.py b/tests/integration_test/services_test.py index adcec2c70..010f25837 100644 --- a/tests/integration_test/services_test.py +++ b/tests/integration_test/services_test.py @@ -1593,9 +1593,9 @@ def test_get_llm_metadata(): assert "model" in actual["data"]["openai"]["properties"] assert actual["data"]["openai"]["properties"]["model"]["enum"] == ["common_openai_model1", "common_openai_model2"] - assert "anthropic" in actual["data"] - assert "model" in actual["data"]["anthropic"]["properties"] - assert actual["data"]["anthropic"]["properties"]["model"]["enum"] == [] + assert "anthropic" not in actual["data"] + #assert "model" in actual["data"]["anthropic"]["properties"] + #assert actual["data"]["anthropic"]["properties"]["model"]["enum"] == [] LLMSecret.objects.delete()