U h(@sddlZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z m Z m Z ddlmZddlmZejejddeeZeejZe e ee ee effddd d ZGd d d Zdd ddZdd ddZedkredS)N)Path)DictListMappingOptionalUnioncastenv)utilsz %(message)s)levelformat)services_statusreturnc Csg}|D]Z}t|dt|dd}ttt|dg}|rXddd|D|d<||qtd d |D}td d |D}d d|d d|d dg}|D]F}|d|d }|d|d } |dd} ||| | q|dt d |dS)NServiceStatus)rrZ Publishersz, cSsg|]}t|dqS)Z PublishedPort)str).0Z publisherr6/tmp/pip-unpacked-wheel-cqvhoa9t/langsmith/cli/main.py sz#pprint_services..ZPublishedPortscss|]}t|dVqdS)rNlenrservicerrr "sz"pprint_services..css|]}t|dVqdS)rNrrrrrr#s zPublished Portsz To connect, set the following environment variables in your LangChain application: LANGSMITH_TRACING_V2=true LANGSMITH_ENDPOINT=http://localhost:80/api) rrrrgetjoinappendmaxljustloggerinfo) rZservicesrZservice_statusZ publishersZmax_service_lenZ max_state_lenZservice_messageZ service_strZ state_strZ ports_strrrrpprint_servicess<       r&c@seZdZdZddddZeeedddZeddd d Z ddd d Z d deddddZ dd de eeeddddZ deddddZddddZddddZdS)LangSmithCommandz$Manage the LangSmith Tracing server.NrcCsttjd|_dS)Nzdocker-compose.yaml)r__file__absoluteparentdocker_compose_fileselfrrr__init__<szLangSmithCommand.__init__cCstSN)ls_utilsZget_docker_compose_commandr-rrrdocker_compose_commandAsz'LangSmithCommand.docker_compose_command)urlrcCs,ztd|gWntk r&YnXdS)Nopen) subprocessrunFileNotFoundError)r.r3rrr _open_browserEszLangSmithCommand._open_browsercCsN|jdt|jf}t|d tdtdtd|ddS) N-fup --quiet-pull--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 LANGSMITH_TRACING=truez, LANGSMITH_ENDPOINT=http://localhost:80/api zhttp://localhost)r:r;r<)r2rr,r5r6r$r%r8)r.commandrrr _start_localKs"  zLangSmithCommand._start_local0.5.7version)rArcCs*|tjd<t|jdt|jdfdS)zPull the latest LangSmith images. Args: version: The LangSmith version to use for LangSmith. Defaults to 0.5.7 Z_LANGSMITH_IMAGE_VERSIONr9pullN)osenvironr5r6r2rr,)r.rArrrrBes zLangSmithCommand.pull)openai_api_keyrA)rElangsmith_license_keyrArcCs<|dk r|tjd<|dk r$|tjd<|j|d|dS)aRun the LangSmith server locally. Args: 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. version: The LangSmith version to use for LangSmith. Defaults to latest. NOPENAI_API_KEYLANGSMITH_LICENSE_KEYr@)rCrDrBr>)r.rErFrArrrstartys    zLangSmithCommand.startF) clear_volumesrcCsR|jdt|jdf}|rDtd}|dkr:tddS|dt|dS)Stop the LangSmith server.r9ZdownzYou 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) r2rr,inputlowerprintr!r5r6)r.rJcmdZconfirmrrrstops  zLangSmithCommand.stopcCs t|jdt|jdfdS)z)Print the logs from the LangSmith server.r9logsN)r5r6r2rr,r-rrrrRszLangSmithCommand.logscCs|jdt|jdddf}tj|tjtjd}z|jd}t |}Wn"tj k rjt dYdSX|rt d t|nt d dSdS) z6Provide information about the status LangSmith server.r9Zpsz--formatjson)stdoutstderrzutf-8z'Error checking LangSmith server status.Nz*The LangSmith server is currently running.z$The LangSmith server is not running.)r2rr,r5r6PIPErTdecoderSloadsJSONDecodeErrorr$errorr%r&)r.r=resultZcommand_stdoutrrrrstatuss.      zLangSmithCommand.status)F)__name__ __module__ __qualname____doc__r/propertyrrr2r8r>rBrrIboolrQrRr\rrrrr'9s,   r'r(cCsrt}|t|ttdd|D}td| D] \}}t||d|qLdS)z*Print the runtime environment information.css|]}t|VqdSr0r)rkeyrrrrszenv..zLangChain Environment:z: N) ls_envZget_runtime_environmentupdateZget_docker_environmentZget_langchain_env_varsr"keysr$r%items)r Zmax_key_lengthkvrrrr s r c sxtdtdt}|jdd}t|jddd}|jdtdd d |jd td d d |jdddd |j fddd|jddd}|jdddd|j fddd|jddd}|jdddd |j fddd|jddd}|j fd dd|jd!d"d}|j fd#dd|d$}|j d%dd| }t |d&sj| d'S| |d'S)(zMain entrypoint for the CLI.zThe LangSmith version to use for LangSmith. Defaults to 0.5.7.csj|j|j|jdS)N)rErFrA)rIrErFrAargsZserver_commandrrszmain..)funcrQrKz--clear-volumes store_truez?Delete all the locally cached LangSmith containers and volumes.)actionrlcsj|jdS)N)rJ)rQrJrmrorrrprBz!Pull the latest LangSmith images.csj|jdS)Nr@)rBrArmrorrrprtrRzShow the LangSmith server logs.csSr0)rRrmrorrrp#rtr\z!Show the LangSmith server status.csSr0)r\rmrorrrp'rtr cSstSr0r rmrrrrp)rtrqN)rOargparseArgumentParseradd_subparsersr' add_parser add_argumentrCgetenv set_defaults parse_argshasattr print_helprq) parserZ subparsersZserver_start_parserZserver_stop_parserZserver_pull_parserZserver_logs_parserZserver_status_parserZ env_parserrnrrormains      r__main__)rurSloggingrCr5pathlibrtypingrrrrrrZ langsmithr rdr r1 basicConfigINFO getLoggerr]r$r)r+Z_DIRrr&r'rrrrrs$      ('R