Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 =
|
| 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(
|