🐛 fix(endpoints.py): fix response and session_id assignment in process_flow function
🐛 fix(schemas.py): add session_id field to ProcessResponse schema 🐛 fix(process.py): fix session_id assignment in process_graph_cached function
This commit is contained in:
parent
fc5670f534
commit
6af7d04d4e
3 changed files with 6 additions and 4 deletions
|
|
@ -91,10 +91,10 @@ async def process_flow(
|
||||||
graph_data = process_tweaks(graph_data, tweaks)
|
graph_data = process_tweaks(graph_data, tweaks)
|
||||||
except Exception as exc:
|
except Exception as exc:
|
||||||
logger.error(f"Error processing tweaks: {exc}")
|
logger.error(f"Error processing tweaks: {exc}")
|
||||||
response = process_graph_cached(graph_data, inputs, clear_cache, session_id)
|
response, session_id = process_graph_cached(
|
||||||
return ProcessResponse(
|
graph_data, inputs, clear_cache, session_id
|
||||||
result=response,
|
|
||||||
)
|
)
|
||||||
|
return ProcessResponse(result=response, session_id=session_id)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
# Log stack trace
|
# Log stack trace
|
||||||
logger.exception(e)
|
logger.exception(e)
|
||||||
|
|
|
||||||
|
|
@ -47,6 +47,7 @@ class ProcessResponse(BaseModel):
|
||||||
"""Process response schema."""
|
"""Process response schema."""
|
||||||
|
|
||||||
result: dict
|
result: dict
|
||||||
|
session_id: str = None
|
||||||
|
|
||||||
|
|
||||||
class ChatMessage(BaseModel):
|
class ChatMessage(BaseModel):
|
||||||
|
|
|
||||||
|
|
@ -117,6 +117,7 @@ def process_graph_cached(
|
||||||
logger.debug("Cleared cache")
|
logger.debug("Cleared cache")
|
||||||
|
|
||||||
langchain_object, artifacts = get_build_result(data_graph, session_id)
|
langchain_object, artifacts = get_build_result(data_graph, session_id)
|
||||||
|
session_id = build_sorted_vertices_with_caching.hash
|
||||||
|
|
||||||
logger.debug("Loaded LangChain object")
|
logger.debug("Loaded LangChain object")
|
||||||
if inputs is None:
|
if inputs is None:
|
||||||
|
|
@ -151,7 +152,7 @@ def process_graph_cached(
|
||||||
raise ValueError(
|
raise ValueError(
|
||||||
f"Unknown langchain_object type: {type(langchain_object).__name__}"
|
f"Unknown langchain_object type: {type(langchain_object).__name__}"
|
||||||
)
|
)
|
||||||
return result
|
return result, session_id
|
||||||
|
|
||||||
|
|
||||||
def load_flow_from_json(
|
def load_flow_from_json(
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue