HCidZddlZddlZddlZddlZddlmZmZmZm Z ddl m Z m Z ejdxsejdxsdZejdxsejd xsd Zejd xsejd xsd Zde e dfdedeedededee eeefdeef dZ ddededeedefdZy)z Grok API Integration (server-side). SECURITY: - Never hardcode API keys in this repository. - Provide the key via environment variables on the server (systemd override recommended). N)OptionalDictAnyList)MAX_NEW_TOKENS TEMPERATURE XAI_API_KEY GROK_API_KEY XAI_API_URL GROK_API_URLz$https://api.x.ai/v1/chat/completions XAI_MODEL GROK_MODELzgrok-4-1-fast-reasoningprompt system_prompt max_tokens temperatureconversation_historyreturncddtdd}g}|r|jd|d|r|j||jd|dt|||dd} t j}t j t||d } t j|z } | jd k(rB| j} d | vrt| d d kDr| d d ddStd| ytd| jd| jy#t jj$r(tdt jz ddYyt jj$r} td| Yd} ~ yd} ~ wt $r} td| Yd} ~ yd} ~ wwxYw)a Call Grok Fast API to generate a response Args: prompt: The user prompt/question system_prompt: Optional system prompt for context max_tokens: Maximum tokens to generate temperature: Sampling temperature conversation_history: Optional conversation history Returns: Generated response text or None if error zBearer zapplication/json) Authorizationz Content-Typesystem)rolecontentuserF)modelmessagesrrstream-)headersjsontimeoutchoicesrmessagerz"Grok API: No choices in response: NzGrok API Error: z - zGrok API Timeout after z.2fszGrok API Request Error: zGrok API Unexpected Error: )r appendextendrtimerequestspostr status_coder!lenprinttext exceptionsTimeoutRequestException Exception) rrrrrr rpayload start_timeresponsedurationdataes ,/var/www/html/leadgen/airagagent/grok_api.py call_grok_apir;"s*#<.1*G H$  ,- OO  " GYY[ ==    99;+   3 &==?DD Si%9A%=Iq))4Y??:4&AB $X%9%9$:#hmm_M N    & & ' j(@'EQGH    / / (,-  +A3/0s8B D#D2%DAF/F/8F  F/F**F/contextquestionsystem_instructionscP|sy|xsd}d|d|d}t||dd}|xsd S) a Generate an answer using Grok Fast API with RAG context Args: context: Formatted RAG context from vector search question: User's question system_instructions: Optional system instructions Returns: Generated answer zYI don't have enough information to answer that question based on the available documents.aYou are a helpful AI assistant that answers questions based on provided context. - Use only the information provided in the context - If the context doesn't contain relevant information, say so explicitly - Be concise and accurate - For technical questions (pH, temperature, measurements), extract specific numbers/ranges - Do NOT provide generic information if the specific answer isn't in the context - Cite sources when mentioned in the contextzFBased on the following context, answer the user's question. CONTEXT: z QUESTION: z ANSWER:i gffffff?)rrrrz?I encountered an error generating a response. Please try again.)r;)r<r=r>r user_promptanswers r:generate_answer_with_grokrBwsc j(4,4M  *  K# F  VVV)N)__doc__osr*r!r)typingrrrrconfigrrgetenvr r rstrintfloatr;rBrCr:rMsM  ,,. BIIm ryy    ryy'n9299^+DnHn BIIk!ryy!  $($$;? R RC=RR R #4S#X#78 R c] Rp*.-W -W-W"#-W -WrC