bstraehle commited on
Commit
08b6d98
·
1 Parent(s): c962a65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
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(rag_option == RAG_OFF,
 
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,