diff --git a/src/backend/langflow/api/v1/schemas.py b/src/backend/langflow/api/v1/schemas.py index 28f351c75..835c9fcb0 100644 --- a/src/backend/langflow/api/v1/schemas.py +++ b/src/backend/langflow/api/v1/schemas.py @@ -52,6 +52,7 @@ class ProcessResponse(BaseModel): result: Any id: Optional[str] = None + session_id: Optional[str] = None # TaskStatusResponse( diff --git a/src/backend/langflow/processing/process.py b/src/backend/langflow/processing/process.py index e887c3a0b..da8996069 100644 --- a/src/backend/langflow/processing/process.py +++ b/src/backend/langflow/processing/process.py @@ -1,3 +1,4 @@ +from dataclasses import dataclass import json from pathlib import Path from langchain.schema import AgentAction @@ -173,6 +174,12 @@ def generate_result(langchain_object: Union[Chain, VectorStore], inputs: dict): # return result, session_id +@dataclass +class Result: + result: Any + session_id: str + + async def process_graph_cached( data_graph: Dict[str, Any], inputs: Optional[dict] = None, @@ -191,7 +198,7 @@ async def process_graph_cached( session_id, data_graph, (langchain_object, artifacts) ) - return result, session_id + return Result(result, session_id) def load_flow_from_json( diff --git a/src/backend/langflow/services/session/manager.py b/src/backend/langflow/services/session/manager.py index 39cefba24..a131d15d1 100644 --- a/src/backend/langflow/services/session/manager.py +++ b/src/backend/langflow/services/session/manager.py @@ -34,5 +34,3 @@ class SessionManager(Service): def clear_session(self, session_id, data_graph): key = self.generate_key(session_id, data_graph) self.cache_manager.delete(key) - - # Additional methods to handle session-related logic