mirror of
https://github.com/LearningCircuit/local-deep-research.git
synced 2026-06-16 03:51:07 +03:00
c8dc94feb1
- Replace standard logging with loguru across codebase (50+ files) - Fix logger.exception usage in exception handlers (100+ fixes) - Update environment variable access to use SettingsManager - Improve pre-commit hooks to handle legitimate use cases - Fix formatting and linting issues Reduced custom code check violations from 268 to 38 (86% reduction) Reduced env var check violations from 73 to 57 (22% reduction)
94 lines
2.5 KiB
Python
94 lines
2.5 KiB
Python
# example_optimization.py - Quick Demo Version
|
|
"""
|
|
Full parameter optimization example for Local Deep Research.
|
|
|
|
This script demonstrates the full parameter optimization functionality.
|
|
|
|
Usage:
|
|
# Install dependencies with PDM
|
|
cd /path/to/local-deep-research
|
|
pdm install
|
|
|
|
# Run the script with PDM
|
|
pdm run python examples/optimization/example_optimization.py
|
|
"""
|
|
|
|
import json
|
|
import os
|
|
from datetime import datetime
|
|
|
|
|
|
# Import the optimization functionality
|
|
from local_deep_research.benchmarks.optimization import (
|
|
optimize_parameters,
|
|
)
|
|
|
|
# Loguru automatically handles logging configuration
|
|
|
|
|
|
def main():
|
|
# Create timestamp for unique output directory
|
|
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
|
output_dir = os.path.join(
|
|
"examples",
|
|
"optimization",
|
|
"results",
|
|
f"optimization_results_{timestamp}",
|
|
)
|
|
os.makedirs(output_dir, exist_ok=True)
|
|
|
|
print(
|
|
f"Starting quick optimization demo - results will be saved to {output_dir}"
|
|
)
|
|
|
|
# Demo with just a single simple optimization
|
|
print("\n=== Running quick demo optimization ===")
|
|
|
|
# Create a very simple parameter set to test
|
|
param_space = {
|
|
"iterations": {
|
|
"type": "int",
|
|
"low": 1,
|
|
"high": 2,
|
|
"step": 1,
|
|
},
|
|
"questions_per_iteration": {
|
|
"type": "int",
|
|
"low": 1,
|
|
"high": 2,
|
|
"step": 1,
|
|
},
|
|
"search_strategy": {
|
|
"type": "categorical",
|
|
"choices": ["rapid"], # Just use the fastest strategy
|
|
},
|
|
}
|
|
|
|
balanced_params, balanced_score = optimize_parameters(
|
|
query="SimpleQA quick demo", # Task descriptor
|
|
search_tool="searxng", # Using SearXNG
|
|
n_trials=2, # Just 2 trials for quick demo
|
|
output_dir=os.path.join(output_dir, "demo"),
|
|
param_space=param_space, # Limited parameter space
|
|
metric_weights={"quality": 0.5, "speed": 0.5},
|
|
)
|
|
|
|
print(f"Best parameters: {balanced_params}")
|
|
print(f"Best score: {balanced_score:.4f}")
|
|
|
|
# Save demo results to a summary file
|
|
summary = {
|
|
"timestamp": timestamp,
|
|
"demo": {"parameters": balanced_params, "score": balanced_score},
|
|
}
|
|
|
|
with open(os.path.join(output_dir, "optimization_summary.json"), "w") as f:
|
|
json.dump(summary, f, indent=2)
|
|
|
|
print(f"\nDemo complete! Results saved to {output_dir}")
|
|
print(f"Recommended parameters: {balanced_params}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|