diff options
| author | Nate Sesti <sestinj@gmail.com> | 2023-07-26 12:53:50 -0700 | 
|---|---|---|
| committer | Nate Sesti <sestinj@gmail.com> | 2023-07-26 12:53:50 -0700 | 
| commit | 00775f54e6c3fa8044a996ea1a7cf0f2205735dd (patch) | |
| tree | f6f62062e7b0544698a75c9d871b865835968595 /continuedev/src | |
| parent | f60f867cfa68be43a737fa6fcb61ac9a537de80c (diff) | |
| download | sncontinue-00775f54e6c3fa8044a996ea1a7cf0f2205735dd.tar.gz sncontinue-00775f54e6c3fa8044a996ea1a7cf0f2205735dd.tar.bz2 sncontinue-00775f54e6c3fa8044a996ea1a7cf0f2205735dd.zip  | |
fix: :ambulance: catch error from meilisearch client.health
Diffstat (limited to 'continuedev/src')
| -rw-r--r-- | continuedev/src/continuedev/core/context.py | 12 | ||||
| -rw-r--r-- | continuedev/src/continuedev/server/meilisearch_server.py | 9 | 
2 files changed, 16 insertions, 5 deletions
diff --git a/continuedev/src/continuedev/core/context.py b/continuedev/src/continuedev/core/context.py index 4a141830..f81fa57a 100644 --- a/continuedev/src/continuedev/core/context.py +++ b/continuedev/src/continuedev/core/context.py @@ -136,8 +136,16 @@ class ContextManager:      @classmethod      async def create(cls, context_providers: List[ContextProvider]):          async with Client('http://localhost:7700') as search_client: -            health = await search_client.health() -            if not health.status == "available": +            meilisearch_running = True +            try: + +                health = await search_client.health() +                if not health.status == "available": +                    meilisearch_running = False +            except: +                meilisearch_running = False + +            if not meilisearch_running:                  print(                      "MeiliSearch not running, avoiding any dependent context providers")                  context_providers = list( diff --git a/continuedev/src/continuedev/server/meilisearch_server.py b/continuedev/src/continuedev/server/meilisearch_server.py index 8beab6c9..86b56838 100644 --- a/continuedev/src/continuedev/server/meilisearch_server.py +++ b/continuedev/src/continuedev/server/meilisearch_server.py @@ -55,10 +55,13 @@ async def check_meilisearch_running() -> bool:      try:          async with Client('http://localhost:7700') as client: -            resp = await client.health() -            if resp["status"] != "available": +            try: +                resp = await client.health() +                if resp["status"] != "available": +                    return False +                return True +            except:                  return False -            return True      except Exception:          return False  | 
