fix assertions in embed integration tests

Author: austin-denoble

tests/integration/control/resources/index/test_configure.py

@@ -47,7 +47,7 @@ def test_configure_index_embed(self, pc, create_sl_index_params):
         create_sl_index_params["dimension"] = 1024
         pc.db.index.create(**create_sl_index_params)
         desc = pc.db.index.describe_index(name)
-        assert desc.embed == None
+        assert desc.embed is None
 
         embed_config = {
             "model": "multilingual-e5-large",
@@ -58,11 +58,13 @@ def test_configure_index_embed(self, pc, create_sl_index_params):
         desc = pc.db.index.describe_index(name)
         assert desc.embed.model == "multilingual-e5-large"
         assert desc.embed.field_map == {"text": "chunk_text"}
-        assert desc.embed.read_parameters == {}
-        assert desc.embed.write_parameters == {}
+        assert desc.embed.read_parameters == {"input_type": "query", "truncate": "END"}
+        assert desc.embed.write_parameters == {
+            "input_type": "passage",
+            "truncate": "END",
+        }
         assert desc.embed.vector_type == "dense"
         assert desc.embed.dimension == 1024
         assert desc.embed.metric == "cosine"
-        assert desc.embed.model_type == "dense"
 
         pc.db.index.delete_index(name)

tests/integration/control/serverless/test_configure_index_embed.py

@@ -4,7 +4,7 @@ def test_convert_index_to_integrated(self, client, create_sl_index_params):
         create_sl_index_params["dimension"] = 1024
         client.create_index(**create_sl_index_params)
         desc = client.describe_index(name)
-        assert desc.embed == None
+        assert desc.embed is None
 
         embed_config = {
             "model": "multilingual-e5-large",
@@ -15,11 +15,13 @@ def test_convert_index_to_integrated(self, client, create_sl_index_params):
         desc = client.describe_index(name)
         assert desc.embed.model == "multilingual-e5-large"
         assert desc.embed.field_map == {"text": "chunk_text"}
-        assert desc.embed.read_parameters == {}
-        assert desc.embed.write_parameters == {}
+        assert desc.embed.read_parameters == {"input_type": "query", "truncate": "END"}
+        assert desc.embed.write_parameters == {
+            "input_type": "passage",
+            "truncate": "END",
+        }
         assert desc.embed.vector_type == "dense"
         assert desc.embed.dimension == 1024
         assert desc.embed.metric == "cosine"
-        assert desc.embed.model_type == "dense"
 
         client.delete_index(name)

tests/integration/control_asyncio/test_configure_index_embed.py

@@ -8,7 +8,7 @@ async def test_convert_index_to_integrated(self, create_sl_index_params):
         create_sl_index_params["dimension"] = 1024
         await pc.create_index(**create_sl_index_params)
         desc = await pc.describe_index(name)
-        assert desc.embed == None
+        assert desc.embed is None
 
         embed_config = {
             "model": "multilingual-e5-large",
@@ -19,11 +19,13 @@ async def test_convert_index_to_integrated(self, create_sl_index_params):
         desc = await pc.describe_index(name)
         assert desc.embed.model == "multilingual-e5-large"
         assert desc.embed.field_map == {"text": "chunk_text"}
-        assert desc.embed.read_parameters == {}
-        assert desc.embed.write_parameters == {}
+        assert desc.embed.read_parameters == {"input_type": "query", "truncate": "END"}
+        assert desc.embed.write_parameters == {
+            "input_type": "passage",
+            "truncate": "END",
+        }
         assert desc.embed.vector_type == "dense"
         assert desc.embed.dimension == 1024
         assert desc.embed.metric == "cosine"
-        assert desc.embed.model_type == "dense"
 
         await pc.delete_index(name)