diff --git a/application/llm/docsgpt_provider.py b/application/llm/docsgpt_provider.py index b7d6a5a..e0c5dba 100644 --- a/application/llm/docsgpt_provider.py +++ b/application/llm/docsgpt_provider.py @@ -20,7 +20,7 @@ class DocsGPTAPILLM(BaseLLM): "max_new_tokens": 30 } ) - response_clean = response.json()['a'].split("###")[0] + response_clean = response.json()['a'].replace("###", "") return response_clean