Skip to content

Commit

Permalink
Fixing python tests with previous preprocess modes
Browse files Browse the repository at this point in the history
  • Loading branch information
deven96 committed Nov 26, 2024
1 parent aab36d6 commit c90b72e
Showing 1 changed file with 2 additions and 6 deletions.
Original file line number Diff line number Diff line change
@@ -53,9 +53,7 @@ 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=[
@@ -196,9 +194,7 @@ def test_ai_client_del_key(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=[

0 comments on commit c90b72e

Please sign in to comment.