NLPGenius commited on
Commit
83a948c
·
1 Parent(s): 0019932

Fix type annotation in RetrieverWithSource class

Browse files
Files changed (1) hide show
  1. 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: any
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__()