fix(community): add mypy assert for data_store_id in VertexAIMultiTurnSearchRetriever

alexrohv · claude · alexrohv · commit b096490fc2b9 · 2026-03-10T19:05:14.000-04:00
Add missing None guard before conversation_path() call to satisfy mypy
type narrowing for the Optional[str] data_store_id field.

Co-Authored-By: Claude Sonnet 4.6 &lt;noreply@anthropic.com&gt;
diff --git a/libs/community/langchain_google_community/vertex_ai_search.py b/libs/community/langchain_google_community/vertex_ai_search.py
@@ -774,6 +774,7 @@ def _get_relevant_documents(
                 TextInput,
             )
 
+        assert self.data_store_id is not None  # noqa: S101
         request = ConverseConversationRequest(
             name=self._client.conversation_path(
                 self.project_id,

Original file line number	Diff line number	Diff line change
`@@ -774,6 +774,7 @@ def _get_relevant_documents(`
`774`	`774`	`TextInput,`
`775`	`775`	`)`
`776`	`776`
	`777`	`+ assert self.data_store_id is not None # noqa: S101`
`777`	`778`	`request = ConverseConversationRequest(`
`778`	`779`	`name=self._client.conversation_path(`
`779`	`780`	`self.project_id,`