summaryrefslogtreecommitdiff
path: root/continuedev
diff options
context:
space:
mode:
authorNate Sesti <sestinj@gmail.com>2023-07-28 14:57:54 -0700
committerNate Sesti <sestinj@gmail.com>2023-07-28 14:57:54 -0700
commit7894c8ed1517394aa00f6e496a97d9e27d204f5f (patch)
treee5d10c4ff0afc0568a24ea69dc548ee0023a1a8e /continuedev
parent62db7c56e3837d1b61b672a24b72fb2ac07e0c40 (diff)
downloadsncontinue-7894c8ed1517394aa00f6e496a97d9e27d204f5f.tar.gz
sncontinue-7894c8ed1517394aa00f6e496a97d9e27d204f5f.tar.bz2
sncontinue-7894c8ed1517394aa00f6e496a97d9e27d204f5f.zip
fix: :goal_net: catch errors when loading to meilisearch index
Diffstat (limited to 'continuedev')
-rw-r--r--continuedev/src/continuedev/core/context.py7
-rw-r--r--continuedev/src/continuedev/libs/util/telemetry.py14
2 files changed, 19 insertions, 2 deletions
diff --git a/continuedev/src/continuedev/core/context.py b/continuedev/src/continuedev/core/context.py
index f81fa57a..12908e02 100644
--- a/continuedev/src/continuedev/core/context.py
+++ b/continuedev/src/continuedev/core/context.py
@@ -166,8 +166,11 @@ class ContextManager:
for item in context_items
]
if len(documents) > 0:
- async with Client('http://localhost:7700') as search_client:
- await search_client.index(SEARCH_INDEX_NAME).add_documents(documents)
+ try:
+ 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)
# def compile_chat_messages(self, max_tokens: int) -> List[Dict]:
# """
diff --git a/continuedev/src/continuedev/libs/util/telemetry.py b/continuedev/src/continuedev/libs/util/telemetry.py
index e1efb668..6d1d4fed 100644
--- a/continuedev/src/continuedev/libs/util/telemetry.py
+++ b/continuedev/src/continuedev/libs/util/telemetry.py
@@ -24,6 +24,18 @@ class PostHogLogger:
self.allow_anonymous_telemetry = allow_anonymous_telemetry or True
def capture_event(self, event_name: str, event_properties: Any):
+ print("------- Logging event -------")
+ telemetry_path = os.path.expanduser("~/.continue/telemetry.log")
+
+ # Make sure the telemetry file exists
+ if not os.path.exists(telemetry_path):
+ os.makedirs(os.path.dirname(telemetry_path), exist_ok=True)
+ open(telemetry_path, "w").close()
+
+ with open(telemetry_path, "a") as f:
+ str_to_write = f"{event_name}: {event_properties}\n{self.unique_id}\n{self.allow_anonymous_telemetry}\n\n"
+ f.write(str_to_write)
+
if not self.allow_anonymous_telemetry:
return
@@ -34,5 +46,7 @@ class PostHogLogger:
self.posthog.capture(self.unique_id, event_name,
clean_pii_from_any(event_properties))
+ print("------- Event logged -------")
+
posthog_logger = PostHogLogger(api_key=POSTHOG_API_KEY)