o y i,@stdZddlZddlZddlZddlmZmZmZddlZddl m Z ddl m Z ddl mZGdddZeZdS) zo Novalite Bedrock Service for Amazon Bedrock Integration Handles text generation using Amazon's Novalite model N)OptionalDictAny) ClientError)Config)Logc @seZdZdZddZddedededeefd d Zddedededefd d Z d edeefddZ de fddZ de eeffddZdS)NovaliteBedrockServicez Service for generating text responses using Amazon Bedrock's Novalite model. Provides cost-effective alternative to OpenAI GPT models. c Cstj|_tj|_tj|_tj|_zt j d|jtj tj d|_ td|jd|jWdStyK}ztd|d|_ WYd}~dSd}~ww)Nzbedrock-runtime)Z region_nameZaws_access_key_idZaws_secret_access_keyu3🤖 Novalite Bedrock Service initialized - Model: z , Region: u)❌ Failed to initialize Bedrock client: )rZNOVALITE_MODEL_IDmodel_id AWS_REGIONregionZNOVALITE_MAX_TOKENS max_tokensZNOVALITE_TEMPERATURE temperatureboto3ZclientAWS_ACCESS_KEY_IDAWS_SECRET_ACCESS_KEYbedrock_clientrinfo Exceptionerror)selfer$services/novalite_bedrock_service.py__init__s" zNovaliteBedrockService.__init__N user_input system_promptconversation_historyreturnc sz6td|ddd||||}||IdH}|r0td|ddd|WStdWdStyR}ztd|WYd}~dSd}~ww) aK Generate a text response using Novalite model. Args: user_input: User's input text system_prompt: System prompt for the bot conversation_history: Previous conversation messages Returns: Generated response text or None if failed u.🤖 Generating Nova Lite response for input: N2...u"✅ Nova Lite response generated: du%⚠️ No response from Nova Lite APIu)❌ Error generating Nova Lite response: )rr_build_messages_call_bedrock_apiZwarningrr)rrrrmessagesZ response_textrrrrgenerate_response*s  z(NovaliteBedrockService.generate_responsec Csg}t|dr |js\tdtdt|dtd|dddd |vr1td nd |vr;td ntd tdt|dtd|dddd|_n tdt|dd|d}|dd|igd|r|ddD]H}t|trd|vr||dd|digdqd|vr|ddkrdnd}||d|digdq|ddt| ddigdq|dd|igd|S)a Build messages array for Novalite API. Args: user_input: Current user input system_prompt: System prompt conversation_history: Previous conversation Returns: Messages array for API _session_initializedu*🧠 Nova Lite System Prompt Source Check:u🧠 System Prompt Length: z charactersu.🧠 System Prompt Preview (first 200 chars): NrzKNOWLEDGE BASE CONTEXTuO🧠 ✅ System prompt includes KNOWLEDGE BASE CONTEXT - from bot configurationz.MCube's professional telephony voice assistantua🧠 ✅ System prompt includes 'MCube's professional telephony voice assistant' - from config.pyuY🧠 ❓ System prompt source unclear - neither bot config nor config.py pattern detectedu,🧠 Final Nova Lite System Message Length: u?🧠 Final Nova Lite System Message Preview (first 300 chars): i,Tu8🧠 Using persistent Nova Lite session (prompt length: z chars)z#[SYSTEM INSTRUCTIONS - MANDATORY] a5 [END OF SYSTEM INSTRUCTIONS] IMPORTANT: You MUST follow ALL instructions above, including: - Collecting user details (name, phone, email) when appropriate - Following behavioral rules - Using correct tone and language - Handling transfers with confirmation Now, please respond to the user's request below.usertextrolecontentir*r+typeZ assistant) hasattrr%rrlenappend isinstancedictstrget)rrrrr#Zfinal_system_messagemsgr*rrrr!JsZ          z&NovaliteBedrockService._build_messagesr#c s |js tddSzltdt|d|jjdddtd|id}t|d  }d |vrmd |d vrm|d d }d |vrmt |d t rm|d D]}d |vrl|d }td|ddd| WSqNtd|WdSt y}zc|jdd}|jdd} td|d| |dkrtdtdtjrdndtdtjrdndtdtjtd|jtdtd |jWYd}~dSd}~wty}ztd!|d"dl} td#| WYd}~dSd}~ww)$z Make API call to Amazon Bedrock. Args: messages: Messages array for Nova Lite API Returns: Generated text or None u9❌ Bedrock client not initialized - cannot make API callNu&🧠 Sending request to Nova API with z messageszamazon.nova-lite-v1:0zapplication/jsonr#)ZmodelIdZ contentTypeZacceptbodyr6outputmessager+r(u🧠 Nova Lite generated: r ru)❌ Unexpected Nova API response format: ErrorZCodeZMessageu❌ Bedrock API error (z): ZUnrecognizedClientExceptionu'🔐 AWS Authentication failed - Check:z 1. AWS_ACCESS_KEY_ID: u✅ Setu ❌ Missingz 2. AWS_SECRET_ACCESS_KEY: z 3. AWS_REGION: z 4. Model ID: u@ → Verify credentials are valid and have Bedrock permissionsu2 → Check if Nova Lite is available in region: u*❌ Unexpected error calling Bedrock API: ru❌ Traceback: )rrrdebugr/Z invoke_modeljsondumpsloadsreadr1liststriprresponserrrr r r r traceback format_exc) rr#rAZ response_bodyr8r+Zgenerated_textrZ error_codeZ error_messagerBrrrr"s^      z(NovaliteBedrockService._call_bedrock_apic sz1|js tdWdSdddddddgi}||IdH}|r+td Wd Std WdStyM}ztd |WYd}~dSd}~ww) z Test connection to Amazon Bedrock. Returns: True if connection successful, False otherwise u"❌ Bedrock client not initializedFr#r'z$System: You are a helpful assistant.r)ZHelloNu/✅ Novalite Bedrock connection test successfulTu+❌ Novalite Bedrock connection test failedu/❌ Error testing Novalite Bedrock connection: )rrrr"rr)rZ test_requestrArrrrtest_connections*   z&NovaliteBedrockService.test_connectioncCs|j|j|j|jddS)z Get information about the current model configuration. Returns: Dictionary with model information Znovalite_bedrock)r r r r Z service_type)r r r r )rrrrget_model_info s z%NovaliteBedrockService.get_model_info)N)__name__ __module__ __qualname____doc__rr3r?rr$r!r"boolrDrrrErrrrrs  ]A!r)rIr;ZasyncioZaiohttptypingrrrrZbotocore.exceptionsrZconfigrZservices.log_utilsrrZnovalite_bedrock_servicerrrrs