Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -38,7 +38,6 @@ MODEL_NAME = "gpt-4"
|
|
38 |
def invoke(openai_api_key, use_rag, prompt):
|
39 |
print(os.listdir("../"))
|
40 |
print(os.listdir("../app/"))
|
41 |
-
#print(os.listdir("../data/"))
|
42 |
llm = ChatOpenAI(model_name = MODEL_NAME,
|
43 |
openai_api_key = openai_api_key,
|
44 |
temperature = 0)
|
@@ -60,6 +59,7 @@ def invoke(openai_api_key, use_rag, prompt):
|
|
60 |
chain_type_kwargs = {"prompt": RAG_CHAIN_PROMPT},
|
61 |
retriever = vector_db.as_retriever(search_kwargs = {"k": 3}),
|
62 |
return_source_documents = True)
|
|
|
63 |
result = rag_chain({"query": prompt})
|
64 |
result = result["result"]
|
65 |
else:
|
|
|
38 |
def invoke(openai_api_key, use_rag, prompt):
|
39 |
print(os.listdir("../"))
|
40 |
print(os.listdir("../app/"))
|
|
|
41 |
llm = ChatOpenAI(model_name = MODEL_NAME,
|
42 |
openai_api_key = openai_api_key,
|
43 |
temperature = 0)
|
|
|
59 |
chain_type_kwargs = {"prompt": RAG_CHAIN_PROMPT},
|
60 |
retriever = vector_db.as_retriever(search_kwargs = {"k": 3}),
|
61 |
return_source_documents = True)
|
62 |
+
print(os.listdir("/data/chroma/"))
|
63 |
result = rag_chain({"query": prompt})
|
64 |
result = result["result"]
|
65 |
else:
|