Spaces:
Sleeping
Sleeping
Update github_repo_analyzer.py
Browse files- 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 |
-
|
153 |
-
|
154 |
-
|
155 |
-
|
156 |
-
|
157 |
-
|
158 |
)
|
159 |
-
return
|
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):
|