fix

f5242389 · Timothy J. Baek · 646832ba · f5242389
Commit f5242389 authored Jun 22, 2024 by Timothy J. Baek
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

backend/main.py backend/main.py +3 -3

No files found.
--- a/backend/main.py
+++ b/backend/main.py
@@ -876,11 +876,11 @@ async def generate_chat_completions(form_data: dict, user=Depends(get_verified_u
            print(pipe_id)

            # Check if function is already loaded
-            if pipe_id not in app.state.FUNCTIONS:
+            if pipe_id not in webui_app.state.FUNCTIONS:
                function_module, function_type = load_function_module_by_id(pipe_id)
-                app.state.FUNCTIONS[pipe_id] = function_module
+                webui_app.state.FUNCTIONS[pipe_id] = function_module
            else:
-                function_module = app.state.FUNCTIONS[pipe_id]
+                function_module = webui_app.state.FUNCTIONS[pipe_id]

            pipe = function_module.pipe