o i@sjddlmZddlZddlmZmZddlmZmZGdddeZ e e e dfZ Gdddee ZdS) ) annotationsN)ABCabstractmethod)ClassVarProtocolc@seZdZUded<dS) _RunnerMetaz ClassVar[str]INFERENCE_METHODN)__name__ __module__ __qualname____annotations__r r a/var/www/html/livekit_bhavya/venv/lib/python3.10/site-packages/livekit/agents/inference_runner.pyrs r_InferenceRunnerc@sDeZdZUiZded<edddZedd d ZedddZ dS)r _RunnersDictregistered_runners runner_classtype[_InferenceRunner]returnNonecCsFttkr td|j|jvrtd|jd||j|j<dS)Nz5InferenceRunner must be registered on the main threadzInferenceRunner z already registered) threadingcurrent_thread main_thread RuntimeErrorrr ValueError)clsrr r rregister_runners  z _InferenceRunner.register_runnercCdS)z8Initialize the runner. This is used to load models, etc.Nr )selfr r r initializez_InferenceRunner.initializedatabytes bytes | NonecCr)z Run inference on the given data.Nr )rr!r r rrun"r z_InferenceRunner.runN)rrrr)rr)r!r"rr#) r r r rr classmethodrrrr$r r r rrs   ) __future__rrabcrrtypingrrrdictstrtyperrr r r rs