From 75cb5ce1109e04afc009f520d6855bb4b7a4675e Mon Sep 17 00:00:00 2001
From: Ran Shemtov <ran@copilotkit.ai>
Date: Tue, 10 Dec 2024 14:26:53 +0100
Subject: [PATCH] fix(python-sdk): fix intermediate state not emitting until
 next async fn (#1063)

---
 sdk-python/copilotkit/langchain.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/sdk-python/copilotkit/langchain.py b/sdk-python/copilotkit/langchain.py
index 0e7f454e10..e6922f1322 100644
--- a/sdk-python/copilotkit/langchain.py
+++ b/sdk-python/copilotkit/langchain.py
@@ -5,6 +5,7 @@ LangChain specific utilities for CopilotKit
 import uuid
 import json
 from typing import List, Optional, Any, Union, Dict, Callable
+import asyncio
 
 from langchain_core.messages import (
     HumanMessage,
@@ -114,6 +115,7 @@ async def copilotkit_exit(config: RunnableConfig):
         {},
         config=config,
     )
+    await asyncio.sleep(0.01)
 
     return True
 
@@ -127,6 +129,7 @@ async def copilotkit_emit_state(config: RunnableConfig, state: Any):
         state,
         config=config,
     )
+    await asyncio.sleep(0.01)
 
     return True
 
@@ -143,6 +146,7 @@ async def copilotkit_emit_message(config: RunnableConfig, message: str):
         },
         config=config,
     )
+    await asyncio.sleep(0.01)
 
     return True
 
@@ -161,5 +165,6 @@ async def copilotkit_emit_tool_call(config: RunnableConfig, *, name: str, args:
         },
         config=config,
     )
+    await asyncio.sleep(0.01)
 
     return True