From c918bb3af5ec4a4a409eb0a3add27951b00c3c59 Mon Sep 17 00:00:00 2001
From: Nate Sesti <sestinj@gmail.com>
Date: Thu, 14 Sep 2023 11:26:20 -0700
Subject: fix: :bug: a few minor fixes

---
 continuedev/src/continuedev/libs/llm/llamacpp.py       |  2 +-
 continuedev/src/continuedev/libs/llm/prompts/edit.py   | 10 ++++------
 continuedev/src/continuedev/models/reference/test.py   |  2 +-
 continuedev/src/continuedev/plugins/steps/core/core.py |  1 -
 4 files changed, 6 insertions(+), 9 deletions(-)

(limited to 'continuedev/src')

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),
             )
-- 
cgit v1.2.3-70-g09d2