From fa511f69778b10e7c03949b36e1ec451b16cceec Mon Sep 17 00:00:00 2001
From: RheagalFire <arishalam121@gmail.com>
Date: Tue, 26 May 2026 02:54:28 +0530
Subject: [PATCH] feat: add LiteLLM as custom model provider with icon

Signed-off-by: RheagalFire <arishalam121@gmail.com>
---
 packages/icons/src/lib/Icons/LiteLLM.svelte |  3 +++
 packages/icons/src/lib/main.ts              |  7 +++++--
 packages/types/src/ai.types.ts              | 11 ++++++++++-
 3 files changed, 18 insertions(+), 3 deletions(-)
 create mode 100644 packages/icons/src/lib/Icons/LiteLLM.svelte

diff --git a/packages/icons/src/lib/Icons/LiteLLM.svelte b/packages/icons/src/lib/Icons/LiteLLM.svelte
new file mode 100644
index 000000000..9cfda7c9b
--- /dev/null
+++ b/packages/icons/src/lib/Icons/LiteLLM.svelte
@@ -0,0 +1,3 @@
+<svg viewBox="0 0 24 24" fill="currentColor" xmlns="http://www.w3.org/2000/svg" {...$$restProps}>
+  <text x="12" y="18" font-size="20" text-anchor="middle">🚅</text>
+</svg>
diff --git a/packages/icons/src/lib/main.ts b/packages/icons/src/lib/main.ts
index 5cdb8a5f9..20fc1e42b 100644
--- a/packages/icons/src/lib/main.ts
+++ b/packages/icons/src/lib/main.ts
@@ -151,6 +151,7 @@ import LinkExternal from './Icons/LinkExternal.svelte'
 import Ollama from './Icons/Ollama.svelte'
 import OpenRouter from './Icons/OpenRouter.svelte'
 import HuggingFace from './Icons/HuggingFace.svelte'
+import LiteLLM from './Icons/LiteLLM.svelte'
 
 import DynamicIcon from './DynamicIcon.svelte'
 
@@ -309,7 +310,8 @@ export const icons = {
   book: Book,
   ollama: Ollama,
   openrouter: OpenRouter,
-  huggingface: HuggingFace
+  huggingface: HuggingFace,
+  litellm: LiteLLM
 }
 
 export type Icons = keyof typeof icons
@@ -470,5 +472,6 @@ export {
   LinkExternal,
   Ollama,
   OpenRouter,
-  HuggingFace
+  HuggingFace,
+  LiteLLM
 }
diff --git a/packages/types/src/ai.types.ts b/packages/types/src/ai.types.ts
index 25bf80716..22c4892a4 100644
--- a/packages/types/src/ai.types.ts
+++ b/packages/types/src/ai.types.ts
@@ -362,7 +362,8 @@ export enum CUSTOM_MODELS {
   Ollama = 'Ollama',
   OpenRouter = 'OpenRouter',
   HuggingFaceTogether = 'Hugging Face Together AI',
-  HuggingFace = 'Hugging Face Inference Endpoint'
+  HuggingFace = 'Hugging Face Inference Endpoint',
+  LiteLLM = 'LiteLLM'
 }
 
 export type CustomModelType = keyof typeof CUSTOM_MODELS
@@ -408,6 +409,14 @@ export const CUSTOM_MODEL_DEFINITIONS: Record<CUSTOM_MODELS, CustomModelDefiniti
     provider_url: 'https://api-inference.huggingface.co/models/',
     model_page: 'https://huggingface.co/models?inference_provider=together&sort=trending',
     api_key_page: 'https://huggingface.co/settings/tokens'
+  },
+  [CUSTOM_MODELS.LiteLLM]: {
+    id: CUSTOM_MODELS.LiteLLM,
+    label: 'LiteLLM',
+    icon: 'litellm',
+    provider_url: 'http://localhost:4000/v1/chat/completions',
+    model_page: 'https://docs.litellm.ai/docs/providers',
+    api_key_page: 'https://docs.litellm.ai/docs/simple_proxy#quick-start'
   }
 }