Update retriever.py
Browse files- retriever.py +1 -1
retriever.py
CHANGED
|
@@ -30,7 +30,7 @@ class GuestInfoRetrieverTool(Tool):
|
|
| 30 |
|
| 31 |
def load_guest_dataset():
|
| 32 |
# Load the dataset
|
| 33 |
-
guest_dataset = datasets.load_dataset("
|
| 34 |
|
| 35 |
# Convert dataset entries into Document objects
|
| 36 |
docs = [
|
|
|
|
| 30 |
|
| 31 |
def load_guest_dataset():
|
| 32 |
# Load the dataset
|
| 33 |
+
guest_dataset = datasets.load_dataset("agents-course/unit3-invitees", split="train")
|
| 34 |
|
| 35 |
# Convert dataset entries into Document objects
|
| 36 |
docs = [
|