*idZddlZddlZddlZddlZddlmZmZmZm Z ddl m Z m Z ejjdZeseddZdZde e dfd ed eed ed ed ee eeefdeef dZ ddededeedefdZy)z_ Grok Fast API Integration Switches RAG system from local LLM to Grok Fast Reasoning Model API N)OptionalDictAnyList)MAX_NEW_TOKENS TEMPERATURE GROK_API_KEYz,GROK_API_KEY environment variable is not setz$https://api.x.ai/v1/chat/completionszgrok-4-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)modelmessagesr r stream-)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 appendextend GROK_MODELtimerequestspost GROK_API_URL status_coderlenprinttext exceptionsTimeoutRequestException Exception) r r r r rrrpayload start_timeresponsedurationdataes -/var/www/html/backtest/airagagent/grok_api.py call_grok_apir7s*#<.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?)r r r r z?I encountered an error generating a response. Please try again.)r7)r8r9r:r user_promptanswers r6generate_answer_with_grokr>ksc j(4,4M  *  K# F  VVV)N)__doc__r%rr$ostypingrrrrairagagent.configrrenvirongetr ValueErrorr'r#strintfloatr7r>r?r6rKs  ,,9zz~~n-  C DD5 $ $($$;? R RC=RR R #4S#X#78 R c] Rp*.-W -W-W"#-W -Wr?