can1357
7eeaba0471
- Extracted internal handlers and logic from AgentSession into dedicated runner, guard, and coordinator modules. - Created standalone modules for bash execution, evaluation runners, IRC bridging, and prewalk coordination. - Established dedicated session components for tracking stats, todos, streams, and retry fallback chains. - Preserved existing session behavior while significantly reducing monolithic class size and complexity.
86 lines
3.3 KiB
TypeScript
86 lines
3.3 KiB
TypeScript
import type { ThinkingLevel } from "@oh-my-pi/pi-agent-core";
|
|
import type { Model } from "@oh-my-pi/pi-ai";
|
|
import type { ModelRegistry } from "../config/model-registry";
|
|
import {
|
|
extractExplicitThinkingSelector,
|
|
formatModelSelectorValue,
|
|
getModelMatchPreferences,
|
|
parseModelString,
|
|
type ResolvedModelRoleValue,
|
|
resolveModelRoleValue,
|
|
} from "../config/model-resolver";
|
|
import type { Settings } from "../config/settings";
|
|
|
|
/** Formats a role assignment while preserving its explicit thinking selector. */
|
|
export function formatRoleModelValue(
|
|
settings: Settings,
|
|
modelRegistry: ModelRegistry,
|
|
role: string,
|
|
model: Model,
|
|
selectorOverride?: string,
|
|
thinkingLevelOverride?: ThinkingLevel,
|
|
): string {
|
|
const modelKey = selectorOverride ?? `${model.provider}/${model.id}`;
|
|
if (thinkingLevelOverride !== undefined) return formatModelSelectorValue(modelKey, thinkingLevelOverride);
|
|
const existingRoleValue = settings.getModelRole(role);
|
|
if (!existingRoleValue) return modelKey;
|
|
const thinkingLevel = extractExplicitThinkingSelector(existingRoleValue, settings, {
|
|
isLiteralModelId: (provider, id) => modelRegistry.find(provider, id) !== undefined,
|
|
});
|
|
return formatModelSelectorValue(modelKey, thinkingLevel);
|
|
}
|
|
|
|
/** Resolves a configured model target relative to the current provider. */
|
|
export function resolveConfiguredModelTarget(
|
|
configuredTarget: string | undefined,
|
|
currentModel: Model,
|
|
availableModels: Model[],
|
|
): Model | undefined {
|
|
const trimmedTarget = configuredTarget?.trim();
|
|
if (!trimmedTarget) return undefined;
|
|
const parsed = parseModelString(trimmedTarget, {
|
|
allowMaxSuffix: true,
|
|
allowAutoAlias: true,
|
|
isLiteralModelId: (provider, id) => availableModels.some(model => model.provider === provider && model.id === id),
|
|
});
|
|
if (parsed) {
|
|
const explicitModel = availableModels.find(model => model.provider === parsed.provider && model.id === parsed.id);
|
|
if (explicitModel) return explicitModel;
|
|
}
|
|
return availableModels.find(model => model.provider === currentModel.provider && model.id === trimmedTarget);
|
|
}
|
|
|
|
/** Resolves a model's configured context-promotion target. */
|
|
export function resolveContextPromotionConfiguredTarget(
|
|
currentModel: Model,
|
|
availableModels: Model[],
|
|
): Model | undefined {
|
|
return resolveConfiguredModelTarget(currentModel.contextPromotionTarget, currentModel, availableModels);
|
|
}
|
|
|
|
/** Resolves a model's configured compaction target. */
|
|
export function resolveCompactionConfiguredTarget(currentModel: Model, availableModels: Model[]): Model | undefined {
|
|
return resolveConfiguredModelTarget(currentModel.compactionModel, currentModel, availableModels);
|
|
}
|
|
|
|
/** Resolves a model role and its explicit thinking selection. */
|
|
export function resolveRoleModelFull(
|
|
settings: Settings,
|
|
role: string,
|
|
availableModels: Model[],
|
|
currentModel: Model | undefined,
|
|
): ResolvedModelRoleValue {
|
|
const roleModelStr =
|
|
role === "default"
|
|
? (settings.getModelRole("default") ??
|
|
(currentModel ? `${currentModel.provider}/${currentModel.id}` : undefined))
|
|
: settings.getModelRole(role);
|
|
if (!roleModelStr) {
|
|
return { model: undefined, thinkingLevel: undefined, explicitThinkingLevel: false, warning: undefined };
|
|
}
|
|
return resolveModelRoleValue(roleModelStr, availableModels, {
|
|
settings,
|
|
matchPreferences: getModelMatchPreferences(settings),
|
|
});
|
|
}
|