o Ujt @sfdZddlmZddlZddlZddlmZejde ee  j ddddl mZmZmZmZddlmZdd lmZmZdd lmZdd lmZed d dZee  jdZejdZ ee Z!ee!ddZ"da#ddZ$GdddeZ%GdddeZ&edd-ddZ'edd.d d!Z(ej)d"e&d#d/d%d&Z*e)d'ed(fd0d+d,Z+dS)1aFastAPI service exposing the travel-insurance agent. uvicorn app.server:app --host 0.0.0.0 --port 8000 The LLM is called lazily, so the API + RAG + cache layer boots even if the vLLM server isn't up yet (useful for testing retrieval/cache in isolation). ) annotationsN)Pathsrc)FastAPIFile HTTPException UploadFile) FileResponse) BaseModelField)Agent) load_configzTravel-Insurance Agentz0.1.0)titleversionZstaticZ AGENT_CONFIGT)Zlazy_llmcCs tdurddlm}|tatS)Nr Transcriber) _transcriberZ agent.sttr_cfgrrPC:\Users\cheed\OneDrive\Desktop\Claude work\travel-insurance-agent\app\server.py_get_transcriber"s rc@s$eZdZUeddddZded<dS) ChatRequest.rz$The user's travel-insurance question)Z min_length descriptionstrqueryN)__name__ __module__ __qualname__r r__annotations__rrrrr+s rc@sveZdZUded<ded<ded<ded<dZded <d Zd ed <d Zded<d Zded<d Zded<dZ ded<d S) ChatResponseranswerz list[str]Z citationsboolcachedZgroundedguardrail_reasonNz float | None similarityrint prompt_tokenscompletion_tokens total_tokensZllmroute) rrrrr%r&r(r)r*r+rrrrr /s      r /returnr cCs ttdS)zNServe the voice web UI (must be loaded via http://localhost so the mic works).z index.htmlN)r STATIC_DIRrrrrindex<s r/z/healthdictcCsddiS)NstatusokrrrrrhealthBsr3z/chat)Zresponse_modelreqcCs t|j}tdit|S)Nr)_agentr!rr Zto_dict)r4ZresprrrchatGs r6z /transcribe.audior c srzt}Wn tytddw|IdH}z ||}Wd|iSty8}ztdd|d}~ww)zXTranscribe uploaded mic audio locally (faster-whisper). Audio never leaves this machine.iz7Local STT not installed. Run: pip install -e '.[voice]'Niztranscription failed: text)r ImportErrorrread transcribe Exception)r7tdatar8errrr;Ms    r;)r-r )r-r0)r4rr-r )r7r r-r0),__doc__ __future__rossyspathlibrpathinsertr__file__resolveparentsZfastapirrrr Zfastapi.responsesr Zpydanticr r Zagent.pipeliner Zagent.settingsrappparentr.environgetZ _config_pathrr5rrrr r/r3postr6r;rrrrs8  $