From ba2da33345fe12b5c0e189a969732085cd617e0a Mon Sep 17 00:00:00 2001 From: mohammadrezapourreza Date: Fri, 10 May 2024 10:04:36 -0400 Subject: [PATCH] DH-5776/fixing black --- dataherald/services/sql_generations.py | 4 ++-- dataherald/sql_generator/__init__.py | 2 +- dataherald/sql_generator/dataherald_sqlagent.py | 4 ++-- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/dataherald/services/sql_generations.py b/dataherald/services/sql_generations.py index 2d9a8121..413101ca 100644 --- a/dataherald/services/sql_generations.py +++ b/dataherald/services/sql_generations.py @@ -63,9 +63,9 @@ def update_the_initial_sql_generation( initial_sql_generation.intermediate_steps = sql_generation.intermediate_steps return self.sql_generation_repository.update(initial_sql_generation) - def create( # noqa: PLR0912 + def create( # noqa: PLR0912 self, prompt_id: str, sql_generation_request: SQLGenerationRequest - ) -> SQLGeneration: # noqa: PLR0912 + ) -> SQLGeneration: # noqa: PLR0912 initial_sql_generation = SQLGeneration( prompt_id=prompt_id, created_at=datetime.now(), diff --git a/dataherald/sql_generator/__init__.py b/dataherald/sql_generator/__init__.py index f2bca9c1..6612332b 100644 --- a/dataherald/sql_generator/__init__.py +++ b/dataherald/sql_generator/__init__.py @@ -187,7 +187,7 @@ def stream_agent_steps( # noqa: PLR0912, C901 sql_generation_repository: SQLGenerationRepository, queue: Queue, metadata: dict = None, - ): # noqa: PLR0912 + ): # noqa: PLR0912 try: with get_openai_callback() as cb: for chunk in agent_executor.stream( diff --git a/dataherald/sql_generator/dataherald_sqlagent.py b/dataherald/sql_generator/dataherald_sqlagent.py index 6d1575f4..414ab089 100644 --- a/dataherald/sql_generator/dataherald_sqlagent.py +++ b/dataherald/sql_generator/dataherald_sqlagent.py @@ -710,13 +710,13 @@ def create_sql_agent( ) @override - def generate_response( # noqa: PLR0912 + def generate_response( # noqa: PLR0912 self, user_prompt: Prompt, database_connection: DatabaseConnection, context: List[dict] = None, metadata: dict = None, - ) -> SQLGeneration: # noqa: PLR0912 + ) -> SQLGeneration: # noqa: PLR0912 context_store = self.system.instance(ContextStore) storage = self.system.instance(DB) response = SQLGeneration(