Update index.mdx

open-webui · Dec 29, 2024 · 51941cb · 51941cb
1 parent f6015b8
commit 51941cb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/features/plugin/migration/index.mdx b/docs/features/plugin/migration/index.mdx
@@ -199,7 +199,7 @@ class Pipe:
 from pydantic import BaseModel
 from fastapi import Request
 
-from open_webui.main import chat_completion
+from open_webui.utils.chat import generate_chat_completion
 
 
 class User(BaseModel):
@@ -222,7 +222,7 @@ class Pipe:
         # Uses the unified endpoint with updated signature
         user = User(**__user__)
         body["model"] = "llama3.2:latest"
-        return await chat_completion(__request__, body, user)
+        return await generate_chat_completion(__request__, body, user)
 ```
 
 ### Important Notes: