.h ddlmZddlZddlmZddlmZddlmZddl m Z ddl m Z dd l mZGd d ejZGd d ZGddZGddeZdS)) annotationsN) AsyncIterator)Any)Agent)TResponseInputItem)RunResultStreaming)Runnerc4eZdZdZejddZdS) VoiceWorkflowBaseal A base class for a voice workflow. You must implement the `run` method. A "workflow" is any code you want, that receives a transcription and yields text that will be turned into speech by a text-to-speech model. In most cases, you'll create `Agent`s and use `Runner.run_streamed()` to run them, returning some or all of the text events from the stream. You can use the `VoiceWorkflowHelper` class to help with extracting text events from the stream. If you have a simple workflow that has a single starting agent and no custom logic, you can use `SingleAgentVoiceWorkflow` directly. transcriptionstrreturnAsyncIterator[str]cdS)aC Run the voice workflow. You will receive an input transcription, and must yield text that will be spoken to the user. You can run whatever logic you want here. In most cases, the final logic will involve calling `Runner.run_streamed()` and yielding any text events from the stream. N)selfr s R/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/agents/voice/workflow.pyrunzVoiceWorkflowBase.runs  Nr rrr)__name__ __module__ __qualname____doc__abcabstractmethodrrrrr r sB         rr c&eZdZeddZdS)VoiceWorkflowHelperresultr rrcK|23d{V}|jdkr|jjdkr|jjWV26dS)zKWraps a `RunResultStreaming` object and yields text events from the stream.Nraw_response_eventzresponse.output_text.delta) stream_eventstypedatadelta)clsr events rstream_text_fromz$VoiceWorkflowHelper.stream_text_from%sz"//11 ' ' ' ' ' ' '% 222JO'CCCj&&&&& 211sA N)r r rr)rrr classmethodr)rrrrr$s2'''['''rrceZdZd dZdS) SingleAgentWorkflowCallbacksworkflowSingleAgentVoiceWorkflowr rrNonecdS)z Called when the workflow is run.Nr)rr-r s ron_runz#SingleAgentWorkflowCallbacks.on_run1s rN)r-r.r rrr/)rrrr1rrrr,r,0s(      rr,c$eZdZdZd ddZdd ZdS)r.a)A simple voice workflow that runs a single agent. Each transcription and result is added to the input history. For more complex workflows (e.g. multiple Runner calls, custom message history, custom logic, custom configs), subclass `VoiceWorkflowBase` and implement your own logic. Nagent Agent[Any] callbacks#SingleAgentWorkflowCallbacks | Nonec0g|_||_||_dS)zCreate a new single agent voice workflow. Args: agent: The agent to run. callbacks: Optional callbacks to call during the workflow. N)_input_history_current_agent _callbacks)rr3r5s r__init__z!SingleAgentVoiceWorkflow.__init__=s9;##rr rrrc`K|jr|j|||jd|dt j|j|j}t|23d{V}|WV 6| |_|j |_dS)Nuser)rolecontent) r:r1r8appendr run_streamedr9rr) to_input_list last_agent)rr r chunks rrzSingleAgentVoiceWorkflow.runHs ? 8 O " "4 7 7 7 ""(     $T%8$:MNN/??GG       %KKKKKH%2244$/s;B)N)r3r4r5r6r)rrrrr;rrrrr.r.6sK $ $ $ $ $000000rr.) __future__rrcollections.abcrtypingrr3ritemsrr r rr ABCr rr,r.rrrrJs="""""" ))))))&&&&&&''''''        . ' ' ' ' ' ' ' '        '0'0'0'0'00'0'0'0'0'0r