Update nova_agent.py

nova_agent.py

@@ -34,19 +34,26 @@ class NovaLiteAgent:
         print(f"NovaLiteAgent received question (first 50 chars): {question[:50]}...")
         
         try:
+            # Create a more focused prompt for concise answers
+            prompt = f"""Answer this question directly and concisely. Provide only the essential information requested, not explanations or step-by-step reasoning unless specifically asked.
+
+Question: {question}
+
+Answer:"""
+            
             # Prepare the request payload for Nova Lite
             payload = {
                 "messages": [
                     {
                         "role": "user",
                         "content": [{
-                            "text": question
+                            "text": prompt
                         }]
                     }
                 ],
                 "inferenceConfig": {
-                    "max_new_tokens": 1000,
-                    "temperature": 0.1
+                    "max_new_tokens": 150,
+                    "temperature": 0.0
                 }
             }
             
@@ -62,8 +69,33 @@ class NovaLiteAgent:
             response_body = json.loads(response['body'].read())
             answer = response_body['output']['message']['content'][0]['text']
             
-            return answer.strip()
+            # Clean up the answer - remove verbose patterns
+            answer = answer.strip()
+            
+            # Remove common verbose beginnings
+            verbose_starts = [
+                "To answer this question",
+                "Based on the information",
+                "According to",
+                "The answer is",
+                "Looking at"
+            ]
+            
+            for start in verbose_starts:
+                if answer.lower().startswith(start.lower()):
+                    sentences = answer.split('. ')
+                    for sentence in sentences[1:]:
+                        if len(sentence.strip()) > 10:
+                            answer = sentence.strip()
+                            break
+            
+            # Limit length
+            if len(answer) > 200:
+                sentences = answer.split('. ')
+                answer = sentences[0] + '.'
+            
+            return answer
             
         except Exception as e:
             print(f"Error calling Nova Lite: {e}")
-            return f"I apologize, but I'm currently experiencing technical difficulties: {e}"
+            return "Unable to process request."