Update app.py

app.py

@@ -2165,13 +2165,22 @@ class MyScene(Scene):
                                     api_key=token,
                                 )
                                 
+                                # Split model name to get publisher and model
+                                if "/" in model_name:
+                                    publisher, model_id = model_name.split("/", 1)
+                                else:
+                                    # Default publisher if not specified
+                                    publisher = "openai"
+                                    model_id = model_name
+                                
                                 # Prepare parameters based on model configuration
                                 params = {
                                     "messages": [
                                         {"role": "system", "content": "You are a helpful assistant."},
                                         {"role": "user", "content": "Hello, this is a connection test."}
                                     ],
-                                    "model": model_name
+                                    "model": model_id,
+                                    "publisher": publisher
                                 }
                                 
                                 # Add appropriate token parameter
@@ -2190,6 +2199,8 @@ class MyScene(Scene):
                                     st.session_state.ai_models = {
                                         "openai_client": client,
                                         "model_name": model_name,
+                                        "model_id": model_id,
+                                        "publisher": publisher,
                                         "endpoint": "https://models.github.ai/inference",
                                         "last_loaded": datetime.now().isoformat(),
                                         "category": category