Package com.tnsai.agents

Class AgentCognitiveSupport

java.lang.Object

com.tnsai.agents.AgentCognitiveSupport

public final class AgentCognitiveSupport
extends Object

Manages planning, reasoning, evaluation, environment perception, and feedback collection for an Agent.

This class extracts cognitive support concerns from the main Agent class to reduce its size and improve cohesion. It holds:

• Planning — PlannerHandle for GOAP/HTN planning
• Reasoning — ReasoningStrategyHandle for ReAct, ToT, etc.
• Evaluation — EvalHandle + EvalContext for metric hooks
• Environment — perception and action on environments
• Feedback — FeedbackCollector for learned preferences

Used by Agent and AgentCapabilities internally.

Method Summary

Modifier and Type	Method	Description
ActionOutcome	actOnEnvironment(String actionName, Map<String,Object> parameters, String agentClassName)
void	addEvalHook(EvalHook hook)
EvalContext	clearEvalContext()
void	ensureEvalContext()
PlannerHandle.PlanResult	executePlan(List<PlannerHandle.PlanStep> steps, List<Role> roles)
Map<String,Object>	extractCurrentState(List<Role> roles)
void	fireOnAfterChat(String result, long latencyMs)
void	fireOnAfterToolCall(String toolName, Object result, boolean success, long latencyMs)
void	fireOnAgentStop(String status)
void	fireOnBeforeChat(String message)
void	fireOnBeforeToolCall(String toolName, Map<String,Object> arguments)
void	fireOnError(Exception e, String phase)
Optional<Environment>	getEnvironment()
EvalContext	getEvalContext()
EvalHandle	getEvalHookManager()
FeedbackCollector	getFeedbackCollector()
Optional<ReasoningStrategyHandle.ReasoningResult>	getLastReasoningResult()
PaymentBroker	getPaymentBroker()	Returns the configured payment broker; never null.
Optional<PlannerHandle>	getPlannerHandle()
Optional<ReasoningStrategyHandle>	getReasoningStrategy()
ReasoningStrategyHandle	getReasoningStrategyRaw()
boolean	isEnvironmentEnabled()
boolean	isEvalEnabled()
boolean	isPlanningEnabled()
boolean	isReasoningEnabled()
void	onEnvironmentChangeWithRoles(Percept percept, String agentClassName, List<Role> roles)	Handles environment change with access to roles for state extraction.
Percept	perceive(String agentClassName)
List<PlannerHandle.PlanStep>	plan(List<Role> roles)
List<PlannerHandle.PlanStep>	planForGoal(String goalName, List<Role> roles)
void	recordGoalCompletion(String goalId, boolean success, Map<String,Object> details)
void	recordToolOutcome(String toolName, Object result, boolean success, long latencyMs)	Records a tool outcome for feedback learning.
boolean	removeEvalHook(EvalHook hook)
void	setEnvironment(Environment env, String agentClassName)
void	setEvalEnabled(boolean enabled)
void	setFeedbackCollector(FeedbackCollector collector)
void	setLastReasoningResult(ReasoningStrategyHandle.ReasoningResult result)
void	setPaymentBroker(PaymentBroker broker)	Replaces the payment broker.
void	setPlannerHandle(PlannerHandle handle)
void	setReasoningStrategy(ReasoningStrategyHandle strategy)

Methods inherited from class java.lang.Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Details

getEvalHookManager

public EvalHandle getEvalHookManager()

getEvalContext

public EvalContext getEvalContext()

addEvalHook

public void addEvalHook(EvalHook hook)

removeEvalHook

public boolean removeEvalHook(EvalHook hook)

clearEvalContext

public EvalContext clearEvalContext()

setEvalEnabled

public void setEvalEnabled(boolean enabled)

isEvalEnabled

public boolean isEvalEnabled()

recordGoalCompletion

public void recordGoalCompletion(String goalId,
 boolean success,
 Map<String,Object> details)

ensureEvalContext

public void ensureEvalContext()

fireOnBeforeChat

public void fireOnBeforeChat(String message)

fireOnAfterChat

public void fireOnAfterChat(String result,
 long latencyMs)

fireOnError

public void fireOnError(Exception e,
 String phase)

fireOnBeforeToolCall

public void fireOnBeforeToolCall(String toolName,
 Map<String,Object> arguments)

fireOnAfterToolCall

public void fireOnAfterToolCall(String toolName,
 Object result,
 boolean success,
 long latencyMs)

fireOnAgentStop

public void fireOnAgentStop(String status)

getFeedbackCollector

public FeedbackCollector getFeedbackCollector()

setFeedbackCollector

public void setFeedbackCollector(FeedbackCollector collector)

recordToolOutcome

public void recordToolOutcome(String toolName,
 Object result,
 boolean success,
 long latencyMs)

Records a tool outcome for feedback learning. Called from Agent's fireOnAfterToolCall implementation.

getPaymentBroker

public PaymentBroker getPaymentBroker()

Returns the configured payment broker; never null. Defaults to PaymentBroker.noOp().

setPaymentBroker

public void setPaymentBroker(PaymentBroker broker)

Replaces the payment broker. Passing null resets to PaymentBroker.noOp() so callers never observe a null broker via getPaymentBroker().

plan

public List<PlannerHandle.PlanStep> plan(List<Role> roles)

planForGoal

public List<PlannerHandle.PlanStep> planForGoal(String goalName,
 List<Role> roles)

executePlan

public PlannerHandle.PlanResult executePlan(List<PlannerHandle.PlanStep> steps,
 List<Role> roles)

extractCurrentState

public Map<String,Object> extractCurrentState(List<Role> roles)

getPlannerHandle

public Optional<PlannerHandle> getPlannerHandle()

setPlannerHandle

public void setPlannerHandle(PlannerHandle handle)

isPlanningEnabled

public boolean isPlanningEnabled()

setEnvironment

public void setEnvironment(Environment env,
 String agentClassName)

getEnvironment

public Optional<Environment> getEnvironment()

isEnvironmentEnabled

public boolean isEnvironmentEnabled()

perceive

public Percept perceive(String agentClassName)

actOnEnvironment

public ActionOutcome actOnEnvironment(String actionName,
 Map<String,Object> parameters,
 String agentClassName)

onEnvironmentChangeWithRoles

public void onEnvironmentChangeWithRoles(Percept percept,
 String agentClassName,
 List<Role> roles)

Handles environment change with access to roles for state extraction. Called by Agent when it has roles available.

getReasoningStrategy

public Optional<ReasoningStrategyHandle> getReasoningStrategy()

setReasoningStrategy

public void setReasoningStrategy(ReasoningStrategyHandle strategy)

isReasoningEnabled

public boolean isReasoningEnabled()

getLastReasoningResult

public Optional<ReasoningStrategyHandle.ReasoningResult> getLastReasoningResult()

setLastReasoningResult

public void setLastReasoningResult(ReasoningStrategyHandle.ReasoningResult result)

getReasoningStrategyRaw

public ReasoningStrategyHandle getReasoningStrategyRaw()