Update app.py
Browse files
app.py
CHANGED
@@ -115,12 +115,25 @@ db = Qdrant(
|
|
115 |
)
|
116 |
|
117 |
# Create retriever
|
|
|
|
|
|
|
|
|
|
|
118 |
retriever = db.as_retriever(
|
119 |
-
search_type="
|
120 |
-
search_kwargs={"k":
|
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 |
|