Properly pass DB session (#941)

EugeneLightsOn · web-flow · commit aec188d967b8 · 2025-02-19T10:36:05.000-05:00
* Pass DB session

* Fix toolkit event
diff --git a/src/backend/chat/custom/custom.py b/src/backend/chat/custom/custom.py
@@ -231,7 +231,7 @@ async def call_chat(
             if has_tool_calls:
                 # Handle tool calls
                 tool_results = await async_call_tools(
-                    chat_request.chat_history, deployment_model, ctx, **kwargs
+                    chat_request.chat_history, deployment_model, ctx, session=session, **kwargs
                 )
 
                 # Remove the message if tool results are present
diff --git a/src/interfaces/assistants_web/src/components/MessageRow/ToolEvents.tsx b/src/interfaces/assistants_web/src/components/MessageRow/ToolEvents.tsx
@@ -102,7 +102,7 @@ const ToolEvent: React.FC<ToolEventProps> = ({ plan, event, stream_search_result
       >
         {errorDocuments[errorDocuments.length - 1].text}
       </ToolErrorWrapper>
-    ) : toolName && toolName != TOOL_PYTHON_INTERPRETER_ID ? (
+    ) : (toolName && toolName != TOOL_PYTHON_INTERPRETER_ID) || artifacts.length > 0 ? (
       <ToolEventWrapper icon="book-open-text">
         {artifacts.length > 0 ? (
           <>

Original file line number	Diff line number	Diff line change
`@@ -231,7 +231,7 @@ async def call_chat(`
`231`	`231`	`if has_tool_calls:`
`232`	`232`	`# Handle tool calls`
`233`	`233`	`tool_results = await async_call_tools(`
`234`		`- chat_request.chat_history, deployment_model, ctx, **kwargs`
	`234`	`+ chat_request.chat_history, deployment_model, ctx, session=session, **kwargs`
`235`	`235`	`)`
`236`	`236`
`237`	`237`	`# Remove the message if tool results are present`
Original file line number	Diff line number	Diff line change
`@@ -102,7 +102,7 @@ const ToolEvent: React.FC<ToolEventProps> = ({ plan, event, stream_search_result`
`102`	`102`	`>`
`103`	`103`	`{errorDocuments[errorDocuments.length - 1].text}`
`104`	`104`	`</ToolErrorWrapper>`
`105`		`- ) : toolName && toolName != TOOL_PYTHON_INTERPRETER_ID ? (`
	`105`	`+ ) : (toolName && toolName != TOOL_PYTHON_INTERPRETER_ID) \|\| artifacts.length > 0 ? (`
`106`	`106`	`<ToolEventWrapper icon="book-open-text">`
`107`	`107`	`{artifacts.length > 0 ? (`
`108`	`108`	`<>`