fix: litellm

open-webui · Jun 4, 2024 · ecd7f8e · ecd7f8e
1 parent 17bc338
commit ecd7f8e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/pipelines/providers/litellm_manifold_pipeline.py b/examples/pipelines/providers/litellm_manifold_pipeline.py
@@ -109,7 +109,7 @@ def pipe(
             headers["Authorization"] = f"Bearer {self.valves.LITELLM_API_KEY}"
 
         try:
-            payload = {**body, "model": model_id, "user_id": body["user"]["id"]}
+            payload = {**body, "model": model_id, "user": body["user"]["id"]}
             payload.pop("chat_id", None)
             payload.pop("user", None)
             payload.pop("title", None)

diff --git a/examples/pipelines/providers/litellm_subprocess_manifold_pipeline.py b/examples/pipelines/providers/litellm_subprocess_manifold_pipeline.py
@@ -197,7 +197,7 @@ def pipe(
         try:
             r = requests.post(
                 url=f"http://{self.valves.LITELLM_PROXY_HOST}:{self.valves.LITELLM_PROXY_PORT}/v1/chat/completions",
-                json={**body, "model": model_id, "user_id": body["user"]["id"]},
+                json={**body, "model": model_id, "user": body["user"]["id"]},
                 stream=True,
             )