livekit-plugins-langchain 1.2.7__tar.gz → 1.2.8__tar.gz

{livekit_plugins_langchain-1.2.7 → livekit_plugins_langchain-1.2.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: livekit-plugins-langchain
-Version: 1.2.7
+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.7
+Requires-Dist: livekit-agents>=1.2.8
 Description-Content-Type: text/markdown
 
 # LangChain plugin for LiveKit Agents

{livekit_plugins_langchain-1.2.7 → livekit_plugins_langchain-1.2.8}/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-1.2.7 → livekit_plugins_langchain-1.2.8}/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.7"
+__version__ = "1.2.8"

{livekit_plugins_langchain-1.2.7 → livekit_plugins_langchain-1.2.8}/pyproject.toml

@@ -24,7 +24,7 @@ classifiers = [
     "Programming Language :: Python :: 3 :: Only",
 ]
 dependencies = [
-    "livekit-agents>=1.2.7",
+    "livekit-agents>=1.2.8",
     "langchain-core>=0.3.0",
     "langgraph>=0.3.0",
 ]