Update app.py
Browse files
app.py
CHANGED
@@ -22,6 +22,15 @@ from langchain.retrievers.self_query.base import SelfQueryRetriever
|
|
22 |
from langchain.chains.query_constructor.base import AttributeInfo
|
23 |
|
24 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
25 |
def create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type='mmr', k=7, chunk_size=250, chunk_overlap=20):
|
26 |
data_path = "data"
|
27 |
model_name = "Alibaba-NLP/gte-base-en-v1.5"
|
|
|
22 |
from langchain.chains.query_constructor.base import AttributeInfo
|
23 |
|
24 |
|
25 |
+
lang_api_key = os.getenv("lang_api_key")
|
26 |
+
|
27 |
+
os.environ["LANGCHAIN_TRACING_V2"] = "true"
|
28 |
+
os.environ["LANGCHAIN_ENDPOINT"] = "https://api.langchain.plus"
|
29 |
+
os.environ["LANGCHAIN_API_KEY"] = lang_api_key
|
30 |
+
os.environ["LANGCHAIN_PROJECT"] = "Chat with multiple PDFs"
|
31 |
+
|
32 |
+
|
33 |
+
|
34 |
def create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type='mmr', k=7, chunk_size=250, chunk_overlap=20):
|
35 |
data_path = "data"
|
36 |
model_name = "Alibaba-NLP/gte-base-en-v1.5"
|