YxihddlmZddlZddlZddlmZmZ d ddZy)) annotationsN)DictAnyc |d}||dd||dd} d} td|dzD]_} tj|| |} | j| j } | j dxsd j cS| #t$r:}|} td | d |d || |krtj|Yd}~d}~wwxYw) Nz /api/generateFjson) temperaturenum_ctx)modelpromptstreamformatoptions)rtimeoutresponseu#⚠️ Ollama call failed (attempt /z): ) rangerequestspostraise_for_statusrgetstrip Exceptionprinttimesleep)r r hostr rr max_retriesretry_wait_securlpayloadlast_errattemptrdataes //home/sandhiya/dr-transcripts/src/llm_ollama.pyollama_generater)sF- C'  GHK!O, + + cAA   668DHHZ(.B557 7 + N  +H 7y+cRSQTU V$ >*  +sAB C 0CC )zhttp://localhost:11434ig?i )r strr r,rr,r intrfloatrr-rr-r r-returnr,) __future__rrrtypingrrr)r(r4s"  )& & & & &  &  &&& &r3