
    gW                     |    d dl Z d dlmZmZ d dlZd dlmZmZmZmZm	Z	 d dl
mZ d dlmZ  G d de      Z e       Zy)    N)AnyOptional)	CustomLLMImageObjectImageResponse
completionget_llm_provider)AsyncHTTPHandler)ModelResponsec                   8    e Zd ZdefdZdej                  fdZy)MyCustomLLMreturnc                 :    t        j                  ddddgd      S Nzgpt-3.5-turbouserzHello world)rolecontentzHi!)modelmessagesmock_responselitellmr   selfargskwargss      e/var/www/openai/venv/lib/python3.12/site-packages/litellm/proxy/example_config_yaml/custom_handler.pyr   zMyCustomLLM.completion   s&    !!!%-@A
 	
    c                 B   K   t        j                  ddddgd      S wr   r   r   s      r   acompletionzMyCustomLLM.acompletion   s*     !!!%-@A
 	
s   N)__name__
__module____qualname__r   r   r   r     r   r   r   r   
   s     
] 

G4I4I 
r   r   )timetypingr   r   r   r   r   r   r   r	   &litellm.llms.custom_httpx.http_handlerr
   litellm.types.utilsr   r   my_custom_llmr$   r   r   <module>r*      s/        W W C -
) 
  r   