
    i                         U d dl Z d dlmZmZ erd dlmZmZ d dlmZ d dlm	Z	 ddl
mZ dZee   ed<   	 	 	 	 	 	 d
d	Ze	j                   j"                  eiZy)    N)TYPE_CHECKINGOptional)PromptLiteLLMParams
PromptSpec)CustomPromptManagement)SupportedPromptIntegrations   )ArizePhoenixPromptManagerglobal_arize_configc           	      "   t        | dd      xs t        j                  j                  d      }t        | dd      }t        | dd      }|r|st	        d      	 t        d
i |||d| j                  h d	      }|S # t        $ r}|d}~ww xY w)z1
    Initialize a prompt from Arize Phoenix.
    api_keyNPHOENIX_API_KEYapi_base	prompt_idzFapi_key and api_base are required for Arize Phoenix prompt integration)r   r   r   >   r   r   r   )exclude )getattrosenvironget
ValueErrorr
   
model_dump	Exception)litellm_paramsprompt_specr   r   r   arize_prompt_manageres          t/Users/manta/Documents/Projects/TheRoad-I1/.venv/lib/python3.12/site-packages/litellm/integrations/arize/__init__.pyprompt_initializerr      s     ni6 "**..;G ~z48HT:I(T
 	
8 	 
"$& !++@ , 		 
 $# s   $A> >	BB		B)r   r   r   r   returnr   )r   typingr   r   "litellm.types.prompts.init_promptsr   r   -litellm.integrations.custom_prompt_managementr   r   arize_phoenix_prompt_managerr
   r   dict__annotations__r   ARIZE_PHOENIXvalueprompt_initializer_registryr       r   <module>r+      se    	 *RT J C '+ Xd^ *)8DF  --335G r*   