fixed async streaming

libs/vertexai/langchain_google_vertexai/chat_models.py

@@ -168,6 +168,7 @@ def _convert_to_parts(message: BaseMessage) -> List[Part]:
     convert_system_message_to_human_content = None
     system_instruction = None
     for i, message in enumerate(history):
+        print(i, message)
         if isinstance(message, SystemMessage):
             if system_instruction is not None:
                 raise ValueError(
@@ -255,6 +256,7 @@ def _convert_to_parts(message: BaseMessage) -> List[Part]:
             convert_system_message_to_human_content = None
 
         vertex_message = Content(role=role, parts=parts)
+        print(vertex_messages)
         vertex_messages.append(vertex_message)
     return system_instruction, vertex_messages
 
@@ -733,7 +735,6 @@ async def _astream(
             project=self.project,
             convert_system_message_to_human=self.convert_system_message_to_human,
         )
-        message = history_gemini.pop()
         self.client = _get_client_with_sys_instruction(
             client=self.client,
             system_instruction=system_instruction,

libs/vertexai/pyproject.toml

@@ -1,7 +1,7 @@
 [tool.poetry]
 name = "langchain-google-vertexai"
 
-version = "0.1.3"
+version = "1.0.1"
 description = "An integration package connecting Google VertexAI and LangChain"
 authors = []
 readme = "README.md"

libs/vertexai/tests/integration_tests/test_chat_models.py

@@ -96,7 +96,6 @@ def test_vertexai_stream(model_name: str) -> None:
         assert isinstance(chunk, AIMessageChunk)
 
 
-@pytest.mark.xfail
 @pytest.mark.release
 async def test_vertexai_astream() -> None:
     model = ChatVertexAI(temperature=0, model_name="gemini-pro")
@@ -144,7 +143,6 @@ def test_multimodal() -> None:
 
 
 @pytest.mark.release
-@pytest.mark.xfail(reason="problem on vertex side")
 def test_multimodal_history() -> None:
     llm = ChatVertexAI(model_name="gemini-pro-vision")
     gcs_url = (