livekit-plugins-langchain 1.1.0__py3-none-any.whl

livekit/plugins/langchain/__init__.py

@@ -0,0 +1,42 @@
+# Copyright 2025 LiveKit, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""LMNT plugin for LiveKit Agents
+
+See https://docs.livekit.io/agents/integrations/llm/langchain/ for more information.
+"""
+
+from .langgraph import LangGraphStream, LLMAdapter
+from .version import __version__
+
+__all__ = ["__version__", "LLMAdapter", "LangGraphStream"]
+
+from livekit.agents import Plugin
+
+
+class LangChainPlugin(Plugin):
+    def __init__(self) -> None:
+        super().__init__(__name__, __version__, __package__)
+
+
+Plugin.register_plugin(LangChainPlugin())
+
+# Cleanup docs of unexported modules
+_module = dir()
+NOT_IN_ALL = [m for m in _module if m not in __all__]
+
+__pdoc__ = {}
+
+for n in NOT_IN_ALL:
+    __pdoc__[n] = False

livekit/plugins/langchain/langgraph.py

@@ -0,0 +1,140 @@
+# Copyright 2025 LiveKit, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import annotations
+
+from typing import Any
+
+from langchain_core.messages import AIMessage, BaseMessageChunk, HumanMessage, SystemMessage
+from langchain_core.runnables import RunnableConfig
+from langgraph.pregel.protocol import PregelProtocol
+
+from livekit.agents import llm, utils
+from livekit.agents.llm import ToolChoice
+from livekit.agents.llm.chat_context import ChatContext, ChatMessage
+from livekit.agents.llm.tool_context import FunctionTool, RawFunctionTool
+from livekit.agents.types import (
+    DEFAULT_API_CONNECT_OPTIONS,
+    NOT_GIVEN,
+    APIConnectOptions,
+    NotGivenOr,
+)
+
+
+class LLMAdapter(llm.LLM):
+    def __init__(
+        self,
+        graph: PregelProtocol,
+        *,
+        config: RunnableConfig | None = None,
+    ) -> None:
+        super().__init__()
+        self._graph = graph
+        self._config = config
+
+    def chat(
+        self,
+        *,
+        chat_ctx: ChatContext,
+        tools: list[FunctionTool | RawFunctionTool] | None = None,
+        conn_options: APIConnectOptions = DEFAULT_API_CONNECT_OPTIONS,
+        # these are unused, since tool execution takes place in langgraph
+        parallel_tool_calls: NotGivenOr[bool] = NOT_GIVEN,
+        tool_choice: NotGivenOr[ToolChoice] = NOT_GIVEN,
+        extra_kwargs: NotGivenOr[dict[str, Any]] = NOT_GIVEN,
+    ) -> LangGraphStream:
+        return LangGraphStream(
+            self,
+            chat_ctx=chat_ctx,
+            tools=tools or [],
+            graph=self._graph,
+            conn_options=conn_options,
+            config=self._config,
+        )
+
+
+class LangGraphStream(llm.LLMStream):
+    def __init__(
+        self,
+        llm: LLMAdapter,
+        *,
+        chat_ctx: ChatContext,
+        tools: list[FunctionTool | RawFunctionTool],
+        conn_options: APIConnectOptions,
+        graph: PregelProtocol,
+        config: RunnableConfig | None = None,
+    ):
+        super().__init__(
+            llm,
+            chat_ctx=chat_ctx,
+            tools=tools,
+            conn_options=conn_options,
+        )
+        self._graph = graph
+        self._config = config
+
+    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)
+            if chat_chunk:
+                self._event_ch.send_nowait(chat_chunk)
+
+    def _chat_ctx_to_state(self) -> dict[str, Any]:
+        """Convert chat context to langgraph input"""
+
+        messages: list[AIMessage | HumanMessage | SystemMessage] = []
+        for item in self._chat_ctx.items:
+            # only support chat messages, ignoring tool calls
+            if isinstance(item, ChatMessage):
+                content = item.text_content
+                if content:
+                    if item.role == "assistant":
+                        messages.append(AIMessage(content=content))
+                    elif item.role == "user":
+                        messages.append(HumanMessage(content=content))
+                    elif item.role in ["system", "developer"]:
+                        messages.append(SystemMessage(content=content))
+
+        return {
+            "messages": messages,
+        }
+
+
+def _to_chat_chunk(msg: str | Any) -> llm.ChatChunk | None:
+    message_id = utils.shortuuid("LC_")
+    content: str | None = None
+
+    if isinstance(msg, str):
+        content = msg
+    elif isinstance(msg, BaseMessageChunk):
+        content = msg.text()
+        if msg.id:
+            message_id = msg.id
+
+    if not content:
+        return None
+
+    return llm.ChatChunk(
+        id=message_id,
+        delta=llm.ChoiceDelta(
+            role="assistant",
+            content=content,
+        ),
+    )

livekit/plugins/langchain/version.py

@@ -0,0 +1,15 @@
+# Copyright 2025 LiveKit, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+__version__ = "1.1.0"

livekit_plugins_langchain-1.1.0.dist-info/METADATA

@@ -0,0 +1,61 @@
+Metadata-Version: 2.4
+Name: livekit-plugins-langchain
+Version: 1.1.0
+Summary: LangChain/LangGraph plugin for LiveKit agents
+Project-URL: Documentation, https://docs.livekit.io
+Project-URL: Website, https://livekit.io/
+Project-URL: Source, https://github.com/livekit/agents
+Author-email: LiveKit <hello@livekit.io>
+License-Expression: Apache-2.0
+Keywords: LangChain,LangGraph,agents,livekit
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Multimedia :: Sound/Audio
+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.1.0
+Description-Content-Type: text/markdown
+
+# LangChain plugin for LiveKit Agents
+
+This plugin integrates capabilites from LangChain within LiveKit Agents
+
+## Installation
+
+```bash
+pip install livekit-plugins-langchain
+```
+
+## Usage
+
+### Using LangGraph workflows
+
+You can bring over any existing workflow in LangGraph as an Agents LLM with `langchain.LLMAdapter`. For example:
+
+```python
+from langgraph.graph import StateGraph
+from livekit.agents import Agent, AgentSession, JobContext
+from livekit.plugins import langchain
+
+...
+
+def entrypoint(ctx: JobContext):
+    graph = StateGraph(...).build()
+
+    session = AgentSession(
+        vad=...,
+        stt=...,
+        tts=...,
+    )
+
+    await session.start(
+        agent=Agent(llm=langchain.LLMAdapter(graph)),
+    )
+    ...
+```

livekit_plugins_langchain-1.1.0.dist-info/RECORD

@@ -0,0 +1,7 @@
+livekit/plugins/langchain/__init__.py,sha256=IWe25ou2cWD3Ql7xQlgUGe2-BrvKGr5B6qg-yncrglk,1221
+livekit/plugins/langchain/langgraph.py,sha256=MYC4uiPcBMIIECZhXx1gKXGIYupRhHUQIN3CuvG7Dx8,4372
+livekit/plugins/langchain/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+livekit/plugins/langchain/version.py,sha256=lFVJxRoJYH4JZgg7iJQF0_dlrbfVbRYYWQzbHnv_p-w,600
+livekit_plugins_langchain-1.1.0.dist-info/METADATA,sha256=euWGyAvC9jWgQwF1oxiU8a1RUtli2YOzhtNZzAAjs_0,1733
+livekit_plugins_langchain-1.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+livekit_plugins_langchain-1.1.0.dist-info/RECORD,,

livekit_plugins_langchain-1.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.27.0
+Root-Is-Purelib: true
+Tag: py3-none-any