github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-15 07:36:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update cohere.py (#795)

Browse Source 
When stop tokens are set in Cohere LLM constructor, they are currently
not stripped from the response, and they should be stripped
This commit is contained in:
bair82 2023-01-30 23:55:44 +01:00 committed by GitHub
parent cacf4091c0
commit ae5695ad32
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
langchain/llms/cohere.py
View File

@ -122,6 +122,6 @@ class Cohere(LLM, BaseModel):
text = response.generations[0].text
# If stop tokens are provided, Cohere's endpoint returns them.
# In order to make this consistent with other endpoints, we strip them.
if stop is not None:
text = enforce_stop_tokens(text, stop)
if stop is not None or self.stop is not None:
text = enforce_stop_tokens(text, params["stop_sequences"])
return text