Fixing python tests with previous preprocess modes on merge

deven96 · Nov 26, 2024 · 4f7d123 · 4f7d123
1 parent e8b65a9
commit 4f7d123
Show file tree

Hide file tree

Showing 5 changed files with 2 additions and 9 deletions.
diff --git a/examples/python/book-search/insert_book.py b/examples/python/book-search/insert_book.py
@@ -20,9 +20,7 @@ async def set_client(ai_client, inputs):
     response = await ai_client.set(
         store_name=ai_store_payload_with_predicates["store_name"],
         inputs=inputs,
-        preprocess_action=ai_query.PreprocessAction__RawString(
-            ai_query.StringAction__ErrorIfTokensExceed()
-        ),
+        preprocess_action=ai_query.PreprocessAction__NoPreprocessing(),
     )
 
     print(response)

diff --git a/sdk/ahnlich-client-py/ahnlich_client_py/builders/non_blocking/ai.py b/sdk/ahnlich-client-py/ahnlich_client_py/builders/non_blocking/ai.py
@@ -6,7 +6,6 @@
 
 
 class AsyncAhnlichAIRequestBuilder(AhnlichAIRequestBuilder):
-
     def __init__(self, tracing_id: str = None, client: BaseClient = None) -> None:
         self.queries: typing.List[ai_query.AIQuery] = []
         self.tracing_id = tracing_id

diff --git a/sdk/ahnlich-client-py/ahnlich_client_py/builders/non_blocking/db.py b/sdk/ahnlich-client-py/ahnlich_client_py/builders/non_blocking/db.py
@@ -6,7 +6,6 @@
 
 
 class AsyncAhnlichDBRequestBuilder(AhnlichDBRequestBuilder):
-
     def __init__(self, tracing_id: str = None, client: BaseClient = None) -> None:
         self.queries: typing.List[db_query.Query] = []
         self.tracing_id = tracing_id

diff --git a/sdk/ahnlich-client-py/ahnlich_client_py/tests/ai_client/test_async_client_commands.py b/sdk/ahnlich-client-py/ahnlich_client_py/tests/ai_client/test_async_client_commands.py
@@ -116,9 +116,7 @@ async def test_ai_client_get_pred(spin_up_ahnlich_ai):
     builder.set(
         store_name=ai_store_payload_with_predicates["store_name"],
         inputs=store_inputs,
-        preprocess_action=ai_query.PreprocessAction__RawString(
-            ai_query.StringAction__ErrorIfTokensExceed()
-        ),
+        preprocess_action=ai_query.PreprocessAction__NoPreprocessing(),
     )
     expected = ai_response.AIServerResult(
         results=[

diff --git a/sdk/ahnlich-client-py/ahnlich_client_py/tests/db_client/test_async_client_commands.py b/sdk/ahnlich-client-py/ahnlich_client_py/tests/db_client/test_async_client_commands.py
@@ -23,7 +23,6 @@ async def test_client_sends_ping_to_db_success(module_scopped_ahnlich_db):
     port = module_scopped_ahnlich_db
     db_client = non_blocking.AhnlichDBClient(address="127.0.0.1", port=port)
     try:
-
         response: db_response.ServerResult = await db_client.ping()
         assert len(response.results) == 1
         assert response.results[0] == db_response.Result__Ok(