Spaces:
Running
Running
as-cle-bert
commited on
Update ChatCohere.py
Browse files- 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 |
|