a bg@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 rrrresponserj/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/llms/you.py_requestsrc ksd|i}tfi|ddi}tj||d|d}|dd|D}zddl}||}WntyvtdYn0|D]P} | jd vrq| jd kr| j Vq| jd krt d | j qt d | jqdS)NrstreamT)rrrcss|] }|VqdS)Nr).0 bytestringrrr "z"_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)selfr3r4r5rr+rrrr_call`sz You._callcksB|r tdd|i}t|jfd|ji|D]}t|dVq,dS)Nr6r7r)text)r*r-r8r9r )r:r3r4r5rr+r!rrr_streamos z You._stream)rcCs|jdkrtStS)Nr/)r1SMART_ENDPOINTRESEARCH_ENDPOINTr:rrrr8s zYou._request_endpointcCs|jptjdS)NZ YDC_API_KEY)r2osenvironr@rrrr9sz You._api_keycCsdS)Nzyou.comrr@rrr _llm_typesz You._llm_type)NN)NN)__name__ __module__ __qualname____doc__r r1r__annotations__r2rstrrr rr;rr r=propertyr8r9rCrrrrr.;sB     r.)rAtypingrrrrrrrrZ langchain_core.callbacks.managerr Z#langchain_core.language_models.llmsr Zlangchain_core.outputsr Zpydanticr r>r?rIrr-r.rrrrs$      "