codelion commited on
Commit
3eed064
·
verified ·
1 Parent(s): df81afd

Update github_repo_analyzer.py

Browse files
Files changed (1) hide show
  1. github_repo_analyzer.py +7 -7
github_repo_analyzer.py CHANGED
@@ -149,14 +149,14 @@ def analyze_pull_requests(github_repo, max_prs):
149
  }
150
 
151
  def call_llm(client, prompt, model="nousresearch/hermes-3-llama-3.1-405b:free", max_tokens=4096):
152
- message = client.chat.completions.create(
153
- max_tokens=max_tokens,
154
- model=model,
155
- messages=[
156
- {"role": "user", "content": prompt}
157
- ]
158
  )
159
- return message.content[0].message.content
160
 
161
  def safe_call_llm(client, prompt, retries=3):
162
  for attempt in range(retries):
 
149
  }
150
 
151
  def call_llm(client, prompt, model="nousresearch/hermes-3-llama-3.1-405b:free", max_tokens=4096):
152
+ response = client.chat.completions.create(
153
+ model=model,
154
+ messages=[
155
+ {"role": "user", "content": prompt}
156
+ ],
157
+ max_tokens=max_tokens,
158
  )
159
+ return response.choices[0].message.content
160
 
161
  def safe_call_llm(client, prompt, retries=3):
162
  for attempt in range(retries):