ehagey commited on
Commit
0dc9ad9
·
verified ·
1 Parent(s): e251f43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -124,14 +124,14 @@ def get_model_response(question, options, prompt_template, model_name):
124
  provider = model_config["provider"]
125
 
126
  if provider == "togetherai":
127
- response = togetherai_client.chat.completions.create(
128
  model=model_config["model_id"],
129
  messages=[{"role": "user", "content": prompt}]
130
  )
131
  response_text = response.choices[0].message.content.strip()
132
 
133
  elif provider == "openai":
134
- response = openai_client.chat.completions.create(
135
  model=model_config["model_id"],
136
  messages=[{
137
  "role": "user",
@@ -140,12 +140,12 @@ def get_model_response(question, options, prompt_template, model_name):
140
  response_text = response.choices[0].message.content.strip()
141
 
142
  elif provider == "anthropic":
143
- response = anthropic_client.messages.create(
144
  model=model_config["model_id"],
145
  messages=[{"role": "user", "content": prompt}],
146
  max_tokens=4096
147
  )
148
- response_text = response.content[0].text
149
 
150
  elif provider == "google":
151
  model = genai.GenerativeModel(
 
124
  provider = model_config["provider"]
125
 
126
  if provider == "togetherai":
127
+ response = st.session_state.togetherai_client.chat.completions.create(
128
  model=model_config["model_id"],
129
  messages=[{"role": "user", "content": prompt}]
130
  )
131
  response_text = response.choices[0].message.content.strip()
132
 
133
  elif provider == "openai":
134
+ response = st.session_state.openai_client.chat.completions.create(
135
  model=model_config["model_id"],
136
  messages=[{
137
  "role": "user",
 
140
  response_text = response.choices[0].message.content.strip()
141
 
142
  elif provider == "anthropic":
143
+ response = st.session_state.anthropic_client.messages.create(
144
  model=model_config["model_id"],
145
  messages=[{"role": "user", "content": prompt}],
146
  max_tokens=4096
147
  )
148
+ response_text = response.content[0].text
149
 
150
  elif provider == "google":
151
  model = genai.GenerativeModel(