diff --git a/airflow/providers/openai/hooks/openai.py b/airflow/providers/openai/hooks/openai.py index 5e73ac12fb2db..8d6decf1371e1 100644 --- a/airflow/providers/openai/hooks/openai.py +++ b/airflow/providers/openai/hooks/openai.py @@ -331,7 +331,7 @@ def get_file_by_name(self, file_name: str) -> FileObject | None: files = self.get_files() return next((file for file in files if file.filename == file_name), None) - def delete_file(self, file_id) -> FileDeleted: + def delete_file(self, file_id: str) -> FileDeleted: """ Delete a file. @@ -340,12 +340,12 @@ def delete_file(self, file_id) -> FileDeleted: response = self.conn.files.delete(file_id=file_id) return response - def create_vector_store(self, **kwargs) -> VectorStore: + def create_vector_store(self, **kwargs: Any) -> VectorStore: """Create a vector store.""" vector_store = self.conn.beta.vector_stores.create(**kwargs) return vector_store - def get_vectors_stores(self, **kwargs) -> list[VectorStore]: + def get_vectors_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 @@ -370,7 +370,7 @@ def get_vector_store_by_name(self, vector_store_name: str) -> VectorStore | None (vector_store for vector_store in vector_stores if vector_store.name == vector_store_name), None ) - def modify_vector_store(self, vector_store_id: str, **kwargs) -> VectorStore: + def modify_vector_store(self, vector_store_id: str, **kwargs: Any) -> VectorStore: """ Modify a vector store.