CindyDelage commited on
Commit
47120b7
·
verified ·
1 Parent(s): 347faa8

Update retriever.py

Browse files
Files changed (1) hide show
  1. 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