Spaces:
Sleeping
Sleeping
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 |
|