Update graph_builder.py

graph_builder.py

@@ -71,16 +71,22 @@ def extract_study_info(state: DocumentExtractionState) -> DocumentExtractionStat
         text_for_extraction = ""
         sections = state.get("sections", {})
         
-        # Try to find synopsis or summary section first
-        for section_name in ["synopsis", "summary", "overview"]:
-            if section_name.lower() in [s.lower() for s in sections.keys()]:
-                section_key = next(k for k in sections.keys() if k.lower() == section_name.lower())
-                text_for_extraction = sections[section_key]
-                break
-        
-        # If no synopsis found, use the beginning of the document
-        if not text_for_extraction and "document_text" in state:
-            text_for_extraction = state["document_text"][:20000]  # Use first 20k chars
+        # Check if sections is a list (section names only) or a dict (section name -> content)
+        if isinstance(sections, list):
+            # Just use the document text since we don't have section content
+            if "document_text" in state:
+                text_for_extraction = state["document_text"][:20000]  # Use first 20k chars
+        else:
+            # Try to find synopsis or summary section first
+            for section_name in ["synopsis", "summary", "overview"]:
+                if section_name.lower() in [s.lower() for s in sections.keys()]:
+                    section_key = next(k for k in sections.keys() if k.lower() == section_name.lower())
+                    text_for_extraction = sections[section_key]
+                    break
+            
+            # If no synopsis found, use the beginning of the document
+            if not text_for_extraction and "document_text" in state:
+                text_for_extraction = state["document_text"][:20000]  # Use first 20k chars
         
         if not text_for_extraction:
             return {