Skip to content

Commit

Permalink
Apply suggestions from code review
Browse files Browse the repository at this point in the history
Co-authored-by: Wei Lee <weilee.rx@gmail.com>
  • Loading branch information
nathadfield and Lee-W committed Apr 26, 2024
1 parent a101b7b commit 67eb541
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions airflow/providers/openai/hooks/openai.py
Expand Up @@ -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.
Expand All @@ -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
Expand All @@ -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.
Expand Down

0 comments on commit 67eb541

Please sign in to comment.