🔧 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:
parent
8d484eb046
commit
320bc96a7a
3 changed files with 9 additions and 3 deletions
|
|
@ -52,6 +52,7 @@ class ProcessResponse(BaseModel):
|
|||
|
||||
result: Any
|
||||
id: Optional[str] = None
|
||||
session_id: Optional[str] = None
|
||||
|
||||
|
||||
# TaskStatusResponse(
|
||||
|
|
|
|||
|
|
@ -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(
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue