Spaces:
Sleeping
Sleeping
Update retriever.py
Browse files- retriever.py +0 -19
retriever.py
CHANGED
@@ -49,24 +49,5 @@ def load_guest_dataset():
|
|
49 |
# Return the tool
|
50 |
return GuestInfoRetrieverTool(docs)
|
51 |
|
52 |
-
def find_subject_conversation():
|
53 |
-
# Load the dataset
|
54 |
-
guest_dataset = datasets.load_dataset("agents-course/unit3-invitees", split="train")
|
55 |
-
|
56 |
-
# Convert dataset entries into Document objects
|
57 |
-
docs = [
|
58 |
-
Document(
|
59 |
-
page_content="\n".join([
|
60 |
-
f"Name: {guest['name']}",
|
61 |
-
f"Description: {guest['description']}",
|
62 |
-
]),
|
63 |
-
metadata={"name": guest["name"]}
|
64 |
-
)
|
65 |
-
for guest in guest_dataset
|
66 |
-
]
|
67 |
-
|
68 |
-
# Return the tool
|
69 |
-
return GuestInfoRetrieverTool(docs)
|
70 |
-
|
71 |
|
72 |
|
|
|
49 |
# Return the tool
|
50 |
return GuestInfoRetrieverTool(docs)
|
51 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
52 |
|
53 |
|