o ig@sddlmZddlmZddlmZddlmZmZddl m Z ddl m Z ddl mZdd lmZmZdd lmZd eZGd d d eZGdddeZGdddZdS)) annotations)Any)urljoin)ResponseSession)Self)LogfireConfigError)VERSION) UserTokenUserTokenCollection)UnexpectedResponsezlogfire/c@s eZdZdS)ProjectAlreadyExistsN)__name__ __module__ __qualname__rr^/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/logfire/_internal/client.pyrsrc@seZdZdddZdS) InvalidProjectNamereasonstrreturnNonecCs ||_dSN)r)selfrrrr__init__s zInvalidProjectName.__init__N)rrrr)rrrrrrrrrsrc@seZdZdZd4ddZed5d d Zd6d7ddZd dd8ddZd6d9ddZ d dd:dd Z d;d"d#Z dd-d.Zd>d/d0Zd?d2d3Zd S)@ LogfireClientzA Logfire HTTP client to interact with the API. Args: user_token: The user token to use when authenticating against the API. user_tokenr rrcCs@|jrtd|j|_|j|_t|_|jj|jt ddS)Nz"The provided user token is expired) Authorizationz User-Agent) is_expired RuntimeErrorbase_urltoken_tokenr_sessionheadersupdate UA_HEADER)rrrrrr"s zLogfireClient.__init__r! str | NonercCs|t|dS)aCreate a client from the provided base URL. Args: base_url: The base URL to use when looking for a user token. If `None`, will prompt the user into selecting a token from the token collection (or, if only one available, use it directly). The token collection will be created from the `~/.logfire/default.toml` file (or an empty one if no such file exists). )r)r get_token)clsr!rrrfrom_url*s zLogfireClient.from_urlNendpointrparamsdict[str, Any] | NonercC&|jjt|j||d}t||S)Nr-)r$getrr!r raise_for_status)rr,r-responserrr_get_raw6 zLogfireClient._get_rawr0 error_messagerc C6z |||WSty}zt||d}~wwr)r4jsonr r)rr,r-r6errr_get;  zLogfireClient._getbody Any | NonecCr/)N)r8)r$postrr!r r2)rr,r<r3rrr _post_rawAr5zLogfireClient._post_rawr<c Cr7r)r?r8r r)rr,r<r6r9rrr_postFr;zLogfireClient._postlist[dict[str, Any]]cC|jdddS)z,Get the organizations of the logged-in user.z)/v1/organizations/available-for-projects/z&Error retrieving list of organizationsr6r:rrrrget_user_organizationsLsz$LogfireClient.get_user_organizationsdict[str, Any]cCrC)z)Get information about the logged-in user.z/v1/account/mez!Error retrieving user informationrDrErFrrrget_user_informationRz"LogfireClient.get_user_informationcCrC)z'Get the projects of the logged-in user.z/v1/writable-projects/z!Error retrieving list of projectsrDrErFrrrget_user_projectsVrJzLogfireClient.get_user_projects organization project_namec Csz|jd|dd|id}W|StyG}z)|j}|jdkr$t|jdkr?|dd}|d d dgkr?t|d td d }~ww)zCreate a new project. Args: organization: The organization that should hold the new project. project_name: The name of the project to be created. Returns: The newly created project. /v1/organizations/z /projectsrMr@iidetailrlocr<msgzError creating new projectN)r?r r3 status_coderr8rr)rrLrMr3r9rerrorrrrcreate_new_projectZs     z LogfireClient.create_new_projectcCs|jd|d|dddS)zECreate a write token for the given project in the given organization.rN /projects/z/write-tokens/z"Error creating project write tokenrDrArrLrMrrrcreate_write_tokenrsz LogfireClient.create_write_tokencCs"|jd|d|dddiddS)zDCreate a read token for the given project in the given organization.rNrVz /read-tokens descriptionzCreated by Logfire CLIz!Error creating project read token)r<r6rWrXrrrcreate_read_tokeny zLogfireClient.create_read_tokenissuecCs"|jd|d|dd|iddS)z/Get a prompt to be used with your favorite LLM.rNrVz/promptsr]zError retrieving prompt)r-r6rE)rrLrMr]rrr get_promptr\zLogfireClient.get_prompt)rr rr)r!r(rrr)r,rr-r.rr)r,rr-r.r6rrr)r,rr<r=rr)r,rr<r=r6rrr)rrB)rrH)rLrrMr)rLrrMrrrH)rLrrMrr]rrrH)rrr__doc__r classmethodr+r4r:r?rArGrIrKrUrYr[r^rrrrrs           rN) __future__rtypingr urllib.parserrequestsrrtyping_extensionsrlogfire.exceptionsrlogfire.versionr authr r utilsr r' Exceptionrrrrrrrs