Pi
D d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dl m
Z
d dl m Z m
Z
mZ d dlmZ ej e ZdZd Zd
Zed eej ddf fd Zd efd
Zded dfdZddded e dz fdZ dded dfdZ ddded e fdZ d e dz fdZ!dS ) N) Generator)contextmanager)ApiGetModelRequest)build_kaggle_client)get_kaggle_credentialsset_kaggle_api_tokenset_kaggle_credentials)UnauthenticatedErrori z 
Create an API token from your Kaggle
settings page and paste it below.
z
Thank Youreturnc # K t j } t j | }t | | V t | dS # t | w xY w)zCapture output that is logged using the logger.
Example:
```py
>>> output = capture_logger_output()
>>> logger.info("This is an info message")
>>> logger.error("This is an error message")
>>> print(output)
```
N)ioStringIOlogging
StreamHandler_logger
addHandler
removeHandler)bufferhandlers b/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/kagglehub/auth.py_capture_logger_outputr sy []]F#F++G w ' g&&&&& g&&&&s A% % Bc ddl m} | j }|j D ]}|j dk r dS dS # t
t f$ r Y dS w xY w)zReturn `True` if code is executed in a notebook (Jupyter, Kaggle Notebook, Colab, QTconsole).
Taken from https://stackoverflow.com/a/39662359.
Adapted to make it work with Google colab as well.
r ) get_ipythonZMQInteractiveShellTF)IPythonr __class____mro____name__ NameErrorModuleNotFoundError)r shell_class parent_classs r _is_in_notebookr# 5 s ''''''!kmm- ' / L$(===tt >u * + uus *1 1 AAvalidate_credentialsc d} ddl m} ddlm n # t $ r t | w xY w dddd } d
d } t | t g|
| dt ddf fd }|
| dS )zRPrompt the user for their Kaggle token and save it in a widget (Jupyter or Colab).z;You need the `ipywidgets` module: `pip install ipywidgets`.r )display)widgetsflexcolumncenterz50%)r& flex_flow align_itemswidthz
API Token:) descriptionLogin)layout_r Nc j }d_ d g_ t 5 }t | du rt d d d n # 1 swxY w Y | }n&# t $ r}t | }Y d }~nd }~ww xY wfd| d D _ d S )N z
Connecting...Tc b g | ]+}| | ,S )stripLabel).0liner' s r
zB_notebook_login..on_click_login_button..s s7 &k&k&kt^b^h^h^j^j&kw}}T':':&k&k&k
)
valuer7 childrenr r _validate_credentials_helpergetvalue Exceptionstrsplit) r1 tokencapturedmessageerrorlogin_token_widget token_widgetr$ r' s r on_click_login_buttonz._notebook_login..on_click_login_button` s3 " (/}}_'E'E&F#
!'))
3X$U+++ (4//0222
3
3
3
3
3
3
3
3
3
3
3
3
3
3
3 ''))GG ! ! !%jjGGGGGG ! 'l&k&k&kw}}UYGZGZ&k&k&k###s: B "A( B (A,,B /A,0B
B+B&&B+)IPython.displayr&
ipywidgetsr' ImportErrorLayoutPasswordButtonVBoxHTML NOTEBOOK_LOGIN_TOKEN_HTML_START NOTEBOOK_LOGIN_TOKEN_HTML_ENDrB on_click) r$
library_errorr&
box_layout login_buttonrJ rH rI r' s ` @@@r _notebook_loginrY G se QM)++++++&&&&&&& ) ) )-((() (PX`e ffJ## #==L>>g>66L LL8 9 9 LL6 7 7
& G l l l l l l l l l l l* /00000s /Tverboser[ c H t 5 }| }| | ddt }|j | }d|v r0| rt d |d cd d d S d|v r.|d t k r | rt d n | rt d d d d d S # 1 swxY w Y d S )Nzapi.v1.DiagnosticsServiceHellouserNamez*Kaggle credentials successfully validated.codezInvalid Kaggle credentials. You can obtain a Kaggle API token on your [Kaggle settings page](https://www.kaggle.com/settings/account).z3Unable to validate Kaggle credentials at this time.)
r http_client
_init_session_prepare_requestr _sessionsendjsonr infoINVALID_CREDENTIALS_ERRORrG warning)r[
api_clientr` requestresponses r r? r? x s * ,,.. !!### ../JGUgUiUijj ',,W55::<< ! !
K IJJJJ' x
HV$48Q$Q$Q
] S OO Q R R R# s B D;ADD Dc t rt | dS t j d }t | | sdS t dS )z@Prompt the user for their Kaggle API token and save it globally.Nz9Enter your Kaggle API token (input will not be visible): )r# rY getpassr r? )r$ rD s r loginrn sd ],--- [\\ """""r; c t st t | }|rd|iS t # t $ r}t |d}~ww xY w)zv
Return a dictionary with the username of the authenticated Kaggle user or raise an error if unauthenticated.
rZ usernameN)r r
r? rA )r[ rp es r whoamirr s " # # %"$$$,/@@@ * ) )"$$$ , , ,"$$!+,s A A
A AA c R t d d S # t $ r Y dS w xY w)zWReturns the username of the authenticated logged-in user if configured, otherwise None.FrZ rp N)rr rA r5 r; r get_usernamert s? e$$$Z00 tts
&&)T)"rm r
r collections.abcr
contextlibr (kagglesdk.models.types.model_api_servicer kagglehub.clientsr kagglehub.configr r r kagglehub.exceptionsr
getLoggerr r rg rS rT r r boolr# rY rB r? rn dictrr rt r5 r; r r~ s % % % % % % % % % % % % G G G G G G 1 1 1 1 1 1 6 5 5 5 5 5
'
H
%
% #8 ! ' "+tT*A B ' ' ' '( $.1$ .14 .1 .1 .1 .1b 59 T S4Z *# # # # # # #" # , , ,t ,t , , , , cDj r;