fix build errors

mbektas · Jan 5, 2025 · 03af6e3 · 03af6e3
1 parent 2c2eb1c
commit 03af6e3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/notebook_intelligence/extension.py b/notebook_intelligence/extension.py
@@ -328,7 +328,7 @@ async def _tool_call_loop(tool_call_rounds: list):
 
                 tool_call_response = await tool_to_call.handle_tool_call(request, response, tool_context, args)
 
-                tool_call_args_resp = args | tool_call_response
+                tool_call_args_resp = args.update(tool_call_response)
 
                 function_call_result_message = {
                     "role": "tool",

diff --git a/notebook_intelligence/handlers.py b/notebook_intelligence/handlers.py
@@ -126,7 +126,7 @@ def chat_id(self) -> str:
     def message_id(self) -> str:
         return self.messageId
 
-    def stream(self, data: Union[ResponseStreamData | dict]):
+    def stream(self, data: Union[ResponseStreamData, dict]):
         data_type = ResponseStreamDataType.LLMRaw if type(data) is dict else data.data_type
 
         if data_type == ResponseStreamDataType.Markdown: