run-llama/template-workflow-document-qa
1
0
Fork 0
mirror of https://github.com/run-llama/template-workflow-document-qa.git synced 2026-07-01 21:44:59 -04:00
Code Issues Packages Projects Releases Wiki Activity

handle errors better

Browse Source
This commit is contained in:
Adrian Lyjak
2025-09-23 14:26:57 -04:00
parent 3a7035d722
commit 8058ae5bd2
4 changed files with 66 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/{{ project_name_snake }}/qa_workflows.py
+14 -13
View File
@@ -172,6 +172,12 @@ class QueryConversationHistoryEvent(HumanResponseEvent):
pass
class ErrorEvent(Event):
"""Event emitted when an error occurs"""
error: str
class ChatWorkflowState(BaseModel):
index_name: str | None = None
conversation_history: list[ConversationMessage] = Field(default_factory=list)
@@ -219,7 +225,7 @@ class ChatWorkflow(Workflow):
@step
async def initialize_chat(
self, ev: ChatEvent, ctx: Context[ChatWorkflowState]
) -> InputRequiredEvent | StopEvent:
) -> InputRequiredEvent | StopEvent | None:
"""Initialize the chat session and request first input"""
try:
logger.info(f"Initializing chat {ev.index_name}")
@@ -236,17 +242,14 @@ class ChatWorkflow(Workflow):
if ev.conversation_history:
async with ctx.store.edit_state() as state:
state.conversation_history.extend(ev.conversation_history)
# Request first user input
return InputRequiredEvent()
except Exception as e:
logger.error(f"Error initializing chat: {str(e)}", exc_info=True)
return StopEvent(
result={
"success": False,
"error": f"Failed to initialize chat: {str(e)}",
}
ctx.write_event_to_stream(
ErrorEvent(error=f"Failed to initialize chat: {str(e)}")
)
return InputRequiredEvent()
@step
async def get_conversation_history(
@@ -260,7 +263,7 @@ class ChatWorkflow(Workflow):
@step
async def process_user_response(
self, ev: HumanResponseEvent, ctx: Context[ChatWorkflowState]
) -> InputRequiredEvent | HumanResponseEvent | StopEvent | None:
) -> InputRequiredEvent | HumanResponseEvent | None:
"""Process user input and generate response"""
try:
logger.info(f"Processing user response {ev.response}")
@@ -326,13 +329,11 @@ class ChatWorkflow(Workflow):
assistant_response,
]
)
return InputRequiredEvent()
except Exception as e:
logger.error(f"Error processing query: {str(e)}", exc_info=True)
return StopEvent(
result={"success": False, "error": f"Error processing query: {str(e)}"}
)
ctx.write_event_to_stream(ErrorEvent(error=str(e)))
return InputRequiredEvent()
upload = DocumentUploadWorkflow(timeout=None)
test-proj/src/test_proj/qa_workflows.py
+14 -13
View File
@@ -172,6 +172,12 @@ class QueryConversationHistoryEvent(HumanResponseEvent):
pass
class ErrorEvent(Event):
"""Event emitted when an error occurs"""
error: str
class ChatWorkflowState(BaseModel):
index_name: str | None = None
conversation_history: list[ConversationMessage] = Field(default_factory=list)
@@ -219,7 +225,7 @@ class ChatWorkflow(Workflow):
@step
async def initialize_chat(
self, ev: ChatEvent, ctx: Context[ChatWorkflowState]
) -> InputRequiredEvent | StopEvent:
) -> InputRequiredEvent | StopEvent | None:
"""Initialize the chat session and request first input"""
try:
logger.info(f"Initializing chat {ev.index_name}")
@@ -236,17 +242,14 @@ class ChatWorkflow(Workflow):
if ev.conversation_history:
async with ctx.store.edit_state() as state:
state.conversation_history.extend(ev.conversation_history)
# Request first user input
return InputRequiredEvent()
except Exception as e:
logger.error(f"Error initializing chat: {str(e)}", exc_info=True)
return StopEvent(
result={
"success": False,
"error": f"Failed to initialize chat: {str(e)}",
}
ctx.write_event_to_stream(
ErrorEvent(error=f"Failed to initialize chat: {str(e)}")
)
return InputRequiredEvent()
@step
async def get_conversation_history(
@@ -260,7 +263,7 @@ class ChatWorkflow(Workflow):
@step
async def process_user_response(
self, ev: HumanResponseEvent, ctx: Context[ChatWorkflowState]
) -> InputRequiredEvent | HumanResponseEvent | StopEvent | None:
) -> InputRequiredEvent | HumanResponseEvent | None:
"""Process user input and generate response"""
try:
logger.info(f"Processing user response {ev.response}")
@@ -326,13 +329,11 @@ class ChatWorkflow(Workflow):
assistant_response,
]
)
return InputRequiredEvent()
except Exception as e:
logger.error(f"Error processing query: {str(e)}", exc_info=True)
return StopEvent(
result={"success": False, "error": f"Error processing query: {str(e)}"}
)
ctx.write_event_to_stream(ErrorEvent(error=str(e)))
return InputRequiredEvent()
upload = DocumentUploadWorkflow(timeout=None)
test-proj/ui/src/libs/useChatbot.ts
+19
View File
@@ -166,6 +166,9 @@ export function useChatbot({
interface AppendChatMessageData {
message: ChatMessage;
}
interface ErrorEventData {
error: string;
}
interface ChatMessage {
role: "user" | "assistant";
text: string;
@@ -239,6 +242,22 @@ function toMessages(events: WorkflowEvent[]): Message[] {
timestamp: new Date(content.message.timestamp),
isPartial: false,
});
} else if (type.endsWith(".ErrorEvent")) {
if (
lastMessage &&
lastMessage.isPartial &&
lastMessage.role === "assistant"
) {
messages.pop();
}
const content = ev.data as unknown as ErrorEventData;
messages.push({
role: "assistant",
content: content.error,
timestamp: new Date(),
isPartial: false,
error: true,
});
}
}
return messages;
ui/src/libs/useChatbot.ts
+19
View File
@@ -166,6 +166,9 @@ export function useChatbot({
interface AppendChatMessageData {
message: ChatMessage;
}
interface ErrorEventData {
error: string;
}
interface ChatMessage {
role: "user" | "assistant";
text: string;
@@ -239,6 +242,22 @@ function toMessages(events: WorkflowEvent[]): Message[] {
timestamp: new Date(content.message.timestamp),
isPartial: false,
});
} else if (type.endsWith(".ErrorEvent")) {
if (
lastMessage &&
lastMessage.isPartial &&
lastMessage.role === "assistant"
) {
messages.pop();
}
const content = ev.data as unknown as ErrorEventData;
messages.push({
role: "assistant",
content: content.error,
timestamp: new Date(),
isPartial: false,
error: true,
});
}
}
return messages;