tatianija commited on
Commit
cadbe36
·
verified ·
1 Parent(s): 7ae69a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -249,7 +249,7 @@ Provide a brief, focused analysis:"""
249
  messages=messages,
250
  max_tokens=500,
251
  temperature=0.3,
252
- chat_template_kwargs: {"enable_thinking": false}
253
  )
254
 
255
  return response.choices[0].message.content.strip()
@@ -365,7 +365,8 @@ class IntelligentAgent:
365
  response = self.client.chat_completion(
366
  messages=messages,
367
  max_tokens=max_tokens,
368
- temperature=temperature
 
369
  )
370
  return response.choices[0].message.content.strip()
371
  except Exception as chat_error:
@@ -377,6 +378,7 @@ class IntelligentAgent:
377
  prompt,
378
  max_new_tokens=max_tokens,
379
  temperature=temperature,
 
380
  do_sample=temperature > 0
381
  )
382
  return response.strip()
@@ -646,7 +648,7 @@ Question: {question}
646
  Answer:"""
647
 
648
  try:
649
- response = self._chat_completion(answer_prompt, max_tokens=500, temperature=0.3)
650
  return response
651
 
652
  except Exception as e:
 
249
  messages=messages,
250
  max_tokens=500,
251
  temperature=0.3,
252
+ chat_template_kwargs = {"enable_thinking": false}
253
  )
254
 
255
  return response.choices[0].message.content.strip()
 
365
  response = self.client.chat_completion(
366
  messages=messages,
367
  max_tokens=max_tokens,
368
+ temperature=temperature,
369
+ chat_template_kwargs = {"enable_thinking": false}
370
  )
371
  return response.choices[0].message.content.strip()
372
  except Exception as chat_error:
 
378
  prompt,
379
  max_new_tokens=max_tokens,
380
  temperature=temperature,
381
+ chat_template_kwargs = {"enable_thinking": false},
382
  do_sample=temperature > 0
383
  )
384
  return response.strip()
 
648
  Answer:"""
649
 
650
  try:
651
+ response = self._chat_completion(answer_prompt, max_tokens=500, temperature=0.3, chat_template_kwargs = {"enable_thinking": false} )
652
  return response
653
 
654
  except Exception as e: