Update app.py

app.py

@@ -2165,13 +2165,13 @@ 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)
+                                # For GitHub AI models, ensure the model_name includes the publisher
+                                # If it doesn't have a publisher prefix, add "openai/"
+                                if "/" not in model_name:
+                                    full_model_name = f"openai/{model_name}"
+                                    st.info(f"Using full model name: {full_model_name}")
                                 else:
-                                    # Default publisher if not specified
-                                    publisher = "openai"
-                                    model_id = model_name
+                                    full_model_name = model_name
                                 
                                 # Prepare parameters based on model configuration
                                 params = {
@@ -2179,8 +2179,7 @@ class MyScene(Scene):
                                         {"role": "system", "content": "You are a helpful assistant."},
                                         {"role": "user", "content": "Hello, this is a connection test."}
                                     ],
-                                    "model": model_id,
-                                    "publisher": publisher
+                                    "model": full_model_name
                                 }
                                 
                                 # Add appropriate token parameter
@@ -2198,9 +2197,7 @@ class MyScene(Scene):
                                     # Save working connection to session state
                                     st.session_state.ai_models = {
                                         "openai_client": client,
-                                        "model_name": model_name,
-                                        "model_id": model_id,
-                                        "publisher": publisher,
+                                        "model_name": full_model_name,  # Store the full model name
                                         "endpoint": "https://models.github.ai/inference",
                                         "last_loaded": datetime.now().isoformat(),
                                         "category": category