Update alz_companion/agent.py

alz_companion/agent.py

@@ -136,12 +136,26 @@ def detect_tags_from_query(query: str, behavior_options: list, emotion_options:
         if start_brace != -1 and end_brace != -1 and end_brace > start_brace:
             json_str = response_str[start_brace : end_brace + 1]
             result = json.loads(json_str)
-            
+
+
             # Safely process the results from the LLM
-            result_dict["detected_behaviors"] = [b for b in result.get("detected_behaviors", []) if b in behavior_options]
-            result_dict["detected_emotion"] = result.get("detected_emotion") if result.get("detected_emotion") in emotion_options else "None"
-            result_dict["detected_topic"] = result.get("detected_topic") if result.get("detected_topic") in topic_options else "None"
-            result_dict["detected_contexts"] = [c for c in result.get("detected_contexts", []) if c in context_options]
+            behaviors = result.get("detected_behaviors")
+            result_dict["detected_behaviors"] = [b for b in behaviors if b in behavior_options] if behaviors else []
+
+            emotion = result.get("detected_emotion")
+            result_dict["detected_emotion"] = emotion if emotion in emotion_options else "None"
+
+            topic = result.get("detected_topic")
+            result_dict["detected_topic"] = topic if topic in topic_options else "None"
+            
+            contexts = result.get("detected_contexts") # This could be None
+            result_dict["detected_contexts"] = [c for c in contexts if c in context_options] if contexts else [] # Check if contexts is not None
+
+            # old code
+            # result_dict["detected_behaviors"] = [b for b in result.get("detected_behaviors", []) if b in behavior_options]
+            # result_dict["detected_emotion"] = result.get("detected_emotion") if result.get("detected_emotion") in emotion_options else "None"
+            # result_dict["detected_topic"] = result.get("detected_topic") if result.get("detected_topic") in topic_options else "None"
+            # result_dict["detected_contexts"] = [c for c in result.get("detected_contexts", []) if c in context_options]
         # --- END OF ROBUST LOGIC ---
         
         return result_dict