o
    ZhD                     @  s  d Z ddlmZ ddlZddlmZ ddlmZmZm	Z	m
Z
mZmZmZmZmZmZmZ ddlmZ ddlmZ ddlmZmZ dd	lmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z% dd
l&m'Z' ddl(m)Z)m*Z* ddl+m,Z,m-Z-m.Z. ddl/m0Z0m1Z1m2Z2 ddl3m4Z4m5Z5 ddl6m7Z7 ddl8m9Z9m:Z:m;Z;m<Z<m=Z= ddl>m?Z? ede9dZ@eeeAef ee@ ef ZBeee@f ZCeDeEZFd!ddZGd"ddZHG dd  d eZIdS )#zWrapper around Perplexity APIs.    )annotationsN)
itemgetter)AnyDictIteratorListLiteralMappingOptionalTupleTypeTypeVarUnion)CallbackManagerForLLMRun)LanguageModelInput)BaseChatModelgenerate_from_stream)	AIMessageAIMessageChunkBaseMessageBaseMessageChunkChatMessageChatMessageChunkFunctionMessageChunkHumanMessageHumanMessageChunkSystemMessageSystemMessageChunkToolMessageChunk)UsageMetadata)JsonOutputParserPydanticOutputParser)ChatGenerationChatGenerationChunk
ChatResult)RunnableRunnableMapRunnablePassthrough)from_envget_pydantic_field_names)is_basemodel_subclass)	BaseModel
ConfigDictFieldTypeAdaptermodel_validator)Self_BM)boundobjr   returnboolc                 C  s   t | tot| S N)
isinstancetyper*   )r3    r9   a/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/chat_models/perplexity.py_is_pydantic_class;   s   r;   token_usagedictr   c                 C  s6   |  dd}|  dd}|  d|| }t|||dS )NZprompt_tokensr   Zcompletion_tokenstotal_tokensinput_tokensoutput_tokensr>   )getr   )r<   r@   rA   r>   r9   r9   r:   _create_usage_metadata?   s   rC   c                   @  sp  e Zd ZU dZdZded< dZded< 	 dZd	ed
< 	 ee	dZ
ded< 	 eedddddZded< 	 edddZded< 	 dZded< 	 dZded< 	 dZded< 	 ed d!Zed\d$d%Zed&d'ed]d)d*Zed+d'd^d-d.Zed_d/d0Zd`d3d4Zdad:d;ZdbdAdBZ		dcdddGdHZ		dcdedJdKZedfdLdMZedgdNdOZ 	dhdPdddQdidZd[Z!dS )jChatPerplexitya@  `Perplexity AI` Chat models API.

    To use, you should have the ``openai`` python package installed, and the
    environment variable ``PPLX_API_KEY`` set to your API key.
    Any parameters that are valid to be passed to the openai.create call can be passed
    in, even if not explicitly saved on this class.

    Example:
        .. code-block:: python

            from langchain_community.chat_models import ChatPerplexity

            chat = ChatPerplexity(
                model="llama-3.1-sonar-small-128k-online",
                temperature=0.7,
            )
    Nr   clientz!llama-3.1-sonar-small-128k-onlinestrmodelgffffff?floattemperature)default_factoryDict[str, Any]model_kwargsPPLX_API_KEY)defaultapi_key)rJ   aliaszOptional[str]pplx_api_keytimeout)rP   z+Optional[Union[float, Tuple[float, float]]]request_timeout   intmax_retriesFr5   	streamingzOptional[int]
max_tokensT)Zpopulate_by_namer4   Dict[str, str]c                 C  s   ddiS )NrQ   rM   r9   selfr9   r9   r:   
lc_secretsx   s   zChatPerplexity.lc_secretsbefore)modevaluesc              
   C  s   t | }|di }t|D ](}||v rtd| d||vr6td| d| d| d ||||< q|| }|rHtd| d	||d< |S )
z>Build extra kwargs from additional params that were passed in.rL   zFound z supplied twice.z	WARNING! z1 is not a default parameter.
                    zJ was transferred to model_kwargs.
                    Please confirm that z is what you intended.zParameters za should be specified explicitly. Instead they were passed in as part of `model_kwargs` parameter.)	r)   rB   list
ValueErrorloggerwarningpopintersectionkeys)clsr_   Zall_required_field_namesextra
field_nameZinvalid_model_kwargsr9   r9   r:   build_extra|   s,   
zChatPerplexity.build_extraafterr0   c                 C  sT   zddl }W n ty   tdw z|j| jdd| _W | S  ty)   tdw )z?Validate that api key and python package exists in environment.r   NzTCould not import openai python package. Please install it with `pip install openai`.zhttps://api.perplexity.ai)rO   base_urlz`openai` has no `ChatCompletion` attribute, this is likely due to an old version of the openai package. Try upgrading it with `pip install --upgrade openai`.)openaiImportErrorZOpenAIrQ   rE   AttributeErrorra   )r[   rm   r9   r9   r:   validate_environment   s"   
	z#ChatPerplexity.validate_environmentc                 C  s   | j | j| jd| jS )z:Get the default parameters for calling PerplexityChat API.)rX   streamrI   )rX   rW   rI   rL   rZ   r9   r9   r:   _default_params   s   zChatPerplexity._default_paramsmessager   c                 C  sx   t |tr|j|jd}|S t |trd|jd}|S t |tr(d|jd}|S t |tr5d|jd}|S td| )N)rolecontentsystemuser	assistantzGot unknown type )r7   r   rt   ru   r   r   r   	TypeError)r[   rs   Zmessage_dictr9   r9   r:   _convert_message_to_dict   s   
	


z'ChatPerplexity._convert_message_to_dictmessagesList[BaseMessage]stopOptional[List[str]]+Tuple[List[Dict[str, Any]], Dict[str, Any]]c                   sD   t  j}|d urd|v rtd||d<  fdd|D }||fS )Nr}   z2`stop` found in both the input and default params.c                   s   g | ]}  |qS r9   )rz   ).0mrZ   r9   r:   
<listcomp>   s    z8ChatPerplexity._create_message_dicts.<locals>.<listcomp>)r=   _invocation_paramsra   )r[   r{   r}   paramsmessage_dictsr9   rZ   r:   _create_message_dicts   s   
z$ChatPerplexity._create_message_dicts_dictMapping[str, Any]default_classType[BaseMessageChunk]r   c                 C  s  | d}| dpd}i }| dr+t|d }d|v r'|d d u r'd|d< ||d< | dr6|d |d< |dks>|tkrCt|dS |d	ksK|tkrQt||d
S |dksY|tkr^t|dS |dksf|tkrnt||d dS |dksv|tkr~t||d dS |s|tkrt||dS ||dS )Nrt   ru    function_callnameZ
tool_callsrw   )ru   rx   )ru   additional_kwargsrv   function)ru   r   Ztooltool_call_id)ru   r   )ru   rt   )rB   r=   r   r   r   r   r   r   )r[   r   r   rt   ru   r   r   r9   r9   r:   _convert_delta_to_message_chunk   s.   





z.ChatPerplexity._convert_delta_to_message_chunkrun_manager"Optional[CallbackManagerForLLMRun]kwargsIterator[ChatGenerationChunk]c                 k  s   |  ||\}}i ||}t}|dd  |r||d< | jjjjd|dd|}d}	d }
|D ]}t|ts<| }|	d }rgt
|}|
rb|d |
d  |d |
d  |d |
d  d	}n|}|}
nd }t|d
 dkrrq1|d
 d }|	dg }| |d |}t|tr|r||_|	r| jd|iO  _d}	|	d}|d urt|dnd }|j}t||d}|r|j|j|d |V  q1d S )Nrq   Zstop_sequencesT)r{   rq   usager@   rA   r>   r?   choicesr   	citationsdeltaFfinish_reason)r   )rs   generation_info)chunkr9   )r   r   rd   rE   chatcompletionscreater7   r=   rB   rC   lenr   usage_metadatar   	__class__r#   Zon_llm_new_tokentext)r[   r{   r}   r   r   r   r   Zdefault_chunk_classZstream_respZfirst_chunkZprev_total_usager   Ztotal_usageZlc_total_usager   choicer   r   r   r9   r9   r:   _stream   sl   
	
zChatPerplexity._streamr$   c                 K  s   | j r| j|f||d|}|rt|S | ||\}}i ||}| jjjjd	d|i|}t|dd  }	r?t	|	
 }
nd }
t|jd jjd|ji|
d}tt|dgdS )
N)r}   r   r{   r   r   r   )ru   r   r   )rs   )Zgenerationsr9   )rW   r   r   r   rE   r   r   r   getattrrC   Z
model_dumpr   r   rs   ru   r   r$   r"   )r[   r{   r}   r   r   Zstream_iterr   r   responser   r   rs   r9   r9   r:   	_generate)  s,   zChatPerplexity._generatec                 C  s   d| j i}i || jS )z,Get the parameters used to invoke the model.rG   )rG   rr   )r[   Z
pplx_credsr9   r9   r:   r   E  s   z!ChatPerplexity._invocation_paramsc                 C  s   dS )zReturn type of chat model.Zperplexitychatr9   rZ   r9   r9   r:   	_llm_typeM  s   zChatPerplexity._llm_typejson_schema)methodinclude_rawstrictschemaOptional[_DictOrPydanticClass]r   Literal['json_schema']r   r   Optional[bool]-Runnable[LanguageModelInput, _DictOrPydantic]c                K  s  |dkrP|du rt dt|}|rt|dr| }n|r#| }nt|tr+|}nt|jdkr:t	|}|
 }| jdd|idd}	|rLt|d	nt }
nt d
| d|r~tjtd|
B dd d}tjdd d}|j|gdd}t|	d|B S |	|
B S )aP  Model wrapper that returns outputs formatted to match the given schema for Preplexity.
        Currently, Preplexity only supports "json_schema" method for structured output
        as per their official documentation: https://docs.perplexity.ai/guides/structured-outputs

        Args:
            schema:
                The output schema. Can be passed in as:

                - a JSON Schema,
                - a TypedDict class,
                - or a Pydantic class

            method: The method for steering model generation, currently only support:

                - "json_schema": Use the JSON Schema to parse the model output


            include_raw:
                If False then only the parsed structured output is returned. If
                an error occurs during model output parsing it will be raised. If True
                then both the raw model response (a BaseMessage) and the parsed model
                response will be returned. If an error occurs during output parsing it
                will be caught and returned as well. The final output is always a dict
                with keys "raw", "parsed", and "parsing_error".

            kwargs: Additional keyword args aren't supported.

        Returns:
            A Runnable that takes same inputs as a :class:`langchain_core.language_models.chat.BaseChatModel`.

            | If ``include_raw`` is False and ``schema`` is a Pydantic class, Runnable outputs an instance of ``schema`` (i.e., a Pydantic object). Otherwise, if ``include_raw`` is False then Runnable outputs a dict.

            | If ``include_raw`` is True, then Runnable outputs a dict with keys:

            - "raw": BaseMessage
            - "parsed": None if there was a parsing error, otherwise the type depends on the ``schema`` as described above.
            - "parsing_error": Optional[BaseException]

        r   NzIschema must be specified when method is not 'json_schema'. Received None.model_json_schema_TypedDictMetar   )r8   r   )response_format)Zpydantic_objectzSUnrecognized method argument. Expected 'json_schema' Received:                    ''rawc                 S     d S r6   r9   _r9   r9   r:   <lambda>      z7ChatPerplexity.with_structured_output.<locals>.<lambda>)parsedparsing_errorc                 S  r   r6   r9   r   r9   r9   r:   r     r   )r   r   )Zexception_key)r   )ra   r;   hasattrr   r   r7   r=   r8   __name__r.   r   bindr!   r    r'   Zassignr   Zwith_fallbacksr&   )r[   r   r   r   r   r   Zis_pydantic_schemar   adapterZllmZoutput_parserZparser_assignZparser_noneZparser_with_fallbackr9   r9   r:   with_structured_outputR  sR   0


z%ChatPerplexity.with_structured_output)r4   rY   )r_   rK   r4   r   )r4   r0   )r4   rK   )rs   r   r4   rK   )r{   r|   r}   r~   r4   r   )r   r   r   r   r4   r   )NN)
r{   r|   r}   r~   r   r   r   r   r4   r   )
r{   r|   r}   r~   r   r   r   r   r4   r$   )r4   r   )r4   rF   r6   )r   r   r   r   r   r5   r   r   r   r   r4   r   )"r   
__module____qualname____doc__rE   __annotations__rG   rI   r-   r=   rL   r(   rQ   rS   rV   rW   rX   r,   Zmodel_configpropertyr\   r/   classmethodrj   rp   rr   rz   r   r   r   r   r   r   r   r9   r9   r9   r:   rD   J   sl   
 
	

 @rD   )r3   r   r4   r5   )r<   r=   r4   r   )Jr   
__future__r   loggingoperatorr   typingr   r   r   r   r   r	   r
   r   r   r   r   Zlangchain_core.callbacksr   Zlangchain_core.language_modelsr   Z*langchain_core.language_models.chat_modelsr   r   Zlangchain_core.messagesr   r   r   r   r   r   r   r   r   r   r   r   Zlangchain_core.messages.air   Zlangchain_core.output_parsersr    r!   Zlangchain_core.outputsr"   r#   r$   Zlangchain_core.runnablesr%   r&   r'   Zlangchain_core.utilsr(   r)   Zlangchain_core.utils.pydanticr*   Zpydanticr+   r,   r-   r.   r/   Ztyping_extensionsr0   r1   rF   Z_DictOrPydanticClassZ_DictOrPydantic	getLoggerr   rb   r;   rC   rD   r9   r9   r9   r:   <module>   s0    48


