Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -12,6 +12,14 @@ RAG_OFF = "Off"
|
|
12 |
RAG_CHROMA = "Chroma"
|
13 |
RAG_MONGODB = "MongoDB"
|
14 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
15 |
def invoke(openai_api_key, rag_option, prompt):
|
16 |
if (openai_api_key == ""):
|
17 |
raise gr.Error("OpenAI API Key is required.")
|
@@ -57,7 +65,8 @@ def invoke(openai_api_key, rag_option, prompt):
|
|
57 |
finally:
|
58 |
end_time_ms = round(time.time() * 1000)
|
59 |
|
60 |
-
wandb_trace(
|
|
|
61 |
prompt,
|
62 |
completion,
|
63 |
result,
|
|
|
12 |
RAG_CHROMA = "Chroma"
|
13 |
RAG_MONGODB = "MongoDB"
|
14 |
|
15 |
+
config = {
|
16 |
+
"chunk_overlap": 150,
|
17 |
+
"chunk_size": 1500,
|
18 |
+
"k": 3,
|
19 |
+
"model_name": "gpt-4-0613",
|
20 |
+
"temperature": 0,
|
21 |
+
}
|
22 |
+
|
23 |
def invoke(openai_api_key, rag_option, prompt):
|
24 |
if (openai_api_key == ""):
|
25 |
raise gr.Error("OpenAI API Key is required.")
|
|
|
65 |
finally:
|
66 |
end_time_ms = round(time.time() * 1000)
|
67 |
|
68 |
+
wandb_trace(config,
|
69 |
+
rag_option == RAG_OFF,
|
70 |
prompt,
|
71 |
completion,
|
72 |
result,
|