Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -201,7 +201,7 @@ def document_retrieval_chroma(prompt):
|
|
201 |
db = Chroma(embedding_function = embeddings, persist_directory = PATH_WORK + CHROMA_DIR)
|
202 |
print ("Chroma DB bereit ...................")
|
203 |
llm = ChatOpenAI(temperature=0.5)
|
204 |
-
retriever = SelfQueryRetriever.from_llm(llm,vectorstore,document_content_description=prompt,
|
205 |
|
206 |
return db, retriever
|
207 |
|
|
|
201 |
db = Chroma(embedding_function = embeddings, persist_directory = PATH_WORK + CHROMA_DIR)
|
202 |
print ("Chroma DB bereit ...................")
|
203 |
llm = ChatOpenAI(temperature=0.5)
|
204 |
+
retriever = SelfQueryRetriever.from_llm(llm,vectorstore,document_content_description=prompt,enable_limit=True,verbose=True,)
|
205 |
|
206 |
return db, retriever
|
207 |
|