Merge pull request #852 from arc53/feat/premaillm

fix: docsgpt provider
4 months ago · 5dcde67ae9
parent 5b9352a946 ee06fa85f1
commit 5dcde67ae9
1 changed files with 1 additions and 1 deletions
--- 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