Ii3ddlZddlmZddlmZddlmZddlZddlZddl m Z ddl m Z eje Z Gd d ZeZy) N) HTTPAdapter)Retry)Config)gemini_service)chatgpt_servicec0eZdZdZdZddZedZy) ChatServicecLt|_|j|_y)N)rconfig_create_sessionsession)selfs 0/srv/www/ollama/chat-ollama/src/services/chat.py__init__zChatService.__init__ sh ++- cLtj}t|jj|jj |jj |jj}t|}|jd||jd||S)N)totalbackoff_factorstatus_forcelistallowed_methods) max_retrieszhttp://zhttps://) requestsSessionrr RETRY_TOTALRETRY_BACKOFF_FACTORRETRY_STATUS_FORCELISTRETRY_ALLOWED_METHODSrmount)rrretry_strategyadapters rrzChatService._create_sessions{""$++));;;;![[?? KK==   .9 i) j'*rNc |rD|jdr3tjd|tj||ddd|S|rU|jdsd|j vr2tjd |t j||ddd S|jj|jj}|xs|jjd |xsd d d|d g|xsg||jj|jj|jjdd}tjd|tjd||jj!|||jj"|}|j%|ryd} |j'D]1} | s t)j*| } d| vrd| dvr | | ddz } 3tjdtjd| | gddS|j)} tjdtjd| d| vsd| dvr t/d| dd| j1dgddS#t(j,$rYwxYw#t2j4j6$r!tj9dt/dt2j4j:$r!tj9d t/d!t2j4j<$r=} tj9d"t?| t/d#t?| d} ~ wt.$r=} tj9d$t?| t/d%t?| d} ~ wwxYw)&NgeminizUsing Gemini model: gffffff?ig?)message system_prompt temperature max_tokenstop_pmodelgptchatgptzUsing ChatGPT model: )r%r&r'r(r)systemzYou are a helpful AI assistant.)rolecontentuser)r'r)top_k)r*messagescontextstreamoptionszSending request to Ollama API: zRequest payload: )jsontimeoutr4r%r/z8Successfully received streaming response from Ollama APIz Response: T)responser3donez.Successfully received response from Ollama APIz'Invalid response format from Ollama APIr3zRequest to API timed outz(The request timed out. Please try again.zFailed to connect to APIzBCould not connect to the AI service. Please check your connection.zError in API request: z)Error communicating with the AI service: zUnexpected error: zAn unexpected error occurred: ) startswithloggerinforgenerate_responselowerr r OLLAMA_URL CHAT_ENDPOINT OLLAMA_MODELOLLAMA_TEMPERATURE OLLAMA_TOP_P OLLAMA_TOP_KdebugrpostOLLAMA_TIMEOUTraise_for_status iter_linesr6loadsJSONDecodeError Exceptiongetr exceptionsTimeouterrorConnectionErrorRequestExceptionstr)rr%r*r&r3r4urlpayloadr9 full_responselinechunkresultes r send_messagezChatService.send_messages` G))(3 25':;%77#"/ ## %**51Y%++-5O 3E7;<&88#"/ ## [[++,T[[-F-F,GHC:$++":":%-2dCde#8#=b #';;#A#A![[55![[55 G KK9#? @ LL,WI6 7||(( 22 )H  % % ' " $//1%D%$(JJt$4E(E1i5CS6S -y1A)1L L % VW z-9:!.!  " LM z&23F*ivi?P.P#$MNN!'y 1) <%zz)R8 ' $33%$%2""** H LL3 4FG G""22 b LL3 4`a a""33 R LL1#a&: ;GAxPQ Q G LL-c!fX6 7rusG)$ *,   8 $zdzdx} r