Fix latest message index (#13355)

There is a bug which caused the earliest message rather than the latest
message being sent
This commit is contained in:
billytrend-cohere 2023-11-14 17:23:25 +00:00 committed by GitHub
parent 58f5a4d30a
commit 2f6fe6ddf3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -80,9 +80,9 @@ def get_cohere_chat_request(
) )
return { return {
"message": messages[0].content, "message": messages[-1].content,
"chat_history": [ "chat_history": [
{"role": get_role(x), "message": x.content} for x in messages[1:] {"role": get_role(x), "message": x.content} for x in messages[:-1]
], ],
"documents": documents, "documents": documents,
"connectors": maybe_connectors, "connectors": maybe_connectors,