Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -109,6 +109,7 @@ def rag_chain(llm, prompt, db):
|
|
109 |
retriever = db.as_retriever(search_kwargs = {"k": 3}),
|
110 |
return_source_documents = True)
|
111 |
completion = rag_chain({"query": prompt})
|
|
|
112 |
return completion["result"]
|
113 |
|
114 |
def wandb_log(prompt, completion, rag_option):
|
|
|
109 |
retriever = db.as_retriever(search_kwargs = {"k": 3}),
|
110 |
return_source_documents = True)
|
111 |
completion = rag_chain({"query": prompt})
|
112 |
+
print(completion)
|
113 |
return completion["result"]
|
114 |
|
115 |
def wandb_log(prompt, completion, rag_option):
|