as-cle-bert commited on
Commit
0f227dc
·
verified ·
1 Parent(s): 959a39f

Update ChatCohere.py

Browse files
Files changed (1) hide show
  1. ChatCohere.py +1 -1
ChatCohere.py CHANGED
@@ -10,7 +10,7 @@ co = cohere.ClientV2(cohere_api_key)
10
  def chat_completion(message: str, context: str) -> str:
11
  response = co.chat(
12
  model="command-r-plus-08-2024",
13
- messages=[{"role": "system", "content": "You are a Pokemon expert. You know everything about their characteristics, abilities, and evolutions. You always reply with the most accurate information, you are polite and helpful. Your output cannot be larger than 1500 characters (spaces included)."}, {"role": "user", "content": message}, {"role": "assistant", "content": context}],
14
  )
15
  return response.message.content[0].text
16
 
 
10
  def chat_completion(message: str, context: str) -> str:
11
  response = co.chat(
12
  model="command-r-plus-08-2024",
13
+ messages=[{"role": "system", "content": "You are a Pokemon expert. You know everything about their characteristics, abilities, and evolutions. You always reply with the most accurate information, you are polite and helpful. Your output cannot be larger than 1500 characters (spaces included). Don't talk about things that the user did not mention in their message."}, {"role": "user", "content": message}, {"role": "assistant", "content": context}],
14
  )
15
  return response.message.content[0].text
16