Apply suggestions from code review

Co-authored-by: Wei Lee <weilee.rx@gmail.com>
apache · Apr 29, 2024 · 6eb1827 · 6eb1827
1 parent a7a6b2c
commit 6eb1827
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/airflow/providers/openai/hooks/openai.py b/airflow/providers/openai/hooks/openai.py
@@ -328,7 +328,7 @@ def create_vector_store(self, **kwargs: Any) -> VectorStore:
         vector_store = self.conn.beta.vector_stores.create(**kwargs)
         return vector_store
 
-    def get_vectors_stores(self, **kwargs: Any) -> list[VectorStore]:
+    def get_vector_stores(self, **kwargs: Any) -> list[VectorStore]:
         """Return a list of vector stores."""
         vector_stores = self.conn.beta.vector_stores.list(**kwargs)
         return vector_stores.data

diff --git a/tests/providers/openai/hooks/test_openai.py b/tests/providers/openai/hooks/test_openai.py
@@ -443,7 +443,7 @@ def test_get_vector_store(mock_openai_hook, mock_vector_store):
 
 def test_get_vector_stores(mock_openai_hook, mock_vector_store_list):
     mock_openai_hook.conn.beta.vector_stores.list.return_value = mock_vector_store_list
-    vector_stores = mock_openai_hook.get_vectors_stores()
+    vector_stores = mock_openai_hook.get_vector_stores()
     assert isinstance(vector_stores, list)