Madras1 commited on
Commit
fbe0aed
·
verified ·
1 Parent(s): e5ab6c2

Update jade/scholar.py

Browse files
Files changed (1) hide show
  1. jade/scholar.py +2 -2
jade/scholar.py CHANGED
@@ -225,11 +225,11 @@ class LLMEngine:
225
  def __init__(self, api_key=None):
226
  self.api_key = api_key or os.environ.get("GROQ_API_KEY")
227
  self.client = groq.Groq(api_key=self.api_key)
228
- self.model = "llama-3.3-70b-versatile"
229
 
230
  def chat(self, messages: List[Dict], json_mode=False) -> str:
231
  try:
232
- kwargs = {"messages": messages, "model": self.model, "temperature": 0.6}
233
  if json_mode: kwargs["response_format"] = {"type": "json_object"}
234
  return self.client.chat.completions.create(**kwargs).choices[0].message.content
235
  except Exception as e:
 
225
  def __init__(self, api_key=None):
226
  self.api_key = api_key or os.environ.get("GROQ_API_KEY")
227
  self.client = groq.Groq(api_key=self.api_key)
228
+ self.model = "moonshotai/kimi-k2-instruct-0905"
229
 
230
  def chat(self, messages: List[Dict], json_mode=False) -> str:
231
  try:
232
+ kwargs = {"messages": messages, "model": self.model, "temperature": 0.8}
233
  if json_mode: kwargs["response_format"] = {"type": "json_object"}
234
  return self.client.chat.completions.create(**kwargs).choices[0].message.content
235
  except Exception as e: