o ZhM @sdZddlZddlZddlmZmZmZddlmZm Z m Z m Z m Z m Z mZmZddlZddlmZddlmZmZmZdZeGdd d ZeGd d d eZeGd d d eZeGdddeZeGdddeZeGdddeZ d4dedeedede fddZ dedeedefddZ! d4de"dede e e e e"ffddZ#ded e"de e$fd!d"Z%d5d#e"d$e"de edefd%d&Z&d$e"de e$fd'd(Z'd)e$d$e"de e$fd*d+Z(d,e$d$e"de e$fd-d.Z)d,e$d/e$d$e"de e$fd0d1Z*Gd2d3d3eZ+dS)6zUtil that calls clickup.N)asdict dataclassfields)AnyDictListMappingOptionalTupleTypeUnion)get_from_dict_or_env) BaseModel ConfigDictmodel_validatorzhttps://api.clickup.com/api/v2c@s.eZdZdZedeeefddfddZdS) ComponentzBase class for all components.datareturncCstN)NotImplementedErrorclsrr\/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/utilities/clickup.py from_dataszComponent.from_dataN) __name__ __module__ __qualname____doc__ classmethodrstrrrrrrrrs rc@seZdZUdZeed<eed<eed<eed<eed<eed<eed<eed <eeee fed <eeee fed <e eed <e eed <e eed<eed<eed<eed<e deee fddfddZ dS)TaskzClass for a task.idname text_content descriptionstatus creator_idcreator_username creator_email assigneeswatchersprioritydue_date start_datepointsteam_id project_idrrcCs|ddurdn|dd}|did|dd|dd|dd|dd|ddd|ddd |dd d |dd d |d d|dd|d|dd|dd|dd|dd|ddS)Nr,r"r#r$r%r&r'Zcreatorr(usernamer)emailr*r+r-r.r/r0r1projectrr)rrr,rrrr-sD          zTask.from_dataN) rrrrint__annotations__r rrrr rrrrrrr!s(     r!c@seZdZUdZeed<eed<dZeeed<dZ ee ed<dZ ee ed<dZ ee ed<dZee ed <dZeeed <ed ed dfd dZdS)CUListzComponent class for a list. folder_idr#Ncontentr- due_date_timer,assigneer&rrc CsD||d|d|d|d|d|d|d|dd S) Nr8r#r9r-r:r,r;r&)r8r#r9r-r:r,r;r&)getrrrrrQszCUList.from_data)rrrrfloatr6r r9r r-r5r:boolr,r;r&rdictrrrrrr7Ds r7c@sHeZdZUdZeed<eed<eed<eed<ededdfdd Z d S) MemberzComponent class for a member.r"r2r3initialsrrcCs0||dd|dd|dd|dddS)Nuserr"r2r3rA)r"r2r3rArrrrrrhs     zMember.from_dataN) rrrrr5r6r rrrrrrrr@_s r@c@sDeZdZUdZeed<eed<eeed<e de ddfddZ d S) TeamzComponent class for a team.r"r#membersrrcCs(dd|dD}||d|d|dS)NcSsg|]}t|qSr)r@r).0Z member_datarrr |sz"Team.from_data..rDr"r#)r"r#rDr)rrrDrrrrzszTeam.from_dataN) rrrrr5r6r rr@rrrrrrrrCrs  rCc@sXeZdZUdZeed<eed<eed<eee fed<e deee fddfdd Z d S) SpacezComponent class for a space.r"r#privateenabled_featuresrrcCs>|dd}dd|dD}||d|d|d|d S) NspacesrcSsi|] \}}|dr||qS)enabledr)rEfeaturevaluerrr s z#Space.from_data..featuresr"r#rH)r"r#rHrI)items)rrZ space_datarIrrrrs  zSpace.from_dataN) rrrrr5r6r r>rrrrrrrrrGs  rGFr componentfault_tolerantrc Csfzt||WSty2}z|r,dt|dt|d}t||WYd}~S|d}~ww)aParse a dictionary by creating a component and then turning it back into a dictionary. This helps with two things 1. Extract and format data from a dictionary according to schema 2. Provide a central place to do this in a fault-tolerant way z(Error encountered while trying to parse : z' Falling back to returning input data.N)rr Exceptionr warningswarn)rrQrReZ warning_strrrrparse_dict_through_components  rXcCs0i}t|D]}|j|vr||j||j<q|S)a"Extract elements from a dictionary. Args: data: The dictionary to extract elements from. component: The component to extract elements from. Returns: A dictionary containing the elements from the input dictionary that are also in the component. )rr#)rrQoutput attributerrr+extract_dict_elements_from_component_fieldss   r[queryc CsTzt|dfWStjy)}z|r#ddt|dfWYd}~S|d}~ww)aDParse a JSON string and return the parsed object. If parsing fails, returns an error message. :param query: The JSON string to parse. :return: A tuple containing the parsed object or None and an error message or None. Exceptions: json.JSONDecodeError: If the input is not a valid JSON string. Nz5Input must be a valid JSON. Got the following error: z". "Please reformat and try again.)jsonloadsJSONDecodeErrorr )r\rRrWrrr load_querys r`keycCsX||vr*t||dkr*t||dkr"td|d||d||ddSdS)z%Fetch the first id from a dictionary.rzFound multiple rSz. Defaulting to first.r"N)lenrUrV)rrarrrfetch_first_ids rdurl access_tokencCs(d|i}tj|||d}||S)zFetch data from a URL. Authorizationheadersparams)requestsr<raise_for_statusr])rerfr\riresponserrr fetch_datasrncCstd}t||}t|dS)zFetch the team id./teamteams DEFAULT_URLrnrd)rfrerrrr fetch_team_ids   rsr0cC,td|d}t||ddid}t|dS)zFetch the space id./team//spacearchivedfalser\rJrq)r0rfrerrrrfetch_space_id rzspace_idcCrt)zFetch the folder id./space//folderrwrxryfoldersrq)r|rfrerrrrfetch_folder_idr{rr8cCsV|r td|d}ntd|d}t||ddid}|r&d|vr&|dSt|dS) zFetch the list id./folder//listr}rwrxryr"listsrq)r|r8rfrerrrr fetch_list_ids  rc @seZdZUdZdZeeed<dZeeed<dZ eeed<dZ eeed<dZ eeed<e dd Z e d>d ed ed efddZed ededed eefddZeddeded efddZded eeeeffddZd eeeeefffddZd efddZd eeeffd d!Zd efd"d#Zd?d%ed&ed efd'd(Z d efd)d*Z!d%ed efd+d,Z"d efd-d.Z#d%ed efd/d0Z$d%ed efd1d2Z%d%ed efd3d4Z&d%ed efd5d6Z'd%ed efd7d8Z(d%ed efd9d:Z)d;ed%ed efd Add or remove assignees of a specified task. TrNrZusersz3All users must be integers, not strings! "Got user z if type rrZ operationadd)rremrzInvalid operation (z). zValid options ['add', 'rem'].rrr*rrm) r`rr5typerrr ValueErrorr0rkr) rr\rrrBreriZassigne_payloadrjrrmrrrupdate_task_assigneess:    z'ClickupAPIWrapper.update_task_assigneesc Cs|t|dd\}}|durd|iS|j}td|d}d|jd}t|t}|}tj||||d } | } t | tddS) z% Creates a new task. TrNrrrr)rr0)r]rirj) r`rrrr0r[r!rrkrr]rX) rr\rrrrerjrrirmrrrr create_tasks  zClickupAPIWrapper.create_taskc Cst|dd\}}|durd|iS|jr|jn|j}td|d}t|t}|}tj|||d}| } t | t dd} d| vrF| d|_ | S) z% Creates a new list. TrNrrrr]rir") r`r8r|rrr[r!rrkrr]rXr7r) rr\rrlocationrerrirmrZ parsed_listrrr create_list$s  zClickupAPIWrapper.create_listc Csxt|dd\}}|durd|iS|j}td|d}d|di}|}tj|||d}|} d | vr:| d |_| S) z' Creates a new folder. TrNrr}r~r#rr")r`r|rrrrkrr]r) rr\rrr|rerrirmrrrr create_folder<s zClickupAPIWrapper.create_folderrcCs |dkr ||}nh|dkr||}n^|dkr|}nU|dkr'||}nK|dkr1||}nA|dkr;||}n7|dkrD|}n.|dkrM|}n%|d krV|}n|d kr`| |}n|d krj| |}nd d |di}zt |WSt yt|YSw)z Run the API.rrZ get_teamsrrrrrrrrZ ModeErrorzGot unexpected mode .)rrrrrrrrrrrr]dumpsrTr )rrr\rYrrrrunTs8              zClickupAPIWrapper.run)r)T)+rrrrrfr r r6r0r|r8rrZ model_configrrrrrrrr?rrrr bytesrrrrr>rrrrrrrrrrrrrrrrsd          'r)Fr),rr]rU dataclassesrrrtypingrrrrr r r r rkZlangchain_core.utilsr Zpydanticrrrrrrr!r7r@rCrGr?r>rXr[r r`r5rdrnrsrzrrrrrrrsj( +