diff --git a/docker-compose.yaml b/docker-compose.yaml
index 2a0bf62..28d0788 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -1,6 +1,6 @@
 services:
   openai_forward:
-    image: beidongjiedeguang/openai-forward:webui-latest
+    build: .
     container_name: openai-forward-container
     env_file:
       .env
diff --git a/pyproject.toml b/pyproject.toml
index e641b25..2e93f92 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -29,7 +29,7 @@ dependencies = [
     "pytz",
     "slowapi==0.1.9",
     "aiohttp>=3.8.5",
-    "flaxkv[client]>=0.2.7",
+    "flaxkv>=0.2.7",
     "httpx[http2]>=0.26.0",
     "msgpack",
     "tiktoken>=0.5.1",
@@ -37,6 +37,7 @@ dependencies = [
     "tomli-w",
     "pyzmq",
     "pyyaml",
+    "litellm"
 ]
 
 dynamic = ["version"]