Update agent.py

agent.py

@@ -156,9 +156,13 @@ if response.status_code != 200:
     raise Exception(f"Failed to load JSONL from {jsonl_url}. Status code: {response.status_code}")
 
 
+# Ensure the request was successful
+if response.status_code != 200:
+    raise Exception(f"Failed to load JSONL from {jsonl_url}. Status code: {response.status_code}")
+
 # Read and parse the JSONL file line by line
 docs = []
-for line in response.text.splitlines():
+for line_number, line in enumerate(response.text.splitlines(), 1):
     try:
         doc = json.loads(line)  # Parse each line as a separate JSON object
         content = doc.get('content', "").strip()
@@ -171,9 +175,9 @@ for line in response.text.splitlines():
         # Convert the document into a Document object
         docs.append(Document(page_content=content, metadata=doc))
 
-    except json.JSONDecodeError:
-        print("Skipping malformed JSONL line.")
-
+    except json.JSONDecodeError as e:
+        print(f"Skipping malformed JSONL line at line {line_number}: {line}")
+        print(f"Error: {e}")
 
 
 # -------------------------------