U h@sddlZddlmZmZmZmZmZmZmZddl Z ddl m Z ddl m Z ddlmZddlmZdZdZeeeeeefd d d Zeeeeeddfd d d ZGddde ZdS)N)AnyDict GeneratorIteratorListLiteralOptional)CallbackManagerForLLMRun)LLM)GenerationChunk)Fieldzhttps://chat-api.you.com/smartz!https://chat-api.you.com/research)base_urlapi_keykwargsreturncKs(d|i}tj|||d}||S)z NOTE: This function can be replaced by a OpenAPI-generated Python SDK in the future, for better input/output typing support. x-api-key)headersjson)requestspostraise_for_statusr)r rrrresponser@/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/llms/you.py_requestsrc ksd|i}tf|ddi}tj||d|d}|dd|D}zddl}||}Wntk rttdYnX|D]P} | jd krq~| jd kr| j Vq~| jd krt d | j q~t d | jq~dS)NrstreamT)rrrcss|] }|VqdS)Nr).0Z bytestringrrr "sz"_request_stream..rzPCould not import `sseclient`. Please install it with `pip install sseclient-py`.)Zsearch_resultsdonetokenerrorzError in response: zUnknown event type ) dictrrr sseclientZ SSEClient ImportErroreventseventdata ValueErrorNotImplementedError) r rrrparamsrZ event_sourcer"clientr%rrr_request_streams(      r+c@seZdZUdZedddZeded<edddZe e ed <de e e e e e e e d d d Zde e e e e e e eed d dZee dddZee dddZee dddZdS)YouaWrapper around You.com's conversational Smart and Research APIs. Each API endpoint is designed to generate conversational responses to a variety of query types, including inline citations and web results when relevant. Smart Endpoint: - Quick, reliable answers for a variety of questions - Cites the entire web page URL Research Endpoint: - In-depth answers with extensive citations for a variety of questions - Cites the specific web page snippet relevant to the claim To connect to the You.com api requires an API key which you can get at https://api.you.com. For more information, check out the documentations at https://documentation.you.com/api-reference/. Args: endpoint: You.com conversational endpoints. Choose from "smart" or "research" ydc_api_key: You.com API key, if `YDC_API_KEY` is not set in the environment smartzCYou.com conversational endpoints. Choose from "smart" or "research") description)r-ZresearchendpointNz>You.com API key, if `YDC_API_KEY` is not set in the envrioment ydc_api_key)promptstop run_managerrrcKs4|r tdd|i}t|jfd|ji|}|dS)N5Stop words are not implemented for You.com endpoints.queryrZanswer)r(r_request_endpoint_api_key)selfr1r2r3rr)rrrr_call`sz You._callcksB|r tdd|i}t|jfd|ji|D]}t|dVq,dS)Nr4r5r)text)r(r+r6r7r )r8r1r2r3rr)rrrr_streamos z You._stream)rcCs|jdkrtStS)Nr-)r/SMART_ENDPOINTRESEARCH_ENDPOINTr8rrrr6s zYou._request_endpointcCs|jptjdS)NZ YDC_API_KEY)r0osenvironr>rrrr7sz You._api_keycCsdS)Nzyou.comrr>rrr _llm_typesz You._llm_type)NN)NN)__name__ __module__ __qualname____doc__r r/r__annotations__r0rstrrr rr9rr r;propertyr6r7rArrrrr,;sB     r,)r?typingrrrrrrrrZ langchain_core.callbacks.managerr Z#langchain_core.language_models.llmsr Zlangchain_core.outputsr Zlangchain_core.pydantic_v1r r<r=rGrr+r,rrrrs$      "