Spaces:
Runtime error
Runtime error
Update g1.py
Browse files
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 |
-
{
|
|
|
|
|
|
|
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 |
|