move imports inline so that they're not required for running the app

llms.py

@@ -1,33 +1,4 @@
 import os
-import logging
-
-import google.generativeai as genai
-from openai import OpenAI
-import anthropic
-
-
-DEFAULT_PROVIDER = "openai"
-DEFAULT_MODEL = "gpt-4o-mini"
-
-# Load API keys
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
-AI_STUDIO_API_KEY = os.getenv("AI_STUDIO_API_KEY")
-ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY")
-
-assert (
-    OPENAI_API_KEY
-), "OPENAI_API_KEY is not set, please set it in your environment variables."
-assert (
-    AI_STUDIO_API_KEY
-), "AI_STUDIO_API_KEY is not set, please set it in your environment variables."
-assert (
-    ANTHROPIC_API_KEY
-), "ANTHROPIC_API_KEY is not set, please set it in your environment variables."
-
-# Initialize clients
-openai_client = OpenAI(api_key=OPENAI_API_KEY)
-genai.configure(api_key=AI_STUDIO_API_KEY)
-anthropic_client = anthropic.Anthropic(api_key=ANTHROPIC_API_KEY)
 
 
 SYSTEM_PROMPT = "You are a programming assistant. You are solving the 2024 advent of code challenge."
@@ -87,6 +58,12 @@ def get_completion(
     Unified function to get completions from various LLM providers.
     """
     if provider == "openai":
+        from openai import OpenAI
+        OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+        assert (
+            OPENAI_API_KEY
+        ), "OPENAI_API_KEY is not set, please set it in your environment variables."
+        openai_client = OpenAI(api_key=OPENAI_API_KEY)
         completion = openai_client.chat.completions.create(
             model=model,
             messages=[
@@ -99,6 +76,15 @@ def get_completion(
         return completion.choices[0].message.content
 
     elif provider == "gemini":
+
+        # Setup
+        import google.generativeai as genai
+        AI_STUDIO_API_KEY = os.getenv("AI_STUDIO_API_KEY")
+        assert (
+            AI_STUDIO_API_KEY
+        ), "AI_STUDIO_API_KEY is not set, please set it in your environment variables."
+        genai.configure(api_key=AI_STUDIO_API_KEY)
+
         model = genai.GenerativeModel(
             model_name=model,
             system_instruction=system_prompt,
@@ -112,6 +98,15 @@ def get_completion(
         return response.text
 
     elif provider == "anthropic":
+
+        # Setup
+        import anthropic
+        ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY")
+        assert (
+            ANTHROPIC_API_KEY
+        ), "ANTHROPIC_API_KEY is not set, please set it in your environment variables."
+        anthropic_client = anthropic.Anthropic(api_key=ANTHROPIC_API_KEY)
+
         response = anthropic_client.messages.create(
             model=model,
             max_tokens=2048,