🔧 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
|
result: Any
|
||||||
id: Optional[str] = None
|
id: Optional[str] = None
|
||||||
|
session_id: Optional[str] = None
|
||||||
|
|
||||||
|
|
||||||
# TaskStatusResponse(
|
# TaskStatusResponse(
|
||||||
|
|
|
||||||
|
|
@ -1,3 +1,4 @@
|
||||||
|
from dataclasses import dataclass
|
||||||
import json
|
import json
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from langchain.schema import AgentAction
|
from langchain.schema import AgentAction
|
||||||
|
|
@ -173,6 +174,12 @@ def generate_result(langchain_object: Union[Chain, VectorStore], inputs: dict):
|
||||||
# return result, session_id
|
# return result, session_id
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class Result:
|
||||||
|
result: Any
|
||||||
|
session_id: str
|
||||||
|
|
||||||
|
|
||||||
async def process_graph_cached(
|
async def process_graph_cached(
|
||||||
data_graph: Dict[str, Any],
|
data_graph: Dict[str, Any],
|
||||||
inputs: Optional[dict] = None,
|
inputs: Optional[dict] = None,
|
||||||
|
|
@ -191,7 +198,7 @@ async def process_graph_cached(
|
||||||
session_id, data_graph, (langchain_object, artifacts)
|
session_id, data_graph, (langchain_object, artifacts)
|
||||||
)
|
)
|
||||||
|
|
||||||
return result, session_id
|
return Result(result, session_id)
|
||||||
|
|
||||||
|
|
||||||
def load_flow_from_json(
|
def load_flow_from_json(
|
||||||
|
|
|
||||||
|
|
@ -34,5 +34,3 @@ class SessionManager(Service):
|
||||||
def clear_session(self, session_id, data_graph):
|
def clear_session(self, session_id, data_graph):
|
||||||
key = self.generate_key(session_id, data_graph)
|
key = self.generate_key(session_id, data_graph)
|
||||||
self.cache_manager.delete(key)
|
self.cache_manager.delete(key)
|
||||||
|
|
||||||
# Additional methods to handle session-related logic
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue