#iSrSSKrSSKrSSKrSSKJrJrJrJr SSK J r J r Sr Sr SrS\ \ S4S\S \\S \S \S \\\\\4S \\4 SjjrSS\S\S\\S \4Sjjrg)z_ Grok Fast API Integration Switches RAG system from local LLM to Grok Fast Reasoning Model API N)OptionalDictAnyList)MAX_NEW_TOKENS TEMPERATUREzTxai-bYupNvIKS2S5I41aDUOKj7XSzoIitscspt9HvhesX0wK58Bnggck8gPAbK1jXvgc7PzKZJntt8HIheqEz$https://api.x.ai/v1/chat/completionszgrok-4-fast-reasoningprompt system_prompt max_tokens temperatureconversation_historyreturncS[3SS.n/nU(aURSUS.5 U(aURU5 URSUS.5 [UUUSS.n[R"5n[ R "[UUS S 9n [R"5U- n U RS :XaEU R5n S U ;a [U S 5S :aU S S SS$[SU 35 g[SU RSU R35 g![ RRa) [S[R"5W- SS35 g[ RRan [SU 35 Sn A gSn A f[ an [SU 35 Sn A gSn A ff=f)aQ 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: ) GROK_API_KEYappendextend GROK_MODELtimerequestspost GROK_API_URL status_coderlenprinttext exceptionsTimeoutRequestException Exception) r r r r r rrpayload start_timeresponsedurationdataes 5/var/www/html/leadgen/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/0s8#B D&1D&%D&&AG,G F G)F<<Gcontextquestionsystem_instructionschU(dgU=(d SnSUSUS3n[UUSSS9nU=(d S $) z 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>hsc j(4,4M  *  K# F  VVV)N)__doc__r%rr$typingrrrrairagagent.configrrr r'r#strintfloatr7r>r?r6rGs  ,,9f 5 $ $($$;? R RC=RR R #4S#X#78 R c] Rp*.-W -W-W"#-W -Wr?