diff options
Diffstat (limited to 'continuedev')
| -rw-r--r-- | continuedev/requirements.txt | 3 | ||||
| -rw-r--r-- | continuedev/src/continuedev/core/context.py | 10 |
2 files changed, 8 insertions, 5 deletions
diff --git a/continuedev/requirements.txt b/continuedev/requirements.txt index 8edcf70e..0cf909d5 100644 --- a/continuedev/requirements.txt +++ b/continuedev/requirements.txt @@ -21,4 +21,5 @@ meilisearch-python-async==1.4.8 socksio==1.0.0 ripgrepy==2.0.0 replicate==0.11.0 -bs4==0.0.1
\ No newline at end of file +bs4==0.0.1 +redbaron==0.9.2
\ No newline at end of file diff --git a/continuedev/src/continuedev/core/context.py b/continuedev/src/continuedev/core/context.py index 326fb441..d47ad942 100644 --- a/continuedev/src/continuedev/core/context.py +++ b/continuedev/src/continuedev/core/context.py @@ -248,10 +248,12 @@ class ContextManager: for item in context_items ] async with Client("http://localhost:7700") as search_client: - await asyncio.wait_for( - search_client.index(SEARCH_INDEX_NAME).add_documents(documents), - timeout=5, - ) + + async def add_docs(): + index = await search_client.get_index(SEARCH_INDEX_NAME) + await index.add_documents(documents) + + await asyncio.wait_for(add_docs(), timeout=5) @staticmethod async def delete_documents(ids): |