Enhance GenAI Integration and Default Model Handling (#4)

* Update README to clarify GENAI_API_URL usage as private option

* Add default GenAI selection handling and Redis state saving for users

Author: jkpe

README.md

@@ -86,7 +86,7 @@ To use the custom GenAI platform with OpenAI-compatible API, export the followin
 
 ```zsh
 export GENAI_API_KEY=<your-genai-api-key>
-export GENAI_API_URL=<your-genai-api-url>  # Optional, set to public: https://docs.digitalocean.com/products/genai-platform/how-to/manage-ai-agent/use-agent/#set-availability. Example: https://agent-<id>.ondigitalocean.app/api/v1
+export GENAI_API_URL=<your-genai-api-url>  # Optional, can be private: https://docs.digitalocean.com/products/genai-platform/how-to/manage-ai-agent/use-agent/#set-availability. Example: https://agent-<id>.ondigitalocean.app/api/v1
 ```
 
 ### Setup Your Local Project

listeners/events/app_home_opened.py

@@ -2,7 +2,9 @@
 from ai.providers import get_available_providers
 from slack_sdk import WebClient
 from state_store.get_redis_user_state import get_redis_user_state
+from state_store.set_redis_user_state import set_redis_user_state
 import sys
+import os
 
 """
 Callback for handling the 'app_home_opened' event. It checks if the event is for the 'home' tab,
@@ -39,13 +41,27 @@ def app_home_opened_callback(event: dict, logger: Logger, client: WebClient):
             print(f"⚠️ No matching option found for model '{model}', using default")
     else:
         print(f"ℹ️ No provider selection found for user: {user_id}")
-        # add an empty option if the user has no previously selected model.
-        options.append(
-            {
-                "text": {"type": "plain_text", "text": "Select a provider", "emoji": True},
-                "value": "null",
-            }
-        )
+        # Check if GENAI_API_URL is set and genai-agent is available
+        genai_api_url = os.environ.get("GENAI_API_URL")
+        if genai_api_url and any(opt["value"].startswith("genai-agent") for opt in options):
+            print(f"🔄 Using genai-agent as default model for user: {user_id}")
+            initial_option = list(filter(lambda x: x["value"].startswith("genai-agent"), options))
+            if initial_option:
+                # Save the default selection to Redis
+                try:
+                    set_redis_user_state(user_id, "genai", "genai-agent")
+                    print(f"✅ Saved default GenAI selection to Redis for user: {user_id}")
+                except Exception as e:
+                    print(f"❌ Error saving default GenAI selection: {e}", file=sys.stderr)
+                    logger.error(f"Error saving default GenAI selection: {e}")
+        else:
+            # add an empty option if the user has no previously selected model.
+            options.append(
+                {
+                    "text": {"type": "plain_text", "text": "Select a provider", "emoji": True},
+                    "value": "null",
+                }
+            )
 
     try:
         client.views_publish(

state_store/get_redis_user_state.py

@@ -1,7 +1,9 @@
 import logging
 import sys
+import os
 from .redis_state_store import RedisStateStore
 from .user_identity import UserIdentity
+from .set_redis_user_state import set_redis_user_state
 
 # Set higher logging level for debugging Redis issues
 logging.basicConfig(level=logging.INFO)
@@ -27,6 +29,21 @@ def get_redis_user_state(user_id: str, is_app_home: bool, redis_url: str = None)
         user_data = redis_store.get_state(user_id)
 
         if not user_data:
+            # Check if GENAI_API_URL is set and use genai-agent as default
+            genai_api_url = os.environ.get("GENAI_API_URL")
+            if genai_api_url:
+                # Set default to genai-agent and save to Redis
+                try:
+                    print(f"🔄 GENAI_API_URL is set, using genai-agent as default for user: {user_id}")
+                    set_redis_user_state(user_id, "genai", "genai-agent", redis_url)
+                    print(f"✅ Saved default GenAI selection to Redis for user: {user_id}")
+                    return "genai", "genai-agent"
+                except Exception as e:
+                    error_msg = f"❌ Error saving default GenAI selection: {e}"
+                    print(error_msg, file=sys.stderr)
+                    logger.error(error_msg)
+            
+            # If GENAI_API_URL not set or error saving default
             if not is_app_home:
                 print(f"❌ No provider selection found for user: {user_id} (non-app-home context)")
                 raise FileNotFoundError("No provider selection found. Please navigate to the App Home and make a selection.")