Spaces:
Sleeping
Sleeping
Update core/analyze.py
Browse files- core/analyze.py +4 -1
core/analyze.py
CHANGED
|
@@ -13,6 +13,9 @@ logger = logging.getLogger(__name__)
|
|
| 13 |
|
| 14 |
# Configure OpenAI Client
|
| 15 |
api_key = os.getenv("OPENROUTER_API_KEY")
|
|
|
|
|
|
|
|
|
|
| 16 |
client = OpenAI(
|
| 17 |
base_url="https://openrouter.ai/api/v1",
|
| 18 |
api_key=api_key
|
|
@@ -70,7 +73,7 @@ def analyze_transcript(transcript):
|
|
| 70 |
for attempt in range(max_retries):
|
| 71 |
try:
|
| 72 |
response = client.chat.completions.create(
|
| 73 |
-
model=
|
| 74 |
messages=[
|
| 75 |
{"role": "system", "content": "You are a helpful assistant that outputs only valid JSON."},
|
| 76 |
{"role": "user", "content": prompt}
|
|
|
|
| 13 |
|
| 14 |
# Configure OpenAI Client
|
| 15 |
api_key = os.getenv("OPENROUTER_API_KEY")
|
| 16 |
+
MODEL_NAME = os.getenv("OPENROUTER_MODEL", "arcee-ai/trinity-large-preview:free")
|
| 17 |
+
|
| 18 |
+
|
| 19 |
client = OpenAI(
|
| 20 |
base_url="https://openrouter.ai/api/v1",
|
| 21 |
api_key=api_key
|
|
|
|
| 73 |
for attempt in range(max_retries):
|
| 74 |
try:
|
| 75 |
response = client.chat.completions.create(
|
| 76 |
+
model=MODEL_NAME,
|
| 77 |
messages=[
|
| 78 |
{"role": "system", "content": "You are a helpful assistant that outputs only valid JSON."},
|
| 79 |
{"role": "user", "content": prompt}
|