enotkrutoy commited on
Commit
ef67efc
·
verified ·
1 Parent(s): c0aec1e

Update g1.py

Browse files
Files changed (1) hide show
  1. g1.py +6 -1
g1.py CHANGED
@@ -21,12 +21,14 @@ class ResponseGenerator:
21
 
22
  def _make_api_call(self, messages, max_tokens, is_final=False):
23
  try:
 
24
  response = groq.Groq().chat.completions.create(
25
  model=self.model,
26
  messages=messages,
27
  max_tokens=max_tokens,
28
  temperature=0.5,
29
  **({"response_format": {"type": "json_object"}} if not is_final else {})
 
30
  return response.choices[0].message.content
31
  except Exception as e:
32
  logger.error(f"API Error: {str(e)}")
@@ -34,7 +36,10 @@ class ResponseGenerator:
34
 
35
  def generate_response(self, prompt: str) -> Generator:
36
  messages = [
37
- {"role": "system", "content": "You're an AI assistant. Explain your reasoning step-by-step in JSON format."},
 
 
 
38
  {"role": "user", "content": prompt}
39
  ]
40
 
 
21
 
22
  def _make_api_call(self, messages, max_tokens, is_final=False):
23
  try:
24
+ # Исправлено: добавлена закрывающая скобка
25
  response = groq.Groq().chat.completions.create(
26
  model=self.model,
27
  messages=messages,
28
  max_tokens=max_tokens,
29
  temperature=0.5,
30
  **({"response_format": {"type": "json_object"}} if not is_final else {})
31
+ )
32
  return response.choices[0].message.content
33
  except Exception as e:
34
  logger.error(f"API Error: {str(e)}")
 
36
 
37
  def generate_response(self, prompt: str) -> Generator:
38
  messages = [
39
+ {
40
+ "role": "system",
41
+ "content": "You're an AI assistant. Explain your reasoning step-by-step in JSON format."
42
+ },
43
  {"role": "user", "content": prompt}
44
  ]
45