livekit-plugins-langchain 1.2.6__py3-none-any.whl → 1.2.8__py3-none-any.whl

livekit/plugins/langchain/langgraph.py

@@ -38,10 +38,12 @@ class LLMAdapter(llm.LLM):
         graph: PregelProtocol,
         *,
         config: RunnableConfig | None = None,
+        subgraphs: bool = False,
     ) -> None:
         super().__init__()
         self._graph = graph
         self._config = config
+        self._subgraphs = subgraphs
 
     def chat(
         self,
@@ -61,6 +63,7 @@ class LLMAdapter(llm.LLM):
             graph=self._graph,
             conn_options=conn_options,
             config=self._config,
+            subgraphs=self._subgraphs,
         )
 
 
@@ -74,6 +77,7 @@ class LangGraphStream(llm.LLMStream):
         conn_options: APIConnectOptions,
         graph: PregelProtocol,
         config: RunnableConfig | None = None,
+        subgraphs: bool = False,
     ):
         super().__init__(
             llm,
@@ -83,16 +87,33 @@ class LangGraphStream(llm.LLMStream):
         )
         self._graph = graph
         self._config = config
+        self._subgraphs = subgraphs
 
     async def _run(self) -> None:
         state = self._chat_ctx_to_state()
 
-        async for message_chunk, _ in self._graph.astream(
-            state,
-            self._config,
-            stream_mode="messages",
-        ):
-            chat_chunk = _to_chat_chunk(message_chunk)
+        # Some LangGraph versions don't accept the `subgraphs` kwarg yet.
+        # Try with it first; fall back gracefully if it's unsupported.
+        try:
+            aiter = self._graph.astream(
+                state,
+                self._config,
+                stream_mode="messages",
+                subgraphs=self._subgraphs,
+            )
+        except TypeError:
+            aiter = self._graph.astream(
+                state,
+                self._config,
+                stream_mode="messages",
+            )
+
+        async for item in aiter:
+            token_like = _extract_message_chunk(item)
+            if token_like is None:
+                continue
+
+            chat_chunk = _to_chat_chunk(token_like)
             if chat_chunk:
                 self._event_ch.send_nowait(chat_chunk)
 
@@ -112,9 +133,48 @@ class LangGraphStream(llm.LLMStream):
                     elif item.role in ["system", "developer"]:
                         messages.append(SystemMessage(content=content, id=item.id))
 
-        return {
-            "messages": messages,
-        }
+        return {"messages": messages}
+
+
+def _extract_message_chunk(item: Any) -> BaseMessageChunk | str | None:
+    """
+    Normalize outputs from graph.astream(..., stream_mode='messages', [subgraphs]).
+
+    Expected shapes:
+      - (token, meta)
+      - (namespace, (token, meta))                  # with subgraphs=True
+      - (mode, (token, meta))                       # future-friendly
+      - (namespace, mode, (token, meta))            # future-friendly
+    Also tolerate direct token-like values for robustness.
+    """
+    # Already a token-like thing?
+    if isinstance(item, (BaseMessageChunk, str)):
+        return item
+
+    if not isinstance(item, tuple):
+        return None
+
+    # token is usually BaseMessageChunk, but could be a str
+    # (token, meta)
+    if len(item) == 2 and not isinstance(item[1], tuple):
+        token, _meta = item
+        return token  # type: ignore
+
+    # (namespace, (token, meta))  OR  (mode, (token, meta))
+    if len(item) == 2 and isinstance(item[1], tuple):
+        inner = item[1]
+        if len(inner) == 2:
+            token, _meta = inner
+            return token  # type: ignore
+
+    # (namespace, mode, (token, meta))
+    if len(item) == 3 and isinstance(item[2], tuple):
+        inner = item[2]
+        if len(inner) == 2:
+            token, _meta = inner
+            return token  # type: ignore
+
+    return None
 
 
 def _to_chat_chunk(msg: str | Any) -> llm.ChatChunk | None:
@@ -125,8 +185,8 @@ def _to_chat_chunk(msg: str | Any) -> llm.ChatChunk | None:
         content = msg
     elif isinstance(msg, BaseMessageChunk):
         content = msg.text()
-        if msg.id:
-            message_id = msg.id
+        if getattr(msg, "id", None):
+            message_id = msg.id  # type: ignore
 
     if not content:
         return None

livekit/plugins/langchain/version.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "1.2.6"
+__version__ = "1.2.8"

{livekit_plugins_langchain-1.2.6.dist-info → livekit_plugins_langchain-1.2.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: livekit-plugins-langchain
-Version: 1.2.6
+Version: 1.2.8
 Summary: LangChain/LangGraph plugin for LiveKit agents
 Project-URL: Documentation, https://docs.livekit.io
 Project-URL: Website, https://livekit.io/
@@ -19,7 +19,7 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: >=3.9.0
 Requires-Dist: langchain-core>=0.3.0
 Requires-Dist: langgraph>=0.3.0
-Requires-Dist: livekit-agents>=1.2.6
+Requires-Dist: livekit-agents>=1.2.8
 Description-Content-Type: text/markdown
 
 # LangChain plugin for LiveKit Agents

livekit_plugins_langchain-1.2.8.dist-info/RECORD

@@ -0,0 +1,7 @@
+livekit/plugins/langchain/__init__.py,sha256=orJAv6qYRWJatEdyMlf9D93GXxeK-ELYa9XIqYF76hw,1140
+livekit/plugins/langchain/langgraph.py,sha256=SXZTRBY7vOwIwwITrSoG0_ZR76UKqWh5rRr1R1NPOgI,6448
+livekit/plugins/langchain/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+livekit/plugins/langchain/version.py,sha256=A0WkWSMcyRjWChIOCaf9rjKCQpO2BmwPf6v0PURlv-k,600
+livekit_plugins_langchain-1.2.8.dist-info/METADATA,sha256=07Fr7eYO76M4CH9w4vzuSgvpwas2KvVacvHsGi5lvdI,1735
+livekit_plugins_langchain-1.2.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+livekit_plugins_langchain-1.2.8.dist-info/RECORD,,

livekit_plugins_langchain-1.2.6.dist-info/RECORD

@@ -1,7 +0,0 @@
-livekit/plugins/langchain/__init__.py,sha256=orJAv6qYRWJatEdyMlf9D93GXxeK-ELYa9XIqYF76hw,1140
-livekit/plugins/langchain/langgraph.py,sha256=KxbK6AbAkf_2oZ6eACCP9J3jZ-p1SqsTG2NBzLWDK-s,4408
-livekit/plugins/langchain/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-livekit/plugins/langchain/version.py,sha256=rmFW-1Nh90Zrz7VO8AXjb-TFa4_KVV79YegUSrx9qbw,600
-livekit_plugins_langchain-1.2.6.dist-info/METADATA,sha256=1_UOr96PUbT6Fo9BAKh95_enxHji0g4s_cf9LUAcAeg,1735
-livekit_plugins_langchain-1.2.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-livekit_plugins_langchain-1.2.6.dist-info/RECORD,,