brainiac 0.1.8 → 0.1.9
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/lib/brainiac/helpers.rb +83 -19
- data/lib/brainiac/version.rb +1 -1
- metadata +1 -1
checksums.yaml
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
---
|
|
2
2
|
SHA256:
|
|
3
|
-
metadata.gz:
|
|
4
|
-
data.tar.gz:
|
|
3
|
+
metadata.gz: cedcfe4e4dfa26034cf0ded9746d8ff5bade886ec3d83774580d8334777d4d3c
|
|
4
|
+
data.tar.gz: 96d1702d994f06f0a4f41870de6652258fc700a9f9df1b6193c347b2cce3b365
|
|
5
5
|
SHA512:
|
|
6
|
-
metadata.gz:
|
|
7
|
-
data.tar.gz:
|
|
6
|
+
metadata.gz: 16a5c9937e67afe78bcba468dea8a4b75c73864c4a320b8d2da930a966bd618797b6285f765f772a99394d8925ba9438fe2b9012ccb94730908e97642148cdbb
|
|
7
|
+
data.tar.gz: 95cd9275779731da352b0356e1da275f12511b960037512e829614a6b5e2a4a97e3b00ba0ceeefbdbcc8eff71a3f0717b9cc43fb3ee4d21b2b706361e4dfccc0
|
data/lib/brainiac/helpers.rb
CHANGED
|
@@ -107,9 +107,8 @@ end
|
|
|
107
107
|
# Returns the provider name (e.g. "grok") or nil.
|
|
108
108
|
def detect_cli_provider(text: "", tags: [])
|
|
109
109
|
# Inline tag: [cli:grok] — works in any channel
|
|
110
|
-
|
|
111
|
-
|
|
112
|
-
end
|
|
110
|
+
inline = inline_cli_provider_tag(text)
|
|
111
|
+
return inline if inline
|
|
113
112
|
|
|
114
113
|
# Plugin hook: let plugins detect from their own metadata (e.g., card tags)
|
|
115
114
|
results = Brainiac.emit(:detect_cli_provider, text: text, tags: tags)
|
|
@@ -119,6 +118,11 @@ def detect_cli_provider(text: "", tags: [])
|
|
|
119
118
|
nil
|
|
120
119
|
end
|
|
121
120
|
|
|
121
|
+
# The provider named by an inline [cli:X] tag in text, or nil. Text only — no plugin hooks.
|
|
122
|
+
def inline_cli_provider_tag(text)
|
|
123
|
+
text.to_s.match(/\[cli:(\w+)\]/i)&.[](1)&.downcase
|
|
124
|
+
end
|
|
125
|
+
|
|
122
126
|
def default_project_key
|
|
123
127
|
# Find the project marked as default
|
|
124
128
|
default = PROJECTS.find { |_key, config| config["default"] == true }
|
|
@@ -354,9 +358,11 @@ def register_work_item(branch:, worktree: nil, project: nil, agent: nil, source:
|
|
|
354
358
|
end
|
|
355
359
|
|
|
356
360
|
# Update dispatch overrides on a work item (cli_provider, model, effort, profile).
|
|
357
|
-
#
|
|
361
|
+
# Non-nil values are stored; nil values leave the existing override untouched.
|
|
362
|
+
# Keys listed in clear: (e.g. [:model]) are removed from the stored overrides.
|
|
358
363
|
# Returns true if the work item was found and updated, false otherwise.
|
|
359
|
-
def update_work_item_overrides(cli_provider: nil, model: nil, effort: nil, profile: nil, work_item_id: nil, branch: nil
|
|
364
|
+
def update_work_item_overrides(cli_provider: nil, model: nil, effort: nil, profile: nil, work_item_id: nil, branch: nil, # rubocop:disable Naming/PredicateMethod
|
|
365
|
+
clear: [])
|
|
360
366
|
map = load_work_item_map
|
|
361
367
|
|
|
362
368
|
target_id = work_item_id
|
|
@@ -376,6 +382,7 @@ def update_work_item_overrides(cli_provider: nil, model: nil, effort: nil, profi
|
|
|
376
382
|
overrides["model"] = model if model
|
|
377
383
|
overrides["effort"] = effort if effort
|
|
378
384
|
overrides["profile"] = profile if profile
|
|
385
|
+
Array(clear).each { |key| overrides.delete(key.to_s) }
|
|
379
386
|
overrides.compact!
|
|
380
387
|
|
|
381
388
|
if overrides.empty?
|
|
@@ -425,33 +432,78 @@ end
|
|
|
425
432
|
# Unknown profile names are never persisted, so a typo can't pin a work item to a
|
|
426
433
|
# profile that doesn't exist (profile_spawn_env ignores it for this dispatch too).
|
|
427
434
|
#
|
|
435
|
+
# A stored model is a provider-specific ID (kiro's "opus" is "claude-opus-4.5", claude's is
|
|
436
|
+
# "opus"), so switching providers with [cli:X] drops it and the new provider's default
|
|
437
|
+
# applies, unless the same message also carries a model tag. default_cli_provider: is the
|
|
438
|
+
# provider a work item with no stored cli_provider was running on; without it, any inline
|
|
439
|
+
# [cli:X] on such a work item counts as a switch.
|
|
440
|
+
#
|
|
428
441
|
# Returns: { cli_provider:, model:, effort:, profile: } (each String or nil)
|
|
429
442
|
def resolve_work_item_overrides(work_item_id: nil, branch: nil, inline_cli_provider: nil, inline_model: nil, inline_effort: nil,
|
|
430
|
-
inline_profile: nil)
|
|
443
|
+
inline_profile: nil, default_cli_provider: nil)
|
|
431
444
|
stored = work_item_overrides_for(work_item_id: work_item_id, branch: branch)
|
|
432
445
|
inline_profile = inline_profile&.to_s&.downcase
|
|
433
446
|
persist_profile = inline_profile && profile_entry(inline_profile) ? inline_profile : nil
|
|
447
|
+
switched = cli_provider_switched?(inline_cli_provider, stored["cli_provider"] || default_cli_provider)
|
|
448
|
+
stored_model = switched ? nil : stored["model"]
|
|
434
449
|
|
|
435
450
|
# Inline tags override stored values
|
|
436
451
|
resolved = {
|
|
437
452
|
cli_provider: inline_cli_provider || stored["cli_provider"],
|
|
438
|
-
model: inline_model ||
|
|
453
|
+
model: inline_model || stored_model,
|
|
439
454
|
effort: inline_effort || stored["effort"],
|
|
440
455
|
profile: inline_profile || stored["profile"]
|
|
441
456
|
}
|
|
442
457
|
|
|
443
458
|
# Persist any new inline overrides to the work item for future dispatches
|
|
444
|
-
new_overrides = {
|
|
445
|
-
|
|
446
|
-
new_overrides[:
|
|
447
|
-
new_overrides[:effort] = inline_effort if inline_effort
|
|
448
|
-
new_overrides[:profile] = persist_profile if persist_profile
|
|
459
|
+
new_overrides = { cli_provider: inline_cli_provider, model: inline_model, effort: inline_effort,
|
|
460
|
+
profile: persist_profile }.compact
|
|
461
|
+
new_overrides[:clear] = [:model] if switched && !inline_model && stored["model"]
|
|
449
462
|
|
|
450
463
|
update_work_item_overrides(work_item_id: work_item_id, branch: branch, **new_overrides) if new_overrides.any? && (work_item_id || branch)
|
|
451
464
|
|
|
452
465
|
resolved
|
|
453
466
|
end
|
|
454
467
|
|
|
468
|
+
# True when an inline [cli:X] names a different provider than the one previously in use.
|
|
469
|
+
def cli_provider_switched?(inline_cli_provider, previous_cli_provider)
|
|
470
|
+
!inline_cli_provider.nil? && inline_cli_provider != previous_cli_provider
|
|
471
|
+
end
|
|
472
|
+
|
|
473
|
+
# One-stop override resolution for a dispatch: parses [cli:X]/[model]/[effort:X]/[p:X] from
|
|
474
|
+
# the message text (and plugin card tags), merges them with the work item's stored overrides,
|
|
475
|
+
# and persists any new inline values. The model and effort are resolved against the provider
|
|
476
|
+
# that will actually run (inline > stored > agent/project default), so a [sonnet] tag can't be
|
|
477
|
+
# looked up in the wrong provider's model map. Plugins should prefer this over calling
|
|
478
|
+
# detect_model/detect_effort and resolve_work_item_overrides by hand.
|
|
479
|
+
#
|
|
480
|
+
# Returns: { cli_provider:, model:, explicit_model:, effort:, profile: }
|
|
481
|
+
# cli_provider is nil when nothing overrides the agent/project default.
|
|
482
|
+
# explicit_model is true when the model came from a tag (see detect_model_explicit).
|
|
483
|
+
def resolve_dispatch_overrides(project_config, text: "", tags: [], agent_name: nil, work_item_id: nil, branch: nil)
|
|
484
|
+
stored = work_item_overrides_for(work_item_id: work_item_id, branch: branch)
|
|
485
|
+
inline_cli = detect_cli_provider(text: text, tags: tags)
|
|
486
|
+
default_cli = agent_cli_provider_for(agent_name) || project_config["cli_provider"]
|
|
487
|
+
effective_cli = inline_cli || stored["cli_provider"]
|
|
488
|
+
|
|
489
|
+
model, explicit_model = detect_model_explicit(project_config, tags: tags, text: text,
|
|
490
|
+
cli_provider_override: effective_cli, agent_name: agent_name)
|
|
491
|
+
effort, explicit_effort = detect_effort_explicit(project_config, tags: tags, text: text,
|
|
492
|
+
cli_provider_override: effective_cli, agent_name: agent_name)
|
|
493
|
+
|
|
494
|
+
resolved = resolve_work_item_overrides(work_item_id: work_item_id, branch: branch, inline_cli_provider: inline_cli,
|
|
495
|
+
inline_model: explicit_model ? model : nil,
|
|
496
|
+
inline_effort: explicit_effort ? effort : nil,
|
|
497
|
+
inline_profile: detect_profile(text: text, tags: tags),
|
|
498
|
+
default_cli_provider: default_cli)
|
|
499
|
+
# Nothing tagged or stored (or the stored model was dropped by a switch): use the
|
|
500
|
+
# effective provider's defaults, already resolved above.
|
|
501
|
+
resolved[:model] ||= model
|
|
502
|
+
resolved[:effort] ||= effort
|
|
503
|
+
resolved[:explicit_model] = explicit_model
|
|
504
|
+
resolved
|
|
505
|
+
end
|
|
506
|
+
|
|
455
507
|
# Add or update a source on an existing work item.
|
|
456
508
|
# Returns true if the work item was found and updated, false otherwise.
|
|
457
509
|
def register_work_item_source(source:, source_data:, work_item_id: nil, branch: nil) # rubocop:disable Naming/PredicateMethod
|
|
@@ -984,7 +1036,11 @@ def append_model_to_cmd(cmd, model, resolved)
|
|
|
984
1036
|
# This handles cases where different projects use "auto" but each CLI provider maps it differently.
|
|
985
1037
|
effective_model = allowed.key?(model) ? allowed[model] : model
|
|
986
1038
|
is_known = allowed.value?(effective_model) || allowed.key?(effective_model)
|
|
987
|
-
|
|
1039
|
+
if is_known
|
|
1040
|
+
cmd.push(resolved["agent_model_flag"], effective_model)
|
|
1041
|
+
else
|
|
1042
|
+
LOG.warn "[Model] Dropping model '#{model}': not in #{resolved["agent_cli"]}'s model map; using the CLI's default"
|
|
1043
|
+
end
|
|
988
1044
|
end
|
|
989
1045
|
|
|
990
1046
|
# A model is only "explicit" if the caller actually supplied one (i.e. it came from a
|
|
@@ -1344,8 +1400,9 @@ end
|
|
|
1344
1400
|
# while a fell-back-to-default value must NOT write — otherwise every no-tag dispatch
|
|
1345
1401
|
# would clobber each account's per-account default back to the project default.
|
|
1346
1402
|
def detect_model_explicit(project_config, tags: [], text: "", cli_provider_override: nil, agent_name: nil)
|
|
1347
|
-
#
|
|
1348
|
-
|
|
1403
|
+
# Without an explicit override, a [cli:X] in the same text picks the model map (so
|
|
1404
|
+
# "[cli:claude] [sonnet]" resolves through claude's map), then the agent's provider.
|
|
1405
|
+
effective_cli_provider = cli_provider_override || inline_cli_provider_tag(text) || agent_cli_provider_for(agent_name)
|
|
1349
1406
|
resolved = resolve_project_cli_config(project_config, cli_provider_override: effective_cli_provider, agent_name: agent_name)
|
|
1350
1407
|
allowed_models = resolved["allowed_models"] || {}
|
|
1351
1408
|
return [resolved["agent_model"], false] if allowed_models.empty?
|
|
@@ -1368,22 +1425,29 @@ end
|
|
|
1368
1425
|
# If the requested level isn't supported by the current model, returns the closest
|
|
1369
1426
|
# lower level from allowed_efforts.
|
|
1370
1427
|
def detect_effort(project_config, tags: [], text: "", cli_provider_override: nil, agent_name: nil)
|
|
1371
|
-
|
|
1428
|
+
detect_effort_explicit(project_config, tags: tags, text: text,
|
|
1429
|
+
cli_provider_override: cli_provider_override, agent_name: agent_name).first
|
|
1430
|
+
end
|
|
1431
|
+
|
|
1432
|
+
# Like detect_effort, but returns [level, explicit_bool] where explicit is true only when
|
|
1433
|
+
# the level came from an [effort:X] tag or a plugin's card tag (not the provider default).
|
|
1434
|
+
def detect_effort_explicit(project_config, tags: [], text: "", cli_provider_override: nil, agent_name: nil)
|
|
1435
|
+
effective_cli_provider = cli_provider_override || inline_cli_provider_tag(text) || agent_cli_provider_for(agent_name)
|
|
1372
1436
|
resolved = resolve_project_cli_config(project_config, cli_provider_override: effective_cli_provider, agent_name: agent_name)
|
|
1373
1437
|
allowed = resolved["allowed_efforts"] || %w[low medium high xhigh max]
|
|
1374
1438
|
|
|
1375
1439
|
# Inline tag: [effort:high] — works in any channel
|
|
1376
1440
|
if (match = text.match(/\[effort:(\w+)\]/i))
|
|
1377
1441
|
level = match[1].downcase
|
|
1378
|
-
return resolve_effort_level(level, allowed) if allowed.include?(level)
|
|
1442
|
+
return [resolve_effort_level(level, allowed), true] if allowed.include?(level)
|
|
1379
1443
|
end
|
|
1380
1444
|
|
|
1381
1445
|
# Plugin hook: let plugins detect from their own metadata (e.g., card tags)
|
|
1382
1446
|
results = Brainiac.emit(:detect_effort, tags: tags, allowed: allowed)
|
|
1383
1447
|
plugin_result = results.compact.first
|
|
1384
|
-
return resolve_effort_level(plugin_result, allowed) if plugin_result
|
|
1448
|
+
return [resolve_effort_level(plugin_result, allowed), true] if plugin_result
|
|
1385
1449
|
|
|
1386
|
-
resolved["agent_effort"]
|
|
1450
|
+
[resolved["agent_effort"], false]
|
|
1387
1451
|
end
|
|
1388
1452
|
|
|
1389
1453
|
# Detect a profile ([profile:X]/[p:X]) from inline text OR card tags.
|
data/lib/brainiac/version.rb
CHANGED