o i@s>ddlZddlmZddlmZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZmZmZm Z ddl!m"Z"m#Z#ddl$m%Z%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.ej/ddddZ0dde1e2e3Bde2de4fddZ5de2de4fd d!Z6d"e2de4fd#d$Z7e0j8d%d&d'ej9eej:dd(d)d*d+ej:dd,d-d.d+fd/e4d0ee.fd1d%Z1e0j8d2d3d'ej9eej;d4d+ej:dd(d)d5d+fd6e4d/e4fd7d2Zd?d@d+ej:dd(d)dAd+fd6e4dBe4d/e4fdCd8Z=e0j8dDdEd'ej9edFdDZ>GdGdHdHe4eZ?GdIdJdJe4eZ@GdKdLdLe4eZAe0j8dMdNd'ej9eej:dd(d)d5d+ej:ddOdPdQd+ej:e?jBdRdSdTd+ej:dUdVdWdXd+ej:ddYdZd[d+ej:dd\d]d^d+fd6e4d/e4d_e@d`e?dae2dbe4dBe4fdcdMZCe0j8ddded'ej9eej:dd(d)dfd+ej:dt||}g}t|D]%}t||}t|d|}|||}||r8t|t|ndq|}dt|t|}|krUdt|S|ddfdd|DS) z?Generate Unicode sparkline from values, downsampling if needed.ru▁▂▃▄▅▆▇█c3s*|]}tdt|VqdS)r%N)minint).0vblocksloscaleS/home/ubuntu/.local/lib/python3.10/site-packages/pipecatcloud/cli/commands/agent.py >s(zsparkline..)lenranger'appendsumr&maxjoin) rr bucket_size downsampledistartendbuckethir.r*r/ sparkline)s"       r>bcCsR|dkr |dddS|dkr|dddS|dkr$|dddS|d S) z&Format bytes as human-readable string.i@.1fGBiz.0fMBiKBBr.)r?r.r.r/ format_bytesAs rE millicorescCs|dddS)z/Format CPU millicores as human-readable string.iz.2fz coresr.)rFr.r.r/ format_cpuLsrGlistzList agents in an organization.)rrz--organizationz-ozOrganization to list agents for)rz--regionz-rzFilter by region organizationregionc s|ptd}|r+t|IdHs+tIdH}td|dd|dtdStj d|dd d t j ||d IdH\}}|rPtWdS|rXt |d krpt d |dtdtdWdStdddtjd}|d|d|d|d|d|d|D]}|d|dd|d|d|d|d|d qtj|d!|t |d"d#WddS1swYdS)$Norgz[red]Invalid region 'z'. Valid regions are: z, [/red]r#z.[dim]Fetching agents for organization: [bold]''[/bold][/dim]dotsspinner)rKrJrz3[red]No agents found for namespace / organization 'zB'[/red] [dim]Please deploy an agent first using[/dim] [bold cyan]z deploy[/bold cyan]Tdim show_header show_lines border_stylerNamerzAgent IDzActive Deployment ID Created At Updated Atz[bold]r[/bold]rJidactiveDeploymentId createdAt updatedAtzAgents for organization: z results)title title_extra)rgetrrrprintr6typerExitstatusragentsr1errorrr rSIMPLE add_columnadd_rowsuccess)rIrJrK valid_regionsdatarftableservicer.r.r/rHTsV          "rdzGet status of agent deploymentz2Name of the agent to get status of e.g. 'my-agent'z'Organization to get status of agent for agent_namec s|ptd}ttjd|ddd}tj|||dIdH\}}td|| |r=t WdS|sSt d|d t WdSt d d tjd }|d |d |dt|dd|dt|didididd|d}|r|dt||dt|dd|dt|dd|dt|dd|didididi}t|tr|dd } nt|} |d | rd!nd"|d#} d"} | rt| tr| d$} | rd%| d&} |d'| |d(d} | r9td)| d*d+d,d-td.| d/d+d,d-g}tt|d0d1d2d3}d}|d4g}|rt|d+krt d d tjd }|d5|d6|D]}|d7|d8d9d:|d;dpw|dq_t|d?d1d@d3}|dArdBndC}|dArdDtdE|dFndGtdH|dF}ttt||r|ndItdJ|dK|dArdLndMdN|dArdOndPd dQ|r|ndIdR|dSd1d1|dTWddS1swYdS)UNrKz [dim]Looking up agent with name [/dim]rNrOrorKlivezAgent status: z.No deployment data found for agent with name ''F)rSrTrKeyValuez"[bold]Active Session Count:[/bold]activeSessionCountN/Az[bold]Image:[/bold] deploymentmanifestspecimage agentProfilezold]Agent Profile:[/bold]z"[bold]Active Deployment ID:[/bold]r[z[bold]Created At:[/bold]r\z[bold]Updated At:[/bold]r]integratedKeysProxyenabledz[bold]Managed Keys:[/bold]z[green]Enabled[/green]z[dim]Disabled[/dim] krispViva audioFilterz[green]Enabled (z )[/green]z[bold]Krisp VIVA:[/bold] autoScalingz[bold]Minimum Agents[/bold] minReplicasrTexpandz[bold]Maximum Agents[/bold] maxReplicasz#[bold]Scaling configuration:[/bold]leftrQr^ title_alignrUerrorsCodeMessagez [bold red]codez [/bold red]z[red]messagerfz Unknown errorrLz"[bold red]Agent errors:[/bold red]redreadyz bold greenz bold yellowz6[dim]Start a new active session with[/dim] [bold cyan]z agent start [/bold cyan]z;[dim]For more information check logs with[/dim] [bold cyan]z agent logs r"[z ]Health: ReadyStopped[/]greenyellow)rUrzStatus for agent [bold]rY)r^rsubtitle_alignsubtitle)rr`rrrdrrrdebugstoprbrcrfr rrgrhristr isinstancedictboolr rr1rrar)rorIrKrrrlrfdeployment_table agent_profileintegrated_keysintegrated_keys_enabled krisp_vivakrisp_viva_status audio_filterautoscaling_datascaling_renderables scaling_panel error_panelr error_tablecolorrr.r.r/rds    &                $sessionsz!List active sessions for an agent)hiddenz6Name of the agent to list sessions for e.g. 'my-agent'F)r show_defaultz--idz-izSession ID to filter byz!Organization to list sessions for session_idc+ s4|ptd}|s|r|jr|j}n tdtdS|rttjd|dddm}t j ||||dIdH\}}| |rNtWdS|sdtd |d tWdSt |d |d ppd }|dd} |d r| dkrdnd} nd} d|dd| | r| dkrd| dndd|dt |d d|d rt |d nd|ddurd|ddnd d!|d"rd#nd$g} |d%} | r~| d&g} | d'd(}t| d)kr | d*d+d(| d(d+d(}|d}nd }| d| d,|d-|d.d/d0| D}|r,t|nd}| d1d(}| d2d(}| d3|d4t|d5t|d6d0| D}|rZt|nd}t| d7d(}t| d8d(}| d9|d4t|d5t|ttd:| d;|d<|d=d>d? WddS1swYttjd@|ddd}t j|||dAIdH\}}| |rtWdS|stdB|dCtWdS|dDg}t|}dEd0|D}g}|D]7}z*d(dFlm}||d dGdH}||d dGdH} | |}!||!Wqty5YqwdId0|D}"t|"}#t|}$tdJdK|D}%g}&|$r|#r|d(kr|%|dL}'tdM|dNdOdPtdQ|$d(dRdS|$ddRdT|$d)dRdUdOdPtdV|#d(dRdS|#ddRdT|#d)dRdUdOdPtdW|%dX|dY|'dRdZdOdPg}&tdOdOd[t j!d\}(|("d]|("d^|("d_|("d`|("da|("db|("dc|dDgD]}|r|d|krqt |d |d pd}|dd} |d r| dkr d} nd} nd} |d"dOu})|)rddnd|dt |d |d r0t |d nd|| |ddurA|ddnd|d"dOurKd#n |d"deurTd$ndg}*|)rcfdfd0|*D}*|(j#|*qtjt$|&ry|syt%|&dOdgnd|(dh|di|d=djWddS1swYdS)kNrKNo target agent name providedr#z[dim]Looking up session 'z'[/dim]rNrO)rorrKrr Session 'z ' not foundr\endedAtrwcompletionStatusr"500z[red]Error (500)[/red]Completez[yellow]Active[/yellow]z[bold]Session ID:[/bold] sessionIdz[bold]Status:[/bold] z ()z[bold]Duration:[/bold] z[bold]Created:[/bold] z[bold]Ended:[/bold] z[dim]N/A[/dim]botStartSecondsz[bold]Bot Start:[/bold] sz&[bold]Bot Start:[/bold] [dim]N/A[/dim]z[bold]Cold Start:[/bold] coldStartz[red]Yes[/red]NoresourceMetrics timeseries sampleCountrtz[bold]Resource Metrics[/bold] (z samples over z):cSg|]}|ddqS)crr`r(rr.r.r/ zsessions..cpuMillicoresP50cpuMillicoresP99z CPU: z p50: z p99: cSr)mrrrr.r.r/rrmemoryBytesP50memoryBytesP99z Memory:  z Session details for agent [bold]z[/bold] [dim](z)[/dim]rr^rz![dim]Looking up agent with name 'rqz+No session data found for agent with name 'rsrcSsg|] }|dr|qS)rrrr.r.r/rs)datetimeZz+00:00cSs"g|] }|ddur|dqS)rNrrr.r.r/rscss"|] }|ddurdVqdS)rTr#Nrrr.r.r/r0s zsessions..dz[bold]Total Sessions:[/bold] z Trz[bold]Average Duration:[/bold] r@z s [dim](p5: zs, p95: zs)[/dim]z[bold]Bot Start Time:[/bold] z[bold]Cold Starts:[/bold] /z [dim](z%)[/dim]rQrRz Session IDrWzEnded AtDurationStatuszBot Start Timez Cold Startzon redFcsg|] }dd|dqS)r]rr.)r(cell row_styler.r/rs)equalzSession data for agent z [dim]()r^)&rr`rorrfrbrcrrdr agent_sessionrrrr1r3r>rGr'rErjr r6agent_sessionsr fromisoformatreplace total_seconds BaseExceptionrr4r rrgrhrirr)+ deploy_configrorrIrKrrrlrfsession_durationrdstatus_display info_linesmetricsr sample_count ts_durationts_duration_str cpu_values cpu_sparkcpu_p50cpu_p99 mem_values mem_sparkmem_p50mem_p99 sessions_listtotal_sessionscompleted_sessions durationssessionr created_atended_atduration_secondsbot_start_timesbot_start_metricsduration_metricscold_starts_countmetric_renderablescold_start_percentrm is_cold_startrow_datar.rr/r2sd                "R      ((            $r-z"Modify agent runtime configurationcstddS)NzNot implemented)rrfr.r.r.r/r- sc@seZdZdZdZdS) LogFormatTEXTJSONN)__name__ __module__ __qualname__rrr.r.r.r/r'src@ eZdZdZdZdZdZdZdS)LogLevelDEBUGINFOWARNINGERRORCRITICALNrrrrr r r r r.r.r.r/r, rc@r)LogLevelColorsbluerrrzbold redNr r.r.r.r/r4rrlogszGet logs for the given agent.z--levelz-lzLevel of logs to getz--formatz-fz Logs formatrz--limitz-nzNumber of logs to getz --deploymentz-dzFilter logs by deployment IDz --session-idz-szFilter logs by session IDlevelformatlimit deployment_idcs|ptd}d}|rd|d}|rd|d}tjd|d|d|r(|jnd d d d /tj|||||d IdH\} } | rG| dsXtdt dWdSWdn1sbwY| dD]h} | dd} | rt | dd} t j }t D] }|j| vr|}nq|r|j|jkrqk|tjkrtt|tjj}tjt| ddddtt| |dqk|tjkr| | d}tttj|ddddqkdS)NrKrz deployment (rz session (z[dim]Fetching logs for z : [bold]'z#'[/bold] with severity: [bold cyan]ALLz[/bold cyan][/dim]rNrO)rorKrrrrz"[dim]No logs found for agent[/dim]r#logr" timestampzbold dim)style )r;)rrF) ensure_asciigray)rr`rrdvaluer agent_logsrarbrcrrr upperrrgetattrrrr rjsondumps)rorIrrrrrrK status_textrlrfrlog_datarseverity log_severityrliner.r.r/r<sZ          deletezDelete an agent.z!Organization to delete agent fromz--forcezBypass prompt for confirmationforcecs|ptd}|stdIdHstdtdStj d|dddFt j ||d IdH\}}|rDtdWdS|s^t d |d |d tdWdSt d |d WddS1srwYdS)NrKzAre you sure you want to delete this agent? Note: active sessions will not be interrupted and will continue to run until completion.z$[bold]Aborting delete request[/bold]r#z[dim]Deleting agent: [bold]'rMrNrO)rorKAgent 'z)' not found in namespace / organization 'rsz' deleted successfully)rr` questionaryconfirm ask_asyncrrarbrcrdr agent_deleterfrj)rorIr)rKrlrfr.r.r/r(zs*   " deploymentszGet deployments for an agent.z#Organization to get deployments forc std}|p td}d}ztjd|dddt4IdH#}|jtdj||dd d |id IdH}WdIdHn 1IdHsMwY|jd kr`t |j}| | IdH}t d d dt jd}|d|d|d|d|d|dD])} | didi} || dd| dd| dd| dd| ddqtt|d|dd d!WdWdS1swYWdSty} zt| t|d"|WYd} ~ dSd} ~ ww)#NtokenrKz,[dim]Fetching deployments for agent: [bold]'rMrNrOservices_deployments_path)rKrn AuthorizationzBearer )headersTrQrRIDz Node TypeImagerWrXr/ryrzrZrw dailyNodeTyper{r\r]z[bold]Deployments for agent: rYrrzUnable to get deployments for )rr`rrdaiohttp ClientSessionrconstruct_api_urlrrraise_for_statusr!r rrgrhrirar Exceptionrr api_error) rorIr0rK error_coderresponserlrmrxrzer.r.r/r/sh   (              &,  r:zStart an agent instancez*Name of the agent to start e.g. 'my-agent'zStart Configuration)rrich_help_panelz --api-keyz-kz(Public API key to use for starting agentz--dataz,Data to pass to the agent (stringified JSON)z --use-dailyz-Dz+Create a Daily WebRTC session for the agentz--daily-propertiesz-pz(Daily room properties (stringified JSON)z'Organization which the agent belongs toapi_keyrl use_dailydaily_propertiescs^|ptd}|ptd} |rdntd} |pt} |s0| j} | s.tdtdS| }| sItt dt dt d d d d d tdS|r|rzt |Wn+t j y} ztd|tdt| dtdWYd} ~ Sd} ~ ww|s|pd}|rt|dkr|ddd}tt d|d| d| d|d|d| d|dd d d tdIdHstdtdSttjdd d!d"d#}tj|||d$IdH\}}|r|d%s |td&|d't d(|d)tdWdS|tjd*|d+d d!tj|| ||||d,IdH\}}|r;|tdWdS|td&|d-|rvt|trv|d.}|d/}|rv|d0|}td1td2|d3|d4t|tr|d5}|rtd6|WddSWddSWddS1swYdS)7NrKdefault_public_keyz CLI provideddefault_public_key_namerr#zwNo public API key provided. Please provide a public API key using the --api-key flag or set a default using [bold cyan]zx organizations keys use[/bold cyan]. If you have not yet created a public API key, you can do so by running [bold cyan]z' organizations keys create[/bold cyan].zPublic API Key Requiredrrrz/Invalid JSON format for Daily room properties: z[dim]JSON error: rpNonePMz... Agent Name: z Public API Key: z [dim]z[/dim] Use Daily: z Daily Properties: z Data: z[bold]Start Request for agent: rYz@Are you sure you want to start an active session for this agent?z#[bold]Aborting start request[/bold]z#[dim]Checking agent health...[/dim]rNrOr$)refresh_per_secondrqrr*z]' does not exist or is not in a healthy state. Please check the agent status with [bold cyan]z agent status rz [dim]Agent 'z,' is healthy, sending start request...[/dim])rorBrCrlrDrrz' started successfully dailyRoom dailyTokenz?t=z. Join your session by visiting the link below:z[link=rz[/link]r Session ID: )rr`rrorrfrbrcrar rr!loadsJSONDecodeErrorrr1r+r,r-rrdrrrupdate start_agentrjrr)rror)rBrlrCrDrIrKdefault_public_api_keydefault_public_api_key_namepartial_configdefault_agent_namer@daily_props_displayrr health_datarf daily_room daily_tokenurlrr.r.r/r:s1                 ,.$rzStop an active agent sessionz!Name of the agent e.g. 'my-agent'.zID of the session to stopc s|ptd}|p t}|s"|r|jr|j}n tdtdS|sItt d|d|dddd t d  IdHsItd tdStj d |d |ddd-tj|||dIdH\}}|rstdWdStd|dWddS1swYdS)NrKrr#rJrNz[bold]Stop Session[/bold]rrrz+Are you sure you want to stop this session?z"[bold]Aborting stop request[/bold]z[dim]Stopping session [bold]'z'[/bold] for agent [bold]'rMrNrO)rorrKrz' stopped successfully)rr`rrorrfrbrcrar r+r,r-rdragent_session_terminaterj) rrorrIr)rKrUrlrfr.r.r/rs@       ")r)Ir!enumrtypingrr8r+rblogururrichr rich.columnsr rich.consoler rich.liver rich.panelr rich.tabler rich.textr pipecatcloud._utils.async_utilsr pipecatcloud._utils.auth_utilsr !pipecatcloud._utils.console_utilsrrrr pipecatcloud._utils.deploy_utilsrrpipecatcloud._utils.regionsrrpipecatcloud.clirpipecatcloud.cli.apirpipecatcloud.cli.configrpipecatcloud.constantsrTyper agent_clirHr'floatrr>rErGcommandcreate_blockingOptionArgumentrdrr-rrrrrrr(r/r:rr.r.r.r/s                   <     k   ; % @   & #