Daemontatox commited on
Commit
b89373e
·
verified ·
1 Parent(s): 9f2c070

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -3
app.py CHANGED
@@ -115,12 +115,25 @@ db = Qdrant(
115
  )
116
 
117
  # Create retriever
 
 
 
 
 
118
  retriever = db.as_retriever(
119
- search_type="similarity",
120
- search_kwargs={"k": 3}
121
  )
122
 
123
 
 
 
 
 
 
 
 
 
124
  # Load model directly
125
 
126
 
@@ -145,7 +158,6 @@ llm = ChatOpenAI(
145
  max_tokens=None,
146
  timeout=None,
147
  max_retries=6,
148
- streaming=True,
149
 
150
  )
151
 
 
115
  )
116
 
117
  # Create retriever
118
+ # retriever = db.as_retriever(
119
+ # search_type="similarity",
120
+ # search_kwargs={"k": 5}
121
+ # )
122
+
123
  retriever = db.as_retriever(
124
+ search_type="mmr",
125
+ search_kwargs={"k": 5, "fetch_k": 10, "lambda_mult": 0.5}
126
  )
127
 
128
 
129
+
130
+ # retriever = db.as_retriever(
131
+ # search_type="similarity_score_threshold",
132
+ # search_kwargs={"k": 5, "score_threshold": 0.8}
133
+ # )
134
+
135
+
136
+
137
  # Load model directly
138
 
139
 
 
158
  max_tokens=None,
159
  timeout=None,
160
  max_retries=6,
 
161
 
162
  )
163