Skip to content

Commit

Permalink
Rename class to fetch search terms and fix its docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
augusto-herrmann committed Aug 23, 2024
1 parent 7d91d1c commit 8b42676
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions src/schemas.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,8 +26,9 @@ class DBSelect(BaseModel):
conn_id: str = Field(description="Airflow connection ID to use for the SQL query")


class SearchTerms(BaseModel):
"""Represents the search terms in the YAML file."""
class FetchTermsConfig(BaseModel):
"""Represents configuration information for fetching search terms from
a data source."""

from_airflow_variable: Optional[str] = Field(
default=None,
Expand Down Expand Up @@ -63,7 +64,7 @@ class SearchConfig(BaseModel):
description="ID do território no Querido Diário para filtragem "
"baseada em localização",
)
terms: Union[List[str], SearchTerms] = Field(
terms: Union[List[str], FetchTermsConfig] = Field(
description="Lista de termos de pesquisa ou uma forma de buscá-los"
)
field: Optional[str] = Field(
Expand Down

0 comments on commit 8b42676

Please sign in to comment.