j9zddlZddlZddlmZmZmZmZmZddlm Z m Z ddl m Z ddl mZddlmZGdde Zy) N)AnyDictListOptionalcast) AgentAction AgentFinish)BaseCallbackHandler) LLMResult)parsec eZdZUdZdZeed<edZeed<dZeed<dZ eed < d-d ed e ed e ede edd f fd Z de ee fdeede dd fdZdede dd fdZdede dd fdZdede dd fdZde ee fde ee fde dd fdZde ee fde dd fdZdede dd fdZde ee fd ede dd fd!Zd"ede de fd#Z d.d$e d%e ed&e ede dd f d'Zdede dd fd(Zd)ede dd fd*Zd+ede dd fd,ZxZ S)/ArgillaCallbackHandleraCallback Handler that logs into Argilla. Args: dataset_name: name of the `FeedbackDataset` in Argilla. Note that it must exist in advance. If you need help on how to create a `FeedbackDataset` in Argilla, please visit https://docs.argilla.io/en/latest/tutorials_and_integrations/integrations/use_argilla_callback_in_langchain.html. workspace_name: name of the workspace in Argilla where the specified `FeedbackDataset` lives in. Defaults to `None`, which means that the default workspace will be used. api_url: URL of the Argilla Server that we want to use, and where the `FeedbackDataset` lives in. Defaults to `None`, which means that either `ARGILLA_API_URL` environment variable or the default will be used. api_key: API Key to connect to the Argilla Server. Defaults to `None`, which means that either `ARGILLA_API_KEY` environment variable or the default will be used. Raises: ImportError: if the `argilla` package is not installed. ConnectionError: if the connection to Argilla fails. FileNotFoundError: if the `FeedbackDataset` retrieval from Argilla fails. Examples: >>> from langchain_community.llms import OpenAI >>> from langchain_community.callbacks import ArgillaCallbackHandler >>> argilla_callback = ArgillaCallbackHandler( ... dataset_name="my-dataset", ... workspace_name="my-workspace", ... api_url="http://localhost:6900", ... api_key="argilla.apikey", ... ) >>> llm = OpenAI( ... temperature=0, ... callbacks=[argilla_callback], ... verbose=True, ... openai_api_key="API_KEY_HERE", ... ) >>> llm.generate([ ... "What is the best NLP-annotation tool out there? (no bias at all)", ... ]) "Argilla, no doubt about it." z%https://github.com/argilla-io/argillaREPO_URLz/issues ISSUES_URLzphttps://docs.argilla.io/en/latest/tutorials_and_integrations/integrations/use_argilla_callback_in_langchain.htmlBLOG_URLzhttp://localhost:6900DEFAULT_API_URLN dataset_nameworkspace_nameapi_urlapi_keyreturncxt | ddl}|j|_t |jt dkrt d|jd|Dtjd/tjd|jd |j}|ntjd Yt |jt d krd nd |_ tjd|jd|j} |j||||_|xs|j%|_ i}t |jt dkr,tjd|jdt(ddi}|j*j,d'|j"|j&d||_dd g}||j.j4D cgc]} | j6c} k7ret9d!|j"d|j&d"|d#|j.j4D cgc]} | j6c} d$|j2d% i|_tjd&|j dy#t $r t dwxYw#t$r"}td|d|j d|d}~wwxYw#t$rI}t1d|d|j"d|j&d|j2d|j d |d}~wwxYwcc} wcc} w)(a)Initializes the `ArgillaCallbackHandler`. Args: dataset_name: name of the `FeedbackDataset` in Argilla. Note that it must exist in advance. If you need help on how to create a `FeedbackDataset` in Argilla, please visit https://docs.argilla.io/en/latest/tutorials_and_integrations/integrations/use_argilla_callback_in_langchain.html. workspace_name: name of the workspace in Argilla where the specified `FeedbackDataset` lives in. Defaults to `None`, which means that the default workspace will be used. api_url: URL of the Argilla Server that we want to use, and where the `FeedbackDataset` lives in. Defaults to `None`, which means that either `ARGILLA_API_URL` environment variable or the default will be used. api_key: API Key to connect to the Argilla Server. Defaults to `None`, which means that either `ARGILLA_API_KEY` environment variable or the default will be used. Raises: ImportError: if the `argilla` package is not installed. ConnectionError: if the connection to Argilla fails. FileNotFoundError: if the `FeedbackDataset` retrieval from Argilla fails. rNzTo use the Argilla callback manager you need to have the `argilla` Python package installed. Please install it with `pip install argilla`z1.8.0z#The installed `argilla` version is z} but `ArgillaCallbackHandler` requires at least version 1.8.0. Please upgrade `argilla` with `pip install --upgrade argilla`.ARGILLA_API_URLz[Since `api_url` is None, and the env var `ARGILLA_API_URL` is not set, it will default to `z6`, which is the default API URL in Argilla Quickstart.ARGILLA_API_KEYz1.11.0z admin.apikeyz owner.apikeyz[Since `api_key` is None, and the env var `ARGILLA_API_KEY` is not set, it will default to `z6`, which is the default API key in Argilla Quickstart.)rrz.Could not connect to Argilla with exception: 'z'. Please check your `api_key` and `api_url`, and make sure that the Argilla server is up and running. If the problem persists please report it to z as an `integration` issue.1.14.0zYou have Argilla z., but Argilla 1.14.0 or higher is recommended. with_recordsF)name workspacez@`FeedbackDataset` retrieval from Argilla failed with exception `z+`. Please check that the dataset with name=z in the workspace=zz exists in advance. If you need help on how to create a `langchain`-compatible `FeedbackDataset` in Argilla, please visit z.. If the problem persists please report it to promptresponsez`FeedbackDataset` with name=z] had fields that are not supported yet for the`langchain` integration. Supported fields are: z/, and the current `FeedbackDataset` fields are zl. For more information on how to create a `langchain`-compatible `FeedbackDataset` in Argilla, please visit .zThe `ArgillaCallbackHandler` is currently in beta and is subject to change based on updates to `langchain`. Please report any issues to )super__init__argilla __version__ARGILLA_VERSION ImportErrorr osgetenvwarningswarnrDEFAULT_API_KEYinit ExceptionConnectionErrorrr get_workspacer UserWarningFeedbackDataset from_argilladatasetFileNotFoundErrorrfieldsr ValueErrorprompts) selfrrrrrge extra_argssupported_fieldsfield __class__s x/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langchain_community/callbacks/argilla_callback.pyr$zArgillaCallbackHandler.__init__=s<   #%>>D  %% &w 75d6J6J5KLJJ  ?ryy):;C MM1151E1E0FG>> **G ?ryy):;C--.x@#  MM1151E1E0FG>> **G  GGGWG 5),B0@0@0B JT))*U8_< '(<(<'=>.. -e4 :2--::&&--DL%j1  8K8KL8Ku 8KL L.t/@/@.AAS&&'(BBRASTAZ^ZfZfZmZmAnZmQV%**ZmAn@op??C}}oQ P .0  OO$$?A } Y  X !@D'(,&77RT    6 #RSTRUV==A=N=Nz6ArgillaCallbackHandler.on_chain_end.. s" L 4<< LsrErFNr7rVrPrQ r)anyrHr9getrritemsrcrdr5rUrSrWjoinrXr r'rY)r:rfrCr9chain_output_keychain_output_valroutputs` rA on_chain_endz#ArgillaCallbackHandler.on_chain_ends  F?34c&:J6KL    ((VO-D)EF $ $,,""3vh'7#8"=K 3:--/ . .*D1 ((/2';K.L/MNFF %*0,26N,@,@,B' /M)  ((%*-((7*;,<,B,B,D') !3B8 vo& '4<< 7 LL  S!89 : vh DLL 0 LL  S!12 3 %% &x 8 LL ( ( * 9;s"H c y)z+Do nothing when LLM chain outputs an error.Nr"r^s rAon_chain_errorz%ArgillaCallbackHandler.on_chain_error8rNrJ input_strc y)zDo nothing when tool starts.Nr")r:rBrwrCs rA on_tool_startz$ArgillaCallbackHandler.on_tool_start<s rJactionc y)z.Do nothing when agent takes a specific action.Nr")r:rzrCs rAon_agent_actionz&ArgillaCallbackHandler.on_agent_actionErNrJrsobservation_prefix llm_prefixc y)zDo nothing when tool ends.Nr")r:rsr}r~rCs rA on_tool_endz"ArgillaCallbackHandler.on_tool_endIs rJc y)z&Do nothing when tool outputs an error.Nr"r^s rA on_tool_errorz$ArgillaCallbackHandler.on_tool_errorSrNrJrVc yz Do nothingNr")r:rVrCs rAon_textzArgillaCallbackHandler.on_textWrNrJfinishc yrr")r:rrCs rAon_agent_finishz&ArgillaCallbackHandler.on_agent_finish[rNrJ)NNN)NN)!__name__ __module__ __qualname____doc__rrH__annotations__rrrrr$rrrrIrMr r[ BaseExceptionr_rertrvryrr|rrrr r __classcell__)r@s@rArr si)Vrs, 22:8,#R 0R rJ