jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: openllm completion start with prompt, remove it (#1303)

Browse source
This commit is contained in:
takatost 2023-10-10 17:44:19 +08:00 • committed by GitHub
commit 875dfbbf0e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
api/core/third_party/langchain/llms/openllm.py vendored
View file

@ -66,6 +66,7 @@ class OpenLLM(LLM):
json_response = response.json() json_response = response.json()
completion = json_response["responses"][0] completion = json_response["responses"][0]
completion = completion.lstrip(prompt)
if stop is not None: if stop is not None:
completion = enforce_stop_tokens(completion, stop) completion = enforce_stop_tokens(completion, stop)