bstraehle commited on
Commit
4c3e605
·
1 Parent(s): 9102fcd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -133,7 +133,7 @@ def wandb_trace(rag_option, prompt, prompt_template, result, completion, chain_n
133
  end_time_ms = end_time_ms,
134
  inputs = {"rag_option": rag_option, "prompt": prompt, "prompt_template": prompt_template},
135
  outputs = {"result": str(result), "completion": str(completion)},
136
- model_dict = {"_kind": "openai"}
137
  )
138
  trace.log("test")
139
  wandb.finish()
 
133
  end_time_ms = end_time_ms,
134
  inputs = {"rag_option": rag_option, "prompt": prompt, "prompt_template": prompt_template},
135
  outputs = {"result": str(result), "completion": str(completion)},
136
+ model_dict = {"api_type": "openai", "model_alias": "gpt-4"}
137
  )
138
  trace.log("test")
139
  wandb.finish()