Merge remote-tracking branch 'origin/feat/llm_responses' into feat/ll…

…m_responses
deeppavlov · Nov 13, 2024 · ef9baa3 · ef9baa3
2 parents 23d6a31 + 2d4998c
commit ef9baa3
Showing 1 changed file with 5 additions and 11 deletions.
diff --git a/chatsky/slots/llm.py b/chatsky/slots/llm.py
@@ -61,17 +61,11 @@ async def get_value(self, ctx: Context) -> ExtractedGroupSlot:
         result_json = result.model_dump()
         logger.debug(f"Result JSON: {result_json}")
 
-        if self.allow_partially_extracted:
-            res = {
-                name: ExtractedValueSlot.model_construct(is_slot_extracted=True, extracted_value=result_json[name])
-                for name in result_json
-                if result_json[name] is not None
-            }
-        else:
-            res = {
-                name: ExtractedValueSlot.model_construct(is_slot_extracted=True, extracted_value=result_json[name])
-                for name in result_json
-            }
+        res = {
+            name: ExtractedValueSlot.model_construct(is_slot_extracted=True, extracted_value=result_json[name])
+            for name in result_json
+            if result_json[name] is not None or not self.allow_partial_extraction
+        }
         return ExtractedGroupSlot(**res)
 
     def __flatten_llm_group_slot(self, slot, parent_key=""):