tPjUdZddlmZmZddlmZmZmZddlm Z ddl m Z m Z ddl mZddlmZeZeed<Gd d e ZGd d e Zej+d dgeegeefdededefdZy)zU USAGE AI CHAT ENDPOINTS /usage/ai/chat - Stream AI chat responses about usage data )FinalLiteral) APIRouterDependsRequest)StreamingResponse) BaseModelField)UserAPIKeyAuth)user_api_key_authrouterc(eZdZUeded<eed<y) ChatMessage)user assistantrolecontentN)__name__ __module__ __qualname__r__annotations__str/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/management_endpoints/usage_endpoints/endpoints.pyrrs % && LrrcVeZdZUeddZeeed<eddZe dzed<y) UsageAIChatRequest.z&Chat messages (user/assistant history)) descriptionmessagesNzModel to use for AI chat)defaultrmodel) rrrr rlistrrr!rrrrrrs/"'9a"bHd;bd8RSE3:Srrz/usage/ai/chatzBudget & Spend Tracking)tags dependenciesdatarequestuser_api_key_dictcKddlm}m}ddlm}||}|r |j }n||}|j Dcgc]}|j|jd} }t|| |j||dddd  Scc}ww) z AI chat about usage data. Streams SSE events with the AI response. The AI agent has access to tools that query aggregated daily activity data. r)_user_has_admin_view$require_caller_user_id_for_non_admin)stream_usage_ai_chat)rr)rr!user_idis_adminztext/event-streamzno-cacheno)z Cache-ControlzX-Accel-Buffering) media_typeheaders) /litellm.proxy.management_endpoints.common_utilsr)r*@litellm.proxy.management_endpoints.usage_endpoints.ai_usage_chatr+r,rrrrr!) r%r&r'r)r*r+r-r,mrs r usage_ai_chatr4s++<=H#++67HIGK}}U!199=UHU **  '",4H  Vsr=s "///%/B ) TT  # $+,- )00A(B! ! !&!  !r