chore(deps): update invoke llm depending on the model used

Author: krokoko

samples/contract-compliance-analysis/back-end/stack/sfn/common-layer/llm.py

@@ -60,9 +60,19 @@ def invoke_llm(prompt, model_id, temperature=0.5, top_k=None, top_p=0.8, max_new
     response = chain.invoke({})
     content = response.content
 
+    usage_data = None
+    stop_reason = None
+
+    if ('anthropic' in model_id):
+        usage_data = response.response_metadata['usage']
+        stop_reason = response.response_metadata['stop_reason'] 
+    elif('amazon.nova' in model_id):
+        usage_data = response.usage_metadata 
+        stop_reason = response.response_metadata['stopReason']
+        
     if verbose:
         logger.info(f"Model response: {content}")
-        logger.info(f"Model usage: {response.response_metadata['usage']}")
-        logger.info(f"Model stop_reason: {response.response_metadata['stop_reason']}")
+        logger.info(f"Model usage: {usage_data}")
+        logger.info(f"Model stop_reason: {stop_reason}")
 
-    return content, response.response_metadata['usage'], response.response_metadata["stop_reason"]
+    return content, usage_data, stop_reason