ԟjQdZddlmZddlZddlZddlZddlZddlZddlm Z ej dZ GddZ d d Z dS) zDService to automatically monitor and restart local LM Studio server.) annotationsN) get_settingszdscons.lmstudio_managerc4eZdZdZd ddZdd Zdd Zdd Zd S)LMStudioManagerzDManages the lifecycle of the local LM Studio server using 'lms' CLI.http://127.0.0.1:1234/v1/models check_urlstrreturnNonec:||_t|_dS)N)rrsettings)selfrs 3C:\Projects\DSCons\app\services\lmstudio_manager.py__init__zLMStudioManager.__init__s"$ boolcK tjd4d{V}||jd{V}|jdkcdddd{VS#1d{VswxYwYdS#t $rYdSwxYw)z;Pings the LM Studio models endpoint to check if it's alive.g@)timeoutNF)httpx AsyncClientgetr status_code Exception)rclientress ris_server_runningz!LMStudioManager.is_server_runnings- (555 . . . . . . ."JJt~66666666#- . . . . . . . . . . . . . . . . . . . . . . . . . . . . . .   55 s4A0+A A0 A''A0*A'+A00 A>=A>cTKtjdstddStd t jdddtjtjd{V}| d{V\}}|j d krtd d Std | d ddS#t$r&}td|Yd}~dSd}~wwxYw)z;Invokes the 'lms server start' command to start the server.lmsz(lms CLI is not installed or not in PATH.Fz3Starting LM Studio server via 'lms server start'...serverstart)stdoutstderrNrz1Successfully requested LM Studio server to start.TzFailed to start LM Studio: %szutf-8ignore)errorsz&Error executing 'lms server start': %s) shutilwhichloggererrorinfoasynciocreate_subprocess_exec subprocessPIPE communicate returncodedecoder)rprocessr"r#es r start_serverzLMStudioManager.start_server!sO|E""  LLC D D D5 IJJJ #:xG$+#6#6#8#8888888NFF!Q&& OPPPt 3MM'(M;;u    LLA1 E E E55555 sA5C70C77 D'D""D'cK|jjsdS|jjdkp4td|jjD}|sdS|d{Vstd| d{V}|rtdD]S}tj dd{V|d{Vrt ddSTtddStd dStd dS) z7Checks the server status and attempts recovery if down.Nlmstudioc3"K|] }|dkV dS)r6N).0provs r z4LMStudioManager.check_and_recover..Bs@H H #'DJ H H H H H H rz;LM Studio server is NOT running. Attempting auto-restart... z1LM Studio server has recovered and is now active.zBLM Studio requested to start, but did not respond on port in time.z0Failed to execute LM Studio server auto-restart.z'LM Studio server is active and healthy.)r lmstudio_enabled llm_provideranyagent_provider_routingvaluesrr(warningr4ranger+sleepr*r)debug)r uses_lmstudiosuccess_s rcheck_and_recoverz!LMStudioManager.check_and_recover;s}-  F 2j@ CH H +/=+O+V+V+X+XH H H E E   F++-------- D NNM   !--////////G QrA!-*********!3355555555 $WXXX X OPPPPP LLB C C C C CrN)r)rr r r )r rr r )__name__ __module__ __qualname____doc__rrr4rJr8rrrrsuNN'''''4DDDDDDrrr r cKt} |d{Vn2#t$r%}td|Yd}~nd}~wwxYwt jdd{Vi)zIBackground loop that runs indefinitely checking and recovering LM Studio.TNz#Error in LM Studio monitor loop: %s<)rrJrr(r)r+rE)managerr3s rstart_lmstudio_monitor_looprS]sG  C++-- - - - - - - - - C C C LL> B B B B B B B B CmB  s. AAArK)rO __future__rr+loggingr&r-rapp.core.settingsr getLoggerr(rrSr8rrrXsJJ""""""  ******  4 5 5IDIDIDIDIDIDIDIDX       r