o ưi @sdZddlmZmZmZmZmZmZddlZddl m Z ddl m Z ddl mZddlmZdd lmZer?dd lmZeZneZGd d d eZdS) z, CompactifAI chat completion transformation ) TYPE_CHECKINGAnyListOptionalTupleUnionNget_secret_str) ModelResponse OpenAIError)BaseLLMException)OpenAIGPTConfig)Loggingc@seZdZdZdeedeedeeeeeffddZ  dded ej d e d e d e d e de de dedeedeede fddZdededee ejfdefddZdS)CompactifAIChatConfigz Configuration class for CompactifAI chat completions. Since CompactifAI is OpenAI-compatible, we extend OpenAIGPTConfig. api_baseapi_keyreturncCs |pd}|p tdp d}||fS)z@ Get API base and key for CompactifAI provider. zhttps://api.compactif.ai/v1ZCOMPACTIFAI_API_KEYr)selfrrZdynamic_api_keyrc/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/compactifai/chat/transformation.py$_get_openai_compatible_provider_infosz:CompactifAIChatConfig._get_openai_compatible_provider_infoNmodel raw_responsemodel_response logging_obj request_datamessagesoptional_paramslitellm_paramsencoding json_modec Cs|j|| |jd|id|} | rA| dD]*} | d}|r@|dr@|dg}t|dkr@|ddd d |d <d |d<qtdi| }d ||_|S)z Transform CompactifAI response to LiteLLM format. Since CompactifAI is OpenAI-compatible, we can use the standard OpenAI transformation. Zcomplete_input_dict)inputroriginal_responseZadditional_argschoicesmessage tool_callsrfunction argumentsrcontentNz compactifai/r)Z post_calltextjsongetlenr r)rrrrrrrr r!r"rr#Z response_jsonchoicer'r(Zreturned_responserrrtransform_response*s&     z(CompactifAIChatConfig.transform_response error_message status_codeheaderscCst|||dS)z Get the appropriate error class for CompactifAI errors. Since CompactifAI is OpenAI-compatible, we use OpenAI error handling. )r4r'r5r )rr3r4r5rrrget_error_classYs z%CompactifAIChatConfig.get_error_class)NN)__name__ __module__ __qualname____doc__rstrrrhttpxResponser LiteLLMLoggingObjdictrrboolr2intrZHeadersr r6rrrrrsZ      / r)r:typingrrrrrrr<Zlitellm.secret_managers.mainr Zlitellm.types.utilsr Z litellm.llms.openai.common_utilsr Z)litellm.llms.base_llm.chat.transformationr Zopenai.chat.gpt_transformationrZ*litellm.litellm_core_utils.litellm_loggingrZ_LiteLLMLoggingObjr>rrrrrs