Spaces:
Running
Running
Update agent.py
Browse files
agent.py
CHANGED
@@ -136,8 +136,12 @@ sys_msg = SystemMessage(content=system_prompt)
|
|
136 |
# -------------------------------
|
137 |
# Step 1: Load documents from CSV file (max 165 rows)
|
138 |
# -------------------------------
|
139 |
-
|
140 |
-
|
|
|
|
|
|
|
|
|
141 |
|
142 |
# Check if 'content' column exists
|
143 |
assert 'content' in df.columns, "'content' column is required in the CSV file."
|
|
|
136 |
# -------------------------------
|
137 |
# Step 1: Load documents from CSV file (max 165 rows)
|
138 |
# -------------------------------
|
139 |
+
|
140 |
+
# -------------------------------
|
141 |
+
# Step 1: Load documents from CSV URL (max 165 rows)
|
142 |
+
# -------------------------------
|
143 |
+
csv_url = "https://huggingface.co/spaces/wt002/Final_Assignment_Project/blob/main/documents.csv" # Replace with your actual URL
|
144 |
+
df = pd.read_csv(csv_url).head(165)
|
145 |
|
146 |
# Check if 'content' column exists
|
147 |
assert 'content' in df.columns, "'content' column is required in the CSV file."
|