mirror of
https://github.com/stablyai/orca.git
synced 2026-10-07 16:02:29 +00:00
Neilandunknown
4dec64d533
* feat(source-control-ai): support OMP text generation Read prompts on stdin, retain OMP configured model by default, and reuse JSON model discovery. Co-authored-by: unknown <1784931579@qq.com> * test(source-control-ai): cover OMP large input and model overrides * fix(omp): keep configured model default out of discovered catalog * fix(omp): hide config default from model discovery catalog * fix(omp): separate terminal discovery from generation defaults * test(omp): keep model probe import compatible with CLI typecheck * test: align Source Control AI registry contracts with OMP --------- Co-authored-by: unknown <1784931579@qq.com>
60 lines
2.5 KiB
TypeScript
60 lines
2.5 KiB
TypeScript
import { describe, expect, it } from 'vitest'
|
|
import { getAgentModelProbeSpec } from './agent-model-probe-spec'
|
|
import { GROK_SESSION_OPTION_CATALOG } from './agent-session-option-catalog-grok'
|
|
import { getCommitMessageAgentSpec, listCommitMessageAgentIds } from './commit-message-agent-spec'
|
|
|
|
describe('getAgentModelProbeSpec', () => {
|
|
it('resolves grok as a discovery-only probe with no static model list', () => {
|
|
const spec = getAgentModelProbeSpec('grok')
|
|
expect(spec).toMatchObject({
|
|
id: 'grok',
|
|
binary: 'grok',
|
|
modelSource: 'dynamic',
|
|
models: [],
|
|
defaultModelId: 'grok-4.6'
|
|
})
|
|
expect(spec?.modelDiscovery).toMatchObject({ binary: 'grok', args: ['models'] })
|
|
})
|
|
|
|
it('wires that probe to the parser that reads a real listing', () => {
|
|
// Pinning binary and args alone would still pass with the wrong parser attached,
|
|
// leaving discovery to publish nothing on every host.
|
|
const parsed = getAgentModelProbeSpec('grok')!.modelDiscovery!.parse(
|
|
'You are logged in with grok.com.\n\nDefault model: grok-4.5\n\nAvailable models:\n * grok-4.5 (default)\n'
|
|
)
|
|
expect(parsed).toEqual([{ id: 'grok-4.5', label: 'Grok 4.5', isDefault: true }])
|
|
})
|
|
|
|
it('keeps the grok probe default in step with the catalog seed', () => {
|
|
expect(getAgentModelProbeSpec('grok')!.defaultModelId).toBe(
|
|
GROK_SESSION_OPTION_CATALOG.models[0].id
|
|
)
|
|
})
|
|
|
|
it('aliases agents without a generation-only default by identity', () => {
|
|
// A lossy adapter here would silently drop fields like
|
|
// `modelDiscovery.stdinPayload` and break Claude discovery.
|
|
for (const id of listCommitMessageAgentIds().filter((agentId) => agentId !== 'omp')) {
|
|
expect(getAgentModelProbeSpec(id)).toBe(getCommitMessageAgentSpec(id))
|
|
}
|
|
})
|
|
|
|
it('removes only the OMP generation sentinel while preserving discovery', () => {
|
|
const generation = getCommitMessageAgentSpec('omp')!
|
|
const probe = getAgentModelProbeSpec('omp')!
|
|
expect(generation.defaultModelId).toBe('default')
|
|
expect(probe.defaultModelId).toBe('')
|
|
expect(probe.models).toEqual([])
|
|
expect(probe.modelDiscovery).toBe(generation.modelDiscovery)
|
|
})
|
|
|
|
it('keeps grok out of the commit-message registry', () => {
|
|
expect(getCommitMessageAgentSpec('grok')).toBeUndefined()
|
|
expect(listCommitMessageAgentIds()).not.toContain('grok')
|
|
})
|
|
|
|
it('is undefined for an agent in neither registry', () => {
|
|
expect(getAgentModelProbeSpec('aider')).toBeUndefined()
|
|
})
|
|
})
|