refactor: :loud_sound: replace all print with logger.debug

3 files changed, 27 insertions, 19 deletions

diff --git a/continuedev/src/continuedev/core/autopilot.py b/continuedev/src/continuedev/core/autopilot.py
index 42a58423..931cfb75 100644
--- a/continuedev/src/continuedev/core/autopilot.py
+++ b/continuedev/src/continuedev/core/autopilot.py
@@ -21,6 +21,7 @@ from ..libs.util.traceback_parsers import get_python_traceback, get_javascript_t
 from openai import error as openai_errors
 from ..libs.util.create_async_task import create_async_task
 from ..libs.util.telemetry import posthog_logger
+from ..libs.util.logging import logger
 
 
 def get_error_title(e: Exception) -> str:
@@ -153,7 +154,7 @@ class Autopilot(ContinueBaseModel):
 
                 await self.update_subscribers()
         except Exception as e:
-            print(e)
+            logger.debug(e)
 
     def handle_manual_edits(self, edits: List[FileEditWithFullContents]):
         for edit in edits:
@@ -258,7 +259,7 @@ class Autopilot(ContinueBaseModel):
                 e)
 
             # Attach an InternalErrorObservation to the step and unhide it.
-            print(
+            logger.error(
                 f"Error while running step: \n{error_string}\n{error_title}")
             posthog_logger.capture_event('step error', {
                                          'error_message': error_string, 'error_title': error_title, 'step_name': step.name, 'params': step.dict()})
diff --git a/continuedev/src/continuedev/core/context.py b/continuedev/src/continuedev/core/context.py
index 12908e02..78a747b2 100644
--- a/continuedev/src/continuedev/core/context.py
+++ b/continuedev/src/continuedev/core/context.py
@@ -7,7 +7,7 @@ from pydantic import BaseModel
 
 from .main import ChatMessage, ContextItem, ContextItemDescription, ContextItemId
 from ..server.meilisearch_server import check_meilisearch_running
-
+from ..libs.util.logging import logger
 
 SEARCH_INDEX_NAME = "continue_context_items"
 
@@ -57,16 +57,22 @@ class ContextProvider(BaseModel):
         Default implementation uses the search index to get the item.
         """
         async with Client('http://localhost:7700') as search_client:
-            result = await search_client.index(
-                SEARCH_INDEX_NAME).get_document(id.to_string())
-            return ContextItem(
-                description=ContextItemDescription(
-                    name=result["name"],
-                    description=result["description"],
-                    id=id
-                ),
-                content=result["content"]
-            )
+            try:
+                result = await search_client.index(
+                    SEARCH_INDEX_NAME).get_document(id.to_string())
+                return ContextItem(
+                    description=ContextItemDescription(
+                        name=result["name"],
+                        description=result["description"],
+                        id=id
+                    ),
+                    content=result["content"]
+                )
+            except Exception as e:
+                logger.warning(
+                    f"Error while retrieving document from meilisearch: {e}")
+
+            return None
 
     async def delete_context_with_ids(self, ids: List[ContextItemId]):
         """
@@ -100,8 +106,8 @@ class ContextProvider(BaseModel):
             if item.description.id.item_id == id.item_id:
                 return
 
-        new_item = await self.get_item(id, query)
-        self.selected_items.append(new_item)
+        if new_item := await self.get_item(id, query):
+            self.selected_items.append(new_item)
 
 
 class ContextManager:
@@ -146,7 +152,7 @@ class ContextManager:
                 meilisearch_running = False
 
             if not meilisearch_running:
-                print(
+                logger.warning(
                     "MeiliSearch not running, avoiding any dependent context providers")
                 context_providers = list(
                     filter(lambda cp: cp.title == "code", context_providers))
@@ -170,7 +176,7 @@ class ContextManager:
                     async with Client('http://localhost:7700') as search_client:
                         await search_client.index(SEARCH_INDEX_NAME).add_documents(documents)
                 except Exception as e:
-                    print("Error loading meilisearch index: ", e)
+                    logger.debug(f"Error loading meilisearch index: {e}")
 
     # def compile_chat_messages(self, max_tokens: int) -> List[Dict]:
     #     """
diff --git a/continuedev/src/continuedev/core/sdk.py b/continuedev/src/continuedev/core/sdk.py
index 264a6852..d75aac00 100644
--- a/continuedev/src/continuedev/core/sdk.py
+++ b/continuedev/src/continuedev/core/sdk.py
@@ -19,6 +19,7 @@ from ..plugins.steps.core.core import *
 from ..libs.llm.proxy_server import ProxyServer
 from ..libs.util.telemetry import posthog_logger
 from ..libs.util.paths import getConfigFilePath
+from ..libs.util.logging import logger
 
 
 class Autopilot:
@@ -161,7 +162,7 @@ class ContinueSDK(AbstractContinueSDK):
             config = sdk._load_config_dot_py()
             sdk.config = config
         except Exception as e:
-            print(e)
+            logger.debug(e)
             sdk.config = ContinueConfig()
             msg_step = MessageStep(
                 name="Invalid Continue Config File", message=e.__repr__())
@@ -273,7 +274,7 @@ class ContinueSDK(AbstractContinueSDK):
 
             return config.config
         except Exception as e:
-            print("Error loading config.py: ", e)
+            logger.debug(f"Error loading config.py: {e}")
             return ContinueConfig() if self._last_valid_config is None else self._last_valid_config
 
     def get_code_context(self, only_editing: bool = False) -> List[RangeInFileWithContents]: