Skip to content

Commit

Permalink
fix linters
Browse files Browse the repository at this point in the history
  • Loading branch information
micpst committed Sep 23, 2024
1 parent c1c871b commit 8eefd9b
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 3 deletions.
3 changes: 2 additions & 1 deletion src/dbally/iql/_type_validators.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
from dataclasses import dataclass
from typing import ( # type: ignore
Annotated,
Any,
Callable,
Dict,
Expand All @@ -13,6 +12,8 @@
get_origin,
)

from typing_extensions import Annotated


@dataclass
class _ValidationResult:
Expand Down
4 changes: 2 additions & 2 deletions src/dbally/views/freeform/text2sql/view.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,11 +100,11 @@ async def ask(
self,
query: str,
llm: LLM,
contexts: Optional[List[BaseCallerContext]] = None,
event_tracker: Optional[EventTracker] = None,
n_retries: int = 3,
dry_run: bool = False,
llm_options: Optional[LLMOptions] = None,
contexts: Optional[Iterable[BaseCallerContext]] = None,
) -> ViewExecutionResult:
"""
Executes the query and returns the result. It generates the SQL query from the natural language query and
Expand All @@ -113,11 +113,11 @@ async def ask(
Args:
query: The natural language query to execute.
llm: The LLM used to execute the query.
contexts: Currently not used.
event_tracker: The event tracker used to audit the query execution.
n_retries: The number of retries to execute the query in case of errors.
dry_run: If True, the query will not be used to fetch data from the datasource.
llm_options: Options to use for the LLM.
contexts: Currently not used.
Returns:
The result of the query.
Expand Down

0 comments on commit 8eefd9b

Please sign in to comment.