o +if.@sddlZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z m Z m Z mZddlmZddlmZejejddeeZeejZde e ee ee effd dfd d ZGd d d ZdddZdddZedkryedSdS)N)Path)DictListLiteralMappingOptionalUnioncastenv)utilsz %(message)s)levelformatservices_statusreturnc Csg}|D]-}t|dt|dd}ttt|dg}|r,ddd|D|d<||qtd d |D}td d |D}d d|d d|d dg}|D]#}|d|d }|d|d } |dd} ||| | qY|dt d |dS)NServiceStatus)rr Publishersz, cSsg|]}t|dqS) PublishedPort)str).0 publisherrK/var/www/html/corbot_env/lib/python3.10/site-packages/langsmith/cli/main.py sz#pprint_services..PublishedPortscs|] }t|dVqdS)rNlenrservicerrr "z"pprint_services..csr)rNrrrrrr!#r" zPublished Portsz To connect, set the following environment variables in your LangChain application: LANGCHAIN_TRACING_V2=true LANGCHAIN_ENDPOINT=http://localhost:80/api) rr rrgetjoinappendmaxljustloggerinfo) rservicesr service_status publishersmax_service_len max_state_lenservice_message service_str state_str ports_strrrrpprint_servicess<       r6c @seZdZdZd#ddZedeefddZdeddfd d Z d#d d Z d ddde e d e de dfdeddfddZ d dddde e d e de dfdeedededdf ddZd$deddfddZd#dd Zd#d!d"ZdS)%LangSmithCommandz$Manage the LangSmith Tracing server.rNcCsttjd|_dS)Nzdocker-compose.yaml)r__file__absoluteparentdocker_compose_fileselfrrr__init__<szLangSmithCommand.__init__cCstSN)ls_utilsget_docker_compose_commandr<rrrdocker_compose_commandAsz'LangSmithCommand.docker_compose_commandurlcCs*z td|gWdStyYdSw)Nopen) subprocessrunFileNotFoundError)r=rCrrr _open_browserEs  zLangSmithCommand._open_browsercCs^g|jdt|j}tg|dddtdtdtd|ddS) N-fupz --quiet-pullz--waitaLangSmith server is running at http://localhost:80/api. To view the app, navigate your browser to http://localhost:80 To connect your LangChain application to the server locally, set the following environment variable when running your LangChain application. z LANGCHAIN_TRACING_V2=truez, LANGCHAIN_ENDPOINT=http://localhost:80/api zhttp://localhost)rBrr;rErFr+r,rH)r=commandrrr _start_localKs0  zLangSmithCommand._start_localprodlateststageversionrPdevbetarQcCsV|dkr dtjd<n |dkrdtjd<|tjd<tg|jdt|jdd S) zPull the latest LangSmith images. Args: stage: Which stage of LangSmith images to pull. One of "prod", "dev", or "beta". rRzdev-_LANGSMITH_IMAGE_PREFIXrSzrc-_LANGSMITH_IMAGE_VERSIONrIpullN)osenvironrErFrBrr;)r=rPrQrrrrVes    zLangSmithCommand.pull)rPopenai_api_keyrQrYlangsmith_license_keycCs>|dur |tjd<|dur|tjd<|j||d|dS)aRun the LangSmith server locally. Args: expose: If True, expose the server to the internet using ngrok. auth_token: The ngrok authtoken to use (visible in the ngrok dashboard). If not provided, ngrok server session length will be restricted. stage: Which set of images to pull when running. One of "prod", "dev", or "beta". openai_api_key: The OpenAI API key to use for LangSmith If not provided, the OpenAI API Key will be read from the OPENAI_API_KEY environment variable. If neither are provided, some features of LangSmith will not be available. langsmith_license_key: The LangSmith license key to use for LangSmith If not provided, the LangSmith license key will be read from the LANGSMITH_LICENSE_KEY environment variable. If neither are provided, Langsmith will not start up. NOPENAI_API_KEYLANGSMITH_LICENSE_KEYrO)rWrXrVrL)r=rPrYrZrQrrrstarts    zLangSmithCommand.startF clear_volumescCsXg|jdt|jd}|r%td}|dkr tddS|dt|dS)Stop the LangSmith server.rIdownzYou are about to delete all the locally cached LangSmith containers and volumes. This operation cannot be undone. Are you sure? [y/N]yz Aborting.Nz --volumes) rBrr;inputlowerprintr(rErF)r=r^cmdconfirmrrrstops$  zLangSmithCommand.stopcCs&tg|jdt|jddS)z)Print the logs from the LangSmith server.rIlogsN)rErFrBrr;r<rrrrhszLangSmithCommand.logscCsg|jdt|jddd}tj|tjtjd}z |jd}t |}Wntj y9t dYdSw|rGt d t|dSt d dS) z6Provide information about the status LangSmith server.rIpsz--formatjson)stdoutstderrzutf-8z'Error checking LangSmith server status.Nz*The LangSmith server is currently running.z$The LangSmith server is not running.)rBrr;rErFPIPErkdecoderjloadsJSONDecodeErrorr+errorr,r6)r=rKresultcommand_stdoutrrrrstatuss<      zLangSmithCommand.statusrN)F)__name__ __module__ __qualname____doc__r>propertyrrrBrHrLrrrVrr]boolrgrhrtrrrrr79sB      r7cCsrt}|t|ttdd|D}td| D]\}}t||d|q&dS)z*Print the runtime environment information.css|]}t|VqdSr?r)rkeyrrrr!szenv..zLangChain Environment:z: N) ls_envget_runtime_environmentupdateget_docker_environmentget_langchain_env_varsr)keysr+r,items)r max_key_lengthkvrrrr s r c stdtdt}|jdd}t|jddd}|jddgd d d |jd td dd|jdtddd|jdddd|j fddd|jddd}|jdddd|j fddd|jd d!d}|jddgd d"d |jdddd|j fd#dd|jd$d%d}|j fd&dd|jd'd(d}|j fd)dd|d*}|j d+dd| }t |d,s| d-S| |d-S).zMain entrypoint for the CLI.zs zmain..)funcrgr_z--clear-volumes store_truez?Delete all the locally cached LangSmith containers and volumes.)actionrcsj|jdS)N)r^)rgr^rrrrr(srVz!Pull the latest LangSmith images.z(Which stage of LangSmith images to pull.csj|j|jdS)NrO)rVrPrQrrrrr=srhzShow the LangSmith server logs.cSr?)rhrrrrrBrtz!Show the LangSmith server status.crr?)rtrrrrrFrr cSstSr?r rrrrrHsrN)rdargparseArgumentParseradd_subparsersr7 add_parser add_argumentrWgetenv set_defaults parse_argshasattr print_helpr) parser subparsersserver_start_parserserver_stop_parserserver_pull_parserserver_logs_parserserver_status_parser env_parserrrrrmains       r__main__ru) rrjloggingrWrEpathlibrtypingrrrrrrr langsmithr r}r r@ basicConfigINFO getLoggerrvr+r8r:_DIRrr6r7rrrrrs( $    *' & e