uPj UdZddlZddlZddlZddlmZddlZdefdZdede fdZ dd iZ ee d <Gd d ejejZy) u& Mock httpx transport that returns valid OpenAI ChatCompletion responses. Activated via `litellm_settings: { network_mock: true }`. Intercepts at the httpx transport layer — the lowest point before bytes hit the wire — so the full proxy -> router -> OpenAI SDK -> httpx path is exercised. N)FinalreturncJdtjjddS)Nzchatcmpl-mock-)uuiduuid4hexm/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/custom_httpx/mock_transport.py_mock_idr s# DJJL,,Ra01 22r modelc rtdttj|ddddddgddd d d S) z-Return a minimal valid ChatCompletion object.zchat.completionr assistantz Mock response)rolecontentstop)indexmessage finish_reason) prompt_tokenscompletion_tokens total_tokens)idobjectcreatedrchoicesusage)r inttime)rs r _chat_completion_jsonr#sVj#tyy{#'."(    !"  r z content-typezapplication/json _JSON_HEADERSceZdZdZedej deee ffdZ dej dejfdZ dej dejfdZ y)MockOpenAITransportz httpx transport that returns canned OpenAI ChatCompletion responses. Supports both async (AsyncOpenAI) and sync (OpenAI) SDK paths. requestrc tj|j}|j dd}|dfS#tjtf$rYywxYw)z$Extract model from the request body.) mock-modelFrr)F)jsonloadsrJSONDecodeError ValueErrorget)r'bodyrs r _parse_requestz"MockOpenAITransport._parse_requestBsT )**W__5Dxx6u~$$j1 )( )s7AAcK|j|\}}tjt|j }t j dt|SwN) status_codeheadersrr0r*dumpsr#encodehttpxResponser$selfr'r_r/s r handle_async_requestz(MockOpenAITransport.handle_async_requestLsQ&&w/qjj!6u!=>EEG~~!  sAAc|j|\}}tjt|j }t j dt|Sr2r6r;s r handle_requestz"MockOpenAITransport.handle_requestUsM&&w/qjj!6u!=>EEG~~!  r N)__name__ __module__ __qualname____doc__ staticmethodr9Requesttuplestrboolr0r:r>r@r r r r&r&;sl  %T 2B %-- ENN  emm  r r&)rDr*r"rtypingrr9rHr dictr#r$__annotations__AsyncBaseTransport BaseTransportr&r r r rOsj  3#3<& u ! %22E4G4G! r