o i @sdZddlmZmZddlmZddlmZmZm Z m Z m Z ddl m Z ddlmZmZzddlmZddlmZWneyRZz ed ed ed Z[wwGd d d eZd S)z,Langchain integration processor for Pipecat.)OptionalUnion)logger)FrameLLMContextFrameLLMFullResponseEndFrameLLMFullResponseStartFrame TextFrame)OpenAILLMContextFrame)FrameDirectionFrameProcessor)AIMessageChunk)RunnablezLIn order to use Langchain, you need to `pip install pipecat-ai[langchain]`. zMissing module: Ncs|eZdZdZddedeffdd Zdefdd Zd ed e ffd d Z e de ee fdefddZdefddZZS)LangchainProcessora.Processor that integrates Langchain runnables with Pipecat's frame pipeline. This processor takes LLM message frames, extracts the latest user message, and processes it through a Langchain runnable chain. The response is streamed back as text frames with appropriate response markers. inputchaintranscript_keycs t||_||_d|_dS)zInitialize the Langchain processor. Args: chain: The Langchain runnable to use for processing messages. transcript_key: The key to use when passing input to the chain. N)super__init___chain_transcript_key_participant_id)selfrr __class__[/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/processors/frameworks/langchain.pyr's  zLangchainProcessor.__init__participant_idcCs ||_dS)zSet the participant ID for session tracking. Args: participant_id: The participant ID to use for session configuration. N)r)rrrrrset_participant_id3s z%LangchainProcessor.set_participant_idframe directioncst||IdHt|ttfr:td|t|tr#|jjn|j }|dd}| | IdHdS| ||IdHdS)zProcess incoming frames and handle LLM message frames. Args: frame: The incoming frame to process. direction: The direction of frame flow in the pipeline. NzGot transcription frame content) r process_frame isinstancerr rdebugcontextmessages get_messages_ainvokestrip push_frame)rrr r'textrrrr#;s  z LangchainProcessor.process_framer,returncCs.|tdr |Stdr|jS dS)zExtract token value from various text types. Args: text: The text or message chunk to extract value from. Returns: The extracted string value. r)strr r")r,rrr__get_token_valueSs  z$LangchainProcessor.__get_token_valuec s:td||tIdHzz,|jj|j|idd|jiid2z3dHW}t| |}d|_ ||IdHq$6Wn/t yPt |dYn+t yn}z|jd||d IdHWYd}~nd}~wwW|tIdHdSW|tIdHdSW|tIdHdS|tIdHw) z5Invoke the Langchain runnable with the provided text.zInvoking chain with N configurable session_id)configTz! generator was closed prematurelyzUnknown error occurred: ) error_msg exception)rr%r+rrastreamrrr $_LangchainProcessor__get_token_valueincludes_inter_frame_spaces GeneratorExitwarning Exception push_errorr)rr,tokenrerrrr)es0  & *zLangchainProcessor._ainvoke)r)__name__ __module__ __qualname____doc__rr/rrrr r# staticmethodrr r7r) __classcell__rrrrrs r)rBtypingrrlogururpipecat.frames.framesrrrrr 1pipecat.processors.aggregators.openai_llm_contextr "pipecat.processors.frame_processorr r langchain_core.messagesr langchain_core.runnablesrModuleNotFoundErrorr>errorr;rrrrrs