alexkueck commited on
Commit
e4efa50
1 Parent(s): ede8fc0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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,metadata_field_info,enable_limit=True,verbose=True,)
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