LiteLLM support

cais/config.py

@@ -11,8 +11,8 @@
 from langchain_google_genai import ChatGoogleGenerativeAI
 from dotenv import load_dotenv
 from langchain_deepseek import ChatDeepSeek
-
 from langchain_together import ChatTogether
+from langchain_litellm import ChatLiteLLM
 
 logger = logging.getLogger(__name__)
 
@@ -82,6 +82,9 @@ def get_llm_client(provider: Optional[str] = None, model_name: Optional[str] = N
             if not api_key:
                 raise ValueError("OPENROUTER_API_KEY not found in environment.")
             return ChatOpenAI(model=model_name, base_url="https://openrouter.ai/api/v1", api_key=api_key, **kwargs)
+
+        elif provider == 'litellm':
+            return ChatLiteLLM(model=model_name, **kwargs)
 
         # Example for Ollama (ensure langchain_community is installed)
         # elif provider == "ollama":