fix: :bug: a number of small fixes + disable summaries

3 files changed, 21 insertions, 8 deletions

diff --git a/continuedev/src/continuedev/core/autopilot.py b/continuedev/src/continuedev/core/autopilot.py
index a943a35f..f24f5ad7 100644
--- a/continuedev/src/continuedev/core/autopilot.py
+++ b/continuedev/src/continuedev/core/autopilot.py
@@ -465,6 +465,9 @@ class Autopilot(ContinueBaseModel):
 
         # Update its description
         async def update_description():
+            if self.continue_sdk.config.disable_summaries:
+                return
+
             description = await step.describe(self.continue_sdk.models)
             if description is not None:
                 step.description = description
@@ -534,11 +537,15 @@ class Autopilot(ContinueBaseModel):
                 ):
                     return
 
-                title = await self.continue_sdk.models.medium.complete(
-                    f'Give a short title to describe the current chat session. Do not put quotes around the title. The first message was: "{user_input}". Do not use more than 10 words. The title is: ',
-                    max_tokens=20,
-                )
-                title = remove_quotes_and_escapes(title)
+                if self.continue_sdk.config.disable_summaries:
+                    title = user_input
+                else:
+                    title = await self.continue_sdk.models.medium.complete(
+                        f'Give a short title to describe the current chat session. Do not put quotes around the title. The first message was: "{user_input}". Do not use more than 10 words. The title is: ',
+                        max_tokens=20,
+                    )
+                    title = remove_quotes_and_escapes(title)
+
                 self.session_info = SessionInfo(
                     title=title,
                     session_id=self.ide.session_id,
diff --git a/continuedev/src/continuedev/core/config.py b/continuedev/src/continuedev/core/config.py
index e8b2c579..d431c704 100644
--- a/continuedev/src/continuedev/core/config.py
+++ b/continuedev/src/continuedev/core/config.py
@@ -93,6 +93,10 @@ class ContinueConfig(BaseModel):
         "https://us-west1-autodebug.cloudfunctions.net",
         description="The URL of the server where development data is sent. No data is sent unless a valid user token is provided.",
     )
+    disable_summaries: Optional[bool] = Field(
+        False,
+        description="If set to `True`, Continue will not generate summaries for each Step. This can be useful if you want to save on compute.",
+    )
 
     @validator("temperature", pre=True)
     def temperature_validator(cls, v):
diff --git a/continuedev/src/continuedev/core/sdk.py b/continuedev/src/continuedev/core/sdk.py
index de209114..12fce1c6 100644
--- a/continuedev/src/continuedev/core/sdk.py
+++ b/continuedev/src/continuedev/core/sdk.py
@@ -2,12 +2,11 @@ import os
 import traceback
 from typing import Coroutine, List, Optional, Union
 
-from ..libs.util.create_async_task import create_async_task
-
 from ..libs.llm import LLM
+from ..libs.util.create_async_task import create_async_task
 from ..libs.util.devdata import dev_data_logger
 from ..libs.util.logging import logger
-from ..libs.util.paths import getConfigFilePath
+from ..libs.util.paths import getConfigFilePath, getDiffsFolderPath
 from ..libs.util.telemetry import posthog_logger
 from ..models.filesystem import RangeInFile
 from ..models.filesystem_edit import (
@@ -69,6 +68,9 @@ class ContinueSDK(AbstractContinueSDK):
         sdk = ContinueSDK(autopilot)
         autopilot.continue_sdk = sdk
 
+        # Create necessary directories
+        getDiffsFolderPath()
+
         try:
             sdk.config = config or sdk._load_config_dot_py()
         except Exception as e: