Add incorrect schema test

Author: dishaprakash

tests/unit_tests/v2/test_async_chat_message_history.py

@@ -126,3 +126,14 @@ async def test_v1_schema_support(async_engine: PGEngine) -> None:
     assert len(messages) == 4
 
     await async_engine._adrop_table(table_name=table_name)
+
+
+async def test_incorrect_schema(async_engine: PGEngine) -> None:
+    table_name = "incorrect_schema_" + str(uuid.uuid4())
+    await async_engine._ainit_vectorstore_table(table_name=table_name, vector_size=1024)
+    with pytest.raises(IndexError):
+        await AsyncPGChatMessageHistory.create(
+            engine=async_engine, session_id="test", table_name=table_name
+        )
+    query = f'DROP TABLE IF EXISTS "{table_name}"'
+    await aexecute(async_engine, query)

tests/unit_tests/v2/test_chat_message_history.py

@@ -74,17 +74,26 @@ def test_chat_table(engine: Any) -> None:
         )
 
 
-@pytest.mark.asyncio
-async def test_chat_schema(engine: Any) -> None:
-    doc_table_name = "test_table" + str(uuid.uuid4())
-    engine.init_document_table(table_name=doc_table_name)
+async def test_incorrect_schema_async(async_engine: PGEngine) -> None:
+    table_name = "incorrect_schema_" + str(uuid.uuid4())
+    await async_engine.ainit_vectorstore_table(table_name=table_name, vector_size=1024)
     with pytest.raises(IndexError):
-        PGChatMessageHistory.create_sync(
-            engine=engine, session_id="test", table_name=doc_table_name
+        await PGChatMessageHistory.create(
+            engine=async_engine, session_id="test", table_name=table_name
         )
+    query = f'DROP TABLE IF EXISTS "{table_name}"'
+    await aexecute(async_engine, query)
 
-    query = f'DROP TABLE IF EXISTS "{doc_table_name}"'
-    await aexecute(engine, query)
+
+async def test_incorrect_schema_sync(async_engine: PGEngine) -> None:
+    table_name = "incorrect_schema_" + str(uuid.uuid4())
+    async_engine.init_vectorstore_table(table_name=table_name, vector_size=1024)
+    with pytest.raises(IndexError):
+        PGChatMessageHistory.create_sync(
+            engine=async_engine, session_id="test", table_name=table_name
+        )
+    query = f'DROP TABLE IF EXISTS "{table_name}"'
+    await aexecute(async_engine, query)
 
 
 @pytest.mark.asyncio