akadmin/headroom
1
0
Fork 0
mirror of https://github.com/headroomlabs-ai/headroom.git synced 2026-08-27 14:17:10 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Format prefix_cache_benchmark.py for ruff 0.15 compatibility

Browse source
This commit is contained in:
chopratejas 2026-03-12 22:03:38 -07:00
parent d2e88d362a
commit 1474355f1c
1 changed files with 8 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

14
benchmarks/prefix_cache_benchmark.py
View file

@ -225,12 +225,14 @@ TOOL_RESPONSES = {
for i, f in enumerate(["auth", "api", "cache"])
]
),
"read_file": lambda q: "# File contents (simulated)\nimport logging\nfrom typing import Optional\n\nlogger = logging.getLogger(__name__)\n\n"
+ "\n".join(
[
f"def function_{i}(arg: str) -> Optional[dict]:\n \"\"\"Process {q}.\"\"\"\n result = {{}}\n for key in ['id', 'name', 'status']:\n result[key] = f'value_{{key}}_{{arg}}'\n logger.info(f'Processed {{arg}}')\n return result\n"
for i in range(8)
]
"read_file": lambda q: (
"# File contents (simulated)\nimport logging\nfrom typing import Optional\n\nlogger = logging.getLogger(__name__)\n\n"
+ "\n".join(
[
f"def function_{i}(arg: str) -> Optional[dict]:\n \"\"\"Process {q}.\"\"\"\n result = {{}}\n for key in ['id', 'name', 'status']:\n result[key] = f'value_{{key}}_{{arg}}'\n logger.info(f'Processed {{arg}}')\n return result\n"
for i in range(8)
]
)
),
"query_database": lambda q: json.dumps(
[