diff options
author | Nate Sesti <sestinj@gmail.com> | 2023-07-28 14:57:54 -0700 |
---|---|---|
committer | Nate Sesti <sestinj@gmail.com> | 2023-07-28 14:57:54 -0700 |
commit | 7894c8ed1517394aa00f6e496a97d9e27d204f5f (patch) | |
tree | e5d10c4ff0afc0568a24ea69dc548ee0023a1a8e /continuedev | |
parent | 62db7c56e3837d1b61b672a24b72fb2ac07e0c40 (diff) | |
download | sncontinue-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.py | 7 | ||||
-rw-r--r-- | continuedev/src/continuedev/libs/util/telemetry.py | 14 |
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) |