Spaces:
Sleeping
Sleeping
Fix type annotation in RetrieverWithSource class
Browse files- lawchatbot/retrievers.py +2 -1
lawchatbot/retrievers.py
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
from langchain_core.retrievers import BaseRetriever
|
| 2 |
from langchain.retrievers.bm25 import BM25Retriever
|
| 3 |
from langchain_core.documents import Document
|
|
|
|
| 4 |
from langchain.retrievers import EnsembleRetriever
|
| 5 |
|
| 6 |
def initialize_semantic_retriever(vectorstore, config) -> BaseRetriever:
|
|
@@ -63,7 +64,7 @@ def wrap_retriever_with_source(retriever):
|
|
| 63 |
Wraps a retriever so every returned document has a 'source' key in metadata.
|
| 64 |
"""
|
| 65 |
class RetrieverWithSource(BaseRetriever):
|
| 66 |
-
base_retriever:
|
| 67 |
|
| 68 |
def __init__(self, base_retriever):
|
| 69 |
super().__init__()
|
|
|
|
| 1 |
from langchain_core.retrievers import BaseRetriever
|
| 2 |
from langchain.retrievers.bm25 import BM25Retriever
|
| 3 |
from langchain_core.documents import Document
|
| 4 |
+
from typing import Any
|
| 5 |
from langchain.retrievers import EnsembleRetriever
|
| 6 |
|
| 7 |
def initialize_semantic_retriever(vectorstore, config) -> BaseRetriever:
|
|
|
|
| 64 |
Wraps a retriever so every returned document has a 'source' key in metadata.
|
| 65 |
"""
|
| 66 |
class RetrieverWithSource(BaseRetriever):
|
| 67 |
+
base_retriever: Any
|
| 68 |
|
| 69 |
def __init__(self, base_retriever):
|
| 70 |
super().__init__()
|