Merge pull request #179 from langchain-ai/nc/7feb/fix-other-bots

Fix streaming for non-assistant bots

backend/app/stream.py

@@ -43,7 +43,7 @@ async def astream_messages(
             last_messages_list = event["data"]["chunk"]
 
             yield last_messages_list
-        elif event["event"] == "on_chat_model_start":
+        elif event["event"] == "on_chat_model_start" and last_messages_list is None:
             input_messages_outer = event["data"]["input"].get("messages")
             input_messages = (
                 input_messages_outer[0]
@@ -55,7 +55,7 @@ async def astream_messages(
                 if input_messages is not None
                 else None
             )
-            if input_messages and input_messages != last_messages_list:
+            if input_messages:
                 last_messages_list = input_messages
                 yield last_messages_list
         elif (

backend/app/tools.py

@@ -175,4 +175,7 @@ class AvailableTools(str, Enum):
 
 # Check if dependencies and env vars for each tool are available
 for k, v in TOOLS.items():
-    v()
+    # Connery requires env vars to be valid even if the tool isn't used,
+    # so we'll skip the check for it
+    if k != AvailableTools.CONNERY:
+        v()