From 53425c36b45cc0d1a1e5b1f502594055554b442c Mon Sep 17 00:00:00 2001 From: ilai Date: Sun, 1 Oct 2023 21:23:54 +0300 Subject: [PATCH] [test] test_app - bug fixes after merging dev --- tests/e2e/test_app.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/e2e/test_app.py b/tests/e2e/test_app.py index e6d3d8cf..705d64e8 100644 --- a/tests/e2e/test_app.py +++ b/tests/e2e/test_app.py @@ -3,7 +3,7 @@ from tenacity import retry, stop_after_attempt, wait_fixed from resin.knoweldge_base import KnowledgeBase -from resin.knoweldge_base.tokenizer import OpenAITokenizer, Tokenizer +from resin.tokenizer import OpenAITokenizer, Tokenizer from resin_cli.app import app, _init_engines, _init_logging from resin_cli.api_models import HealthStatus, ContextUpsertRequest, ContextQueryRequest @@ -21,8 +21,6 @@ def test_e2e(): Tokenizer.initialize(OpenAITokenizer, "gpt-3.5-turbo") kb = KnowledgeBase.create_with_new_index( index_name=os.environ["INDEX_NAME"], - encoder=KnowledgeBase.DEFAULT_RECORD_ENCODER(), - chunker=KnowledgeBase.DEFAULT_CHUNKER(), ) Tokenizer.clear() _init_engines()