Update app.py

app.py

@@ -29,6 +29,9 @@ from pydantic import BaseModel, Field
 import litellm
 from langchain.tools import Tool
 
+LLM._get_litellm_model_name = lambda self, model_name: f"gemini/{model_name}" if not "/" in model_name else model_name
+os.environ["LITELLM_MODEL_DEFAULT_PROVIDER"] = "gemini"
+
 # Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -600,8 +603,7 @@ def create_teaching_plan_crew(file_paths, llm_provider="gemini"):
     tracker.set_placeholder(st.empty())
     
     # Initialize LLM based on provider
-    my_llm = LLM(
-        model='gemini/gemini-2.0-flash',
+    my_llm = LLM(model='gemini/gemini-2.0-flash',
         api_key=os.environ.get("GEMINI_API_KEY")
     )
     
@@ -706,7 +708,9 @@ class BoardPlanAnalyzer:
         self.api_key = api_key
             
         litellm.set_verbose = True
-        
+
+        os.environ["LITELLM_MODEL_DEFAULT_PROVIDER"] = "gemini"
+
         # Create agents
         self.create_agents()
 
@@ -1172,7 +1176,7 @@ if st.session_state.uploaded_files:
                 if st.button("Generate Board Plan", key="board_plan_button"):
                     try:
                         # Select LLM provider
-                        llm_provider = "gemini" if api_key_source == "Google (Gemini)" else "openai"
+                        llm_provider = "gemini"
                         
                         # Initialize the board plan analyzer
                         analyzer = BoardPlanAnalyzer(llm_provider=llm_provider)