administrator/poc-v1
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: Rename metadata_repo to metadata_store for consistency across tools and remove compatibility modules

Browse source
This commit is contained in:
william.dias 2026-01-23 09:55:58 -03:00
parent 5719fdbe18
commit a7afdfac8b
7 changed files with 17 additions and 54 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/sql_optimizer_team/tools/core_tools.py
View file

@ -122,14 +122,14 @@ async def optimize_query_core(
llm = LLMToolFactory.create(settings)
storage = LocalFileStore()
metadata_repo = JsonMetadataStore(settings.storage.metadata_file)
metadata_store = JsonMetadataStore(settings.storage.metadata_file)
generator = PromptGeneratorFactory.create(db_type)
sql_file = _write_sql(sql, output_dir=output_dir, filename_prefix=f"opt_{db_type.value}")
flow = OptimizeQueryFlow(
llm_tool=llm,
storage_tool=storage,
metadata_store=metadata_repo,
metadata_store=metadata_store,
prompt_generator=generator,
)
@ -166,13 +166,13 @@ async def compare_optimizations_core(
llm = LLMToolFactory.create(settings)
storage = LocalFileStore()
metadata_repo = JsonMetadataStore(settings.storage.metadata_file)
metadata_store = JsonMetadataStore(settings.storage.metadata_file)
sql_file = _write_sql(sql, output_dir=output_dir, filename_prefix="compare")
flow = CompareOptimizationsFlow(
llm_tool=llm,
storage_tool=storage,
metadata_store=metadata_repo,
metadata_store=metadata_store,
prompt_generator_factory=PromptGeneratorFactory,
)

4
src/sql_optimizer_team/tools/engine/flows/collect_and_optimize_worst_query.py
View file

@ -47,7 +47,7 @@ class CollectAndOptimizeWorstQueryFlow:
) -> None:
self._llm = llm_tool
self._storage_tool = storage_tool
self._metadata_repo = metadata_store
self._metadata_store = metadata_store
self._prompt_generator = prompt_generator
def _fetch_worst(
@ -216,7 +216,7 @@ class CollectAndOptimizeWorstQueryFlow:
output = await OptimizeQueryFlow(
llm_tool=self._llm,
storage_tool=self._storage_tool,
metadata_store=self._metadata_repo,
metadata_store=self._metadata_store,
prompt_generator=self._prompt_generator,
).execute(
sql_file=worst_query_file,

6
src/sql_optimizer_team/tools/engine/flows/compare_optimizations.py
View file

@ -27,7 +27,7 @@ class CompareOptimizationsFlow:
Attributes:
_llm: LLM tool
_storage: File storage tool
_metadata_repo: Metadata store tool
_metadata_store: Metadata store tool
_prompt_generator_factory: Factory for creating prompt generators
Examples:
@ -60,7 +60,7 @@ class CompareOptimizationsFlow:
"""
self._llm = llm_tool
self._storage = storage_tool
self._metadata_repo = metadata_store
self._metadata_store = metadata_store
self._prompt_factory = prompt_generator_factory
async def execute(
@ -151,7 +151,7 @@ class CompareOptimizationsFlow:
return await OptimizeQueryFlow(
llm_tool=self._llm,
storage_tool=self._storage,
metadata_store=self._metadata_repo,
metadata_store=self._metadata_store,
prompt_generator=self._prompt_factory.create(database_type),
).execute(
sql_file,

8
src/sql_optimizer_team/tools/engine/flows/optimize_query.py
View file

@ -30,7 +30,7 @@ class OptimizeQueryFlow:
Attributes:
_llm: LLM tool
_storage: File storage tool
_metadata_repo: Metadata store tool
_metadata_store: Metadata store tool
_prompt_generator: Prompt generator tool for target database
Examples:
@ -63,7 +63,7 @@ class OptimizeQueryFlow:
"""
self._llm = llm_tool
self._storage = storage_tool
self._metadata_repo = metadata_store
self._metadata_store = metadata_store
self._prompt_generator = prompt_generator
async def execute(
@ -144,7 +144,7 @@ class OptimizeQueryFlow:
pair_review = None
# Persist metadata and output file and return
await self._metadata_repo.save(metadata)
await self._metadata_store.save(metadata)
await self._save_output_file(sql_file, result, pair_review=pair_review, source_object=source_object)
return OptimizationOutput.from_result(result, source_object=source_object)
@ -198,7 +198,7 @@ class OptimizeQueryFlow:
Returns:
Existing or new metadata instance
"""
if existing := await self._metadata_repo.get(query_hash):
if existing := await self._metadata_store.get(query_hash):
return existing
return OptimizationMetadata.create_new(query, database_type)

11
src/sql_optimizer_team/tools/engine/outputs/optimization_dto.py
View file

@ -1,11 +0,0 @@
"""Compatibility module for output objects.
Keep this shim to avoid import breakages while exposing the new output types.
"""
from sql_optimizer_team.tools.engine.outputs.optimization_output import (
ComparisonOutput,
OptimizationOutput,
)
__all__ = ["OptimizationOutput", "ComparisonOutput"]

8
src/sql_optimizer_team/tools/engine/storage_tools/sqlserver_stats_tool.py
View file

@ -84,13 +84,13 @@ class SQLServerStatsTool:
List of dicts containing performance metrics and SQL text.
Examples:
>>> repo = SQLServerStatsTool(instance="sis_vida_prd")
>>> tool = SQLServerStatsTool(instance="sis_vida_prd")
>>> # Get queries with highest CPU consumption
>>> queries = repo.fetch_worst_queries(limit=10, metric=QueryMetric.CPU)
>>> queries = tool.fetch_worst_queries(limit=10, metric=QueryMetric.CPU)
>>> # Get queries executed at least 100 times
>>> queries = repo.fetch_worst_queries(limit=10, metric=QueryMetric.CPU, min_executions=100)
>>> queries = tool.fetch_worst_queries(limit=10, metric=QueryMetric.CPU, min_executions=100)
>>> # Order by most executed first, then by CPU
>>> queries = repo.fetch_worst_queries(limit=10, metric=QueryMetric.CPU, order_by_executions=True)
>>> queries = tool.fetch_worst_queries(limit=10, metric=QueryMetric.CPU, order_by_executions=True)
"""
# Ensure safety and avoid SQL injection
try:

26
src/sql_optimizer_team/tools/engine/types/domain_exceptions.py
View file

@ -1,26 +0,0 @@
"""Compatibility module for tool exceptions.
Keep this shim to avoid import breakages while exposing the new exceptions.
"""
from sql_optimizer_team.tools.engine.types.tool_exceptions import (
ToolError,
InvalidDatabaseTypeError,
InvalidQueryHashError,
InvalidVersionError,
LLMProviderError,
MetadataNotFoundError,
OptimizationError,
QueryValidationError,
)
__all__ = [
"ToolError",
"QueryValidationError",
"LLMProviderError",
"MetadataNotFoundError",
"InvalidDatabaseTypeError",
"InvalidQueryHashError",
"InvalidVersionError",
"OptimizationError",
]