Spaces:
Sleeping
Sleeping
update hybrid_retriever_tool
Browse files
tools/hybrid_retriever_tool.py
CHANGED
|
@@ -110,7 +110,7 @@ class HybridRetrieverTool(RagTool):
|
|
| 110 |
],
|
| 111 |
temperature=0
|
| 112 |
)
|
| 113 |
-
content = response.choices[0].message.
|
| 114 |
try:
|
| 115 |
ranked = json.loads(response.choices[0].message.content)
|
| 116 |
# Keep only valid strings
|
|
|
|
| 110 |
],
|
| 111 |
temperature=0
|
| 112 |
)
|
| 113 |
+
content = response.choices[0].message.content.strip()
|
| 114 |
try:
|
| 115 |
ranked = json.loads(response.choices[0].message.content)
|
| 116 |
# Keep only valid strings
|