o Zh @sddlZddlmZmZmZmZmZmZmZddl Z ddl m Z ddl m Z ddlmZddlmZdZdZd ed ed ed eeeffd dZd ed ed ed eeddffddZGddde ZdS)N)AnyDict GeneratorIteratorListLiteralOptional)CallbackManagerForLLMRun)LLM)GenerationChunk)Fieldzhttps://chat-api.you.com/smartz!https://chat-api.you.com/researchbase_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 rrrresponserS/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/llms/you.py_requestsrc ksd|i}tdi|ddi}tj||d|d}|dd|D}z ddl}||}Wn ty:tdw|D]&} | jd vrGq?| jd krQ| j Vq?| jd kr^t d | j t d | jdS)NrstreamT)rrrcss|]}|VqdS)Nr).0 bytestringrrr "sz"_request_stream..rzPCould not import `sseclient`. Please install it with `pip install sseclient-py`.)Zsearch_resultsdonetokenerrorzError in response: zUnknown event type r) dictrrr sseclientZ SSEClient ImportErroreventseventdata ValueErrorNotImplementedError) r rrrparamsrZ event_sourcer#clientr&rrr_request_streams.      r,c @seZdZUdZedddZeded<edddZe e ed <  dd e d e e e d e e d e de f ddZ  dd e d e e e d e e d e deef ddZede fddZede fddZede fddZdS)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_keypromptstop run_managerrrcKs4|rtdd|i}t|jfd|ji|}|dS)N5Stop words are not implemented for You.com endpoints.queryrZanswer)r)r_request_endpoint_api_key)selfr2r3r4rr*rrrr_call`sz You._callcksD|rtdd|i}t|jfd|ji|D]}t|dVqdS)Nr5r6r)text)r)r,r7r8r )r9r2r3r4rr*r rrr_streamos z You._streamcCs|jdkrtStS)Nr.)r0SMART_ENDPOINTRESEARCH_ENDPOINTr9rrrr7s zYou._request_endpointcCs|jptjdS)NZ YDC_API_KEY)r1osenvironr?rrrr8sz You._api_keycCsdS)Nzyou.comrr?rrr _llm_typesz You._llm_type)NN)__name__ __module__ __qualname____doc__r r0r__annotations__r1rstrrr rr:rr r<propertyr7r8rBrrrrr-;sT     r-)r@typingrrrrrrrrZ langchain_core.callbacks.managerr Z#langchain_core.language_models.llmsr Zlangchain_core.outputsr Zpydanticr r=r>rHrr,r-rrrrs($    "   "