Fixed bug where workflow ControllerClient is not updated when Workflo…

…wServerConfig is updated (#29)
mlrun · Oct 1, 2024 · 89f97cb · 89f97cb
1 parent b575935
commit 89f97cb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/genai_factory/src/genai_factory/workflows/workflow_server.py b/genai_factory/src/genai_factory/workflows/workflow_server.py
@@ -15,7 +15,6 @@
 from urllib.parse import urlparse
 
 import uvicorn
-
 from genai_factory.config import WorkflowServerConfig
 from genai_factory.controller_client import ControllerClient
 from genai_factory.schemas import WorkflowType
@@ -55,6 +54,7 @@ def set_config(self, config: WorkflowServerConfig):
         self._session_store = SessionStore(self._controller_client)
         for workflow in self._workflows.values():
             workflow._server = None
+            workflow._client = self._controller_client
 
     def add_workflow(
         self,