From 3ac222784e12aa833955091ba972916a51c6918d Mon Sep 17 00:00:00 2001 From: Anshul Thakur Date: Tue, 28 Jan 2025 23:36:36 +0530 Subject: [PATCH] fix: Add a clause 'single' to constrain the LLM to not return more than one codebock (#55) --- doc_comments_ai/llm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/doc_comments_ai/llm.py b/doc_comments_ai/llm.py index 663e1a5..030d9a7 100644 --- a/doc_comments_ai/llm.py +++ b/doc_comments_ai/llm.py @@ -57,7 +57,7 @@ def __init__( "Add a detailed doc comment to the following {language} method:\n{code}\n" "The doc comment should describe what the method does. " "{inline_comments} " - "Return the method implementaion with the doc comment as a markdown code block. " + "Return the method implementaion with the doc comment as a single markdown code block. " "Don't include any explanations {haskell_missing_signature}in your response." ) self.prompt = PromptTemplate(