🔧 fix(schemas.py): add session_id field to ProcessResponse schema to support session-based processing

🔧 fix(process.py): add Result dataclass to encapsulate result and session_id for session-based processing

🔧 fix(manager.py): remove unused method clear_session in SessionManager
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-09-04 09:56:17 -03:00
commit 320bc96a7a
3 changed files with 9 additions and 3 deletions

View file

@ -52,6 +52,7 @@ class ProcessResponse(BaseModel):
result: Any
id: Optional[str] = None
session_id: Optional[str] = None
# TaskStatusResponse(

View file

@ -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(

View file

@ -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