Merge pull request #3 from tomkat-cr/GS-154_post_hackathon_fixes

GS-154 - Fix available providers and "status:" attribute
tomkat-cr · Nov 20, 2024 · e0e2d9e · e0e2d9e
2 parents 501cbdf + 82ba5ba
commit e0e2d9e
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 13 deletions.
diff --git a/config/app_config.json b/config/app_config.json
@@ -26,36 +26,36 @@
     "REFINE_VIDEO_PROMPT_TEXT": "[refine_video_prompt.txt]",
     "REFINE_LLM_PROMPT_TEXT": "[refine_llm_prompt.txt]",
     "LLM_PROVIDERS": {
+        "together_ai": {
+            "requirements": ["TOGETHER_AI_API_KEY"],
+            "active": true
+        },
         "openai": {
             "requirements": ["OPENAI_API_KEY"]
         },
         "groq": {
             "requirements": ["GROQ_API_KEY"],
-            "active:": true
+            "active": true
         },
         "ollama": {
             "requirements": [],
-            "active:": true
+            "active": false
         },
         "rhymes": {
             "requirements": ["RHYMES_ARIA_API_KEY"],
-            "active:": true
+            "active": true
         },
         "nvidia": {
             "requirements": ["NVIDIA_API_KEY"],
-            "active:": true
+            "active": true
         },
         "huggingface": {
             "requirements": ["HUGGINGFACE_API_KEY"],
-            "active:": true
-        },
-        "together_ai": {
-            "requirements": ["TOGETHER_AI_API_KEY"],
-            "active:": true
+            "active": true
         },
         "xai": {
             "requirements": ["XAI_API_KEY"],
-            "active:": true
+            "active": true
         }
     },
     "NO_SYSTEM_PROMPT_ALLOWED_PROVIDERS": [
@@ -104,11 +104,11 @@
     "TEXT_TO_IMAGE_PROVIDERS": {
         "huggingface": {
             "requirements": ["HUGGINGFACE_API_KEY"],
-            "active:": true
+            "active": true
         },
         "openai": {
             "requirements": ["OPENAI_API_KEY"],
-            "active:": true
+            "active": true
         }
     },
     "TEXT_TO_IMAGE_AVAILABLE_MODELS": {
@@ -124,7 +124,7 @@
     "TEXT_TO_VIDEO_PROVIDERS": {
         "rhymes": {
             "requirements": ["RHYMES_ALLEGRO_API_KEY"],
-            "active:": true
+            "active": true
         }
     },
     "TEXT_TO_VIDEO_AVAILABLE_MODELS": {

diff --git a/lib/codegen_streamlit_lib.py b/lib/codegen_streamlit_lib.py
@@ -1092,6 +1092,12 @@ def get_available_ai_providers(
             param_values = os.environ
         result = []
         for model_name, model_attr in self.get_par_value(param_name).items():
+            # log_debug(f"get_available_ai_providers | "
+            #           '\nmodel_attr.get("active", True): '
+            #           f'{model_attr.get("active", True)}'
+            #           f"\nmodel_name: {model_name} | "
+            #           f"\nmodel_attr: {model_attr}",
+            #           debug=DEBUG)
             if not model_attr.get("active", True):
                 continue
             model_to_add = model_name