Spaces:
Runtime error
Runtime error
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 |
|