ex510 commited on
Commit
5ec2b90
·
verified ·
1 Parent(s): 577e8bf

Update core/analyze.py

Browse files
Files changed (1) hide show
  1. 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="arcee-ai/trinity-large-preview:free",
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}