diff options
Diffstat (limited to 'continuedev')
4 files changed, 6 insertions, 9 deletions
diff --git a/continuedev/src/continuedev/libs/llm/llamacpp.py b/continuedev/src/continuedev/libs/llm/llamacpp.py index 60d9961f..c795bd15 100644 --- a/continuedev/src/continuedev/libs/llm/llamacpp.py +++ b/continuedev/src/continuedev/libs/llm/llamacpp.py @@ -1,5 +1,5 @@ import json -from typing import Any, Callable, Dict, Optional +from typing import Any, Callable, Dict import aiohttp from pydantic import Field diff --git a/continuedev/src/continuedev/libs/llm/prompts/edit.py b/continuedev/src/continuedev/libs/llm/prompts/edit.py index b4892669..7da5a192 100644 --- a/continuedev/src/continuedev/libs/llm/prompts/edit.py +++ b/continuedev/src/continuedev/libs/llm/prompts/edit.py @@ -2,19 +2,18 @@ from textwrap import dedent simplified_edit_prompt = dedent( """\ - [INST] Consider the following code: + Consider the following code: ``` {{code_to_edit}} ``` Edit the code to perfectly satisfy the following user request: {{user_input}} - Output nothing except for the code. No code block, no English explanation, no start/end tags. - [/INST]""" + Output nothing except for the code. No code block, no English explanation, no start/end tags.""" ) simplest_edit_prompt = dedent( """\ - [INST] Here is the code before editing: + Here is the code before editing: ``` {{code_to_edit}} ``` @@ -22,8 +21,7 @@ simplest_edit_prompt = dedent( Here is the edit requested: "{{user_input}}" - Here is the code after editing: - [/INST]""" + Here is the code after editing:""" ) codellama_infill_edit_prompt = "{{file_prefix}}<FILL>{{file_suffix}}" diff --git a/continuedev/src/continuedev/models/reference/test.py b/continuedev/src/continuedev/models/reference/test.py index 1c7608ec..1cebfc36 100644 --- a/continuedev/src/continuedev/models/reference/test.py +++ b/continuedev/src/continuedev/models/reference/test.py @@ -54,7 +54,7 @@ def docs_from_schema(schema, filepath, ignore_properties=[], inherited=[]): if only_required != required or prop in ignore_properties: return "" required = "true" if required else "false" - return f"""<ClassPropertyRef name='{prop}' details='{html.escape(json.dumps(details))}' required={{{required}}} default="{html.escape(str(details.get("default", "")))}"/>""" + return f"""<ClassPropertyRef name='{prop}' details='{html.escape(json.dumps(details))}' required={{{required}}} default="{html.escape(str(details.get("default", "")))}"/>\n""" for prop, details in schema["properties"].items(): property = add_property(prop, details, True) diff --git a/continuedev/src/continuedev/plugins/steps/core/core.py b/continuedev/src/continuedev/plugins/steps/core/core.py index 5e1f1cd4..61de6578 100644 --- a/continuedev/src/continuedev/plugins/steps/core/core.py +++ b/continuedev/src/continuedev/plugins/steps/core/core.py @@ -624,7 +624,6 @@ Please output the code to be inserted at the cursor in order to fulfill the user generator = model_to_use.stream_complete( rendered, - raw=True, temperature=sdk.config.temperature, max_tokens=min(max_tokens, model_to_use.context_length // 2), ) |