@cogitator-ai/core 0.18.6 → 0.19.2
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.
- package/README.md +103 -31
- package/dist/agent-tool.d.ts +2 -2
- package/dist/agent-tool.d.ts.map +1 -1
- package/dist/agent-tool.js +3 -2
- package/dist/agent-tool.js.map +1 -1
- package/dist/agent.d.ts.map +1 -1
- package/dist/agent.js +10 -2
- package/dist/agent.js.map +1 -1
- package/dist/cache/cache-key.d.ts.map +1 -1
- package/dist/cache/cache-key.js +3 -0
- package/dist/cache/cache-key.js.map +1 -1
- package/dist/cache/index.d.ts +5 -5
- package/dist/cache/index.js +4 -4
- package/dist/cache/storage/index.d.ts +3 -3
- package/dist/cache/storage/index.js +2 -2
- package/dist/cache/storage/memory.d.ts.map +1 -1
- package/dist/cache/storage/memory.js +3 -1
- package/dist/cache/storage/memory.js.map +1 -1
- package/dist/cache/storage/redis.d.ts +1 -0
- package/dist/cache/storage/redis.d.ts.map +1 -1
- package/dist/cache/storage/redis.js +47 -22
- package/dist/cache/storage/redis.js.map +1 -1
- package/dist/cache/tool-cache.d.ts.map +1 -1
- package/dist/cache/tool-cache.js +14 -7
- package/dist/cache/tool-cache.js.map +1 -1
- package/dist/causal/capabilities/causal-explainer.js +2 -2
- package/dist/causal/capabilities/causal-planner.js +1 -1
- package/dist/causal/causal-reasoner.d.ts +4 -4
- package/dist/causal/causal-reasoner.d.ts.map +1 -1
- package/dist/causal/causal-reasoner.js +17 -13
- package/dist/causal/causal-reasoner.js.map +1 -1
- package/dist/causal/discovery/causal-extractor.js +1 -1
- package/dist/causal/discovery/causal-validator.js +1 -1
- package/dist/causal/discovery/hypothesis-generator.js +1 -1
- package/dist/causal/discovery/prompts.d.ts.map +1 -1
- package/dist/causal/discovery/prompts.js +44 -10
- package/dist/causal/discovery/prompts.js.map +1 -1
- package/dist/causal/graph/causal-graph.d.ts.map +1 -1
- package/dist/causal/graph/causal-graph.js +3 -3
- package/dist/causal/graph/causal-graph.js.map +1 -1
- package/dist/causal/graph/graph-builder.js +1 -1
- package/dist/causal/index.d.ts +27 -27
- package/dist/causal/index.js +17 -17
- package/dist/causal/inference/adjustment.js +1 -1
- package/dist/causal/inference/d-separation.d.ts.map +1 -1
- package/dist/causal/inference/d-separation.js +12 -31
- package/dist/causal/inference/d-separation.js.map +1 -1
- package/dist/causal/inference/inference-engine.d.ts +2 -1
- package/dist/causal/inference/inference-engine.d.ts.map +1 -1
- package/dist/causal/inference/inference-engine.js +55 -10
- package/dist/causal/inference/inference-engine.js.map +1 -1
- package/dist/causal/stores/causal-graph-store.d.ts.map +1 -1
- package/dist/causal/stores/causal-graph-store.js +33 -3
- package/dist/causal/stores/causal-graph-store.js.map +1 -1
- package/dist/causal/stores/causal-pattern-store.d.ts.map +1 -1
- package/dist/causal/stores/causal-pattern-store.js +13 -4
- package/dist/causal/stores/causal-pattern-store.js.map +1 -1
- package/dist/causal/stores/intervention-log.d.ts.map +1 -1
- package/dist/causal/stores/intervention-log.js +17 -3
- package/dist/causal/stores/intervention-log.js.map +1 -1
- package/dist/cogitator/index.d.ts +5 -5
- package/dist/cogitator/index.d.ts.map +1 -1
- package/dist/cogitator/index.js +5 -5
- package/dist/cogitator/index.js.map +1 -1
- package/dist/cogitator/initializers.d.ts +6 -6
- package/dist/cogitator/initializers.js +7 -7
- package/dist/cogitator/initializers.js.map +1 -1
- package/dist/cogitator/message-builder.d.ts +2 -2
- package/dist/cogitator/message-builder.js +1 -1
- package/dist/cogitator/streaming.d.ts +3 -3
- package/dist/cogitator/streaming.d.ts.map +1 -1
- package/dist/cogitator/streaming.js +20 -1
- package/dist/cogitator/streaming.js.map +1 -1
- package/dist/cogitator/tool-executor.d.ts +9 -4
- package/dist/cogitator/tool-executor.d.ts.map +1 -1
- package/dist/cogitator/tool-executor.js +31 -15
- package/dist/cogitator/tool-executor.js.map +1 -1
- package/dist/cogitator.d.ts +1 -1
- package/dist/cogitator.js +1 -1
- package/dist/constitutional/constitutional-ai.d.ts.map +1 -1
- package/dist/constitutional/constitutional-ai.js +17 -9
- package/dist/constitutional/constitutional-ai.js.map +1 -1
- package/dist/constitutional/critique-reviser.js +2 -2
- package/dist/constitutional/index.d.ts +12 -12
- package/dist/constitutional/index.js +7 -7
- package/dist/constitutional/input-filter.d.ts.map +1 -1
- package/dist/constitutional/input-filter.js +18 -11
- package/dist/constitutional/input-filter.js.map +1 -1
- package/dist/constitutional/output-filter.d.ts.map +1 -1
- package/dist/constitutional/output-filter.js +18 -11
- package/dist/constitutional/output-filter.js.map +1 -1
- package/dist/context/context-manager.d.ts.map +1 -1
- package/dist/context/context-manager.js +3 -4
- package/dist/context/context-manager.js.map +1 -1
- package/dist/context/index.d.ts +3 -3
- package/dist/context/index.js +2 -2
- package/dist/context/strategies/hybrid.js +4 -4
- package/dist/context/strategies/index.d.ts +5 -5
- package/dist/context/strategies/index.js +5 -5
- package/dist/context/strategies/sliding-window.js +1 -1
- package/dist/context/strategies/summarize.js +1 -1
- package/dist/context/strategies/truncate.d.ts +3 -0
- package/dist/context/strategies/truncate.d.ts.map +1 -1
- package/dist/context/strategies/truncate.js +61 -1
- package/dist/context/strategies/truncate.js.map +1 -1
- package/dist/cost-routing/budget-enforcer.d.ts +1 -1
- package/dist/cost-routing/cost-estimator.d.ts +1 -1
- package/dist/cost-routing/cost-estimator.d.ts.map +1 -1
- package/dist/cost-routing/cost-estimator.js +5 -15
- package/dist/cost-routing/cost-estimator.js.map +1 -1
- package/dist/cost-routing/cost-router.d.ts +1 -1
- package/dist/cost-routing/cost-router.js +4 -4
- package/dist/cost-routing/index.d.ts +7 -7
- package/dist/cost-routing/index.js +7 -7
- package/dist/cost-routing/model-selector.d.ts.map +1 -1
- package/dist/cost-routing/model-selector.js +6 -12
- package/dist/cost-routing/model-selector.js.map +1 -1
- package/dist/cost-routing/task-analyzer.d.ts.map +1 -1
- package/dist/cost-routing/task-analyzer.js +15 -8
- package/dist/cost-routing/task-analyzer.js.map +1 -1
- package/dist/index.d.ts +42 -39
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +24 -23
- package/dist/index.js.map +1 -1
- package/dist/learning/ab-testing.d.ts +1 -0
- package/dist/learning/ab-testing.d.ts.map +1 -1
- package/dist/learning/ab-testing.js +34 -13
- package/dist/learning/ab-testing.js.map +1 -1
- package/dist/learning/agent-optimizer.d.ts +5 -3
- package/dist/learning/agent-optimizer.d.ts.map +1 -1
- package/dist/learning/agent-optimizer.js +43 -15
- package/dist/learning/agent-optimizer.js.map +1 -1
- package/dist/learning/auto-optimizer.d.ts +4 -4
- package/dist/learning/auto-optimizer.d.ts.map +1 -1
- package/dist/learning/auto-optimizer.js +3 -0
- package/dist/learning/auto-optimizer.js.map +1 -1
- package/dist/learning/demo-selector.d.ts +1 -0
- package/dist/learning/demo-selector.d.ts.map +1 -1
- package/dist/learning/demo-selector.js +20 -11
- package/dist/learning/demo-selector.js.map +1 -1
- package/dist/learning/index.d.ts +23 -23
- package/dist/learning/index.js +12 -12
- package/dist/learning/instruction-optimizer.js +1 -1
- package/dist/learning/metrics.d.ts.map +1 -1
- package/dist/learning/metrics.js +4 -1
- package/dist/learning/metrics.js.map +1 -1
- package/dist/learning/postgres-trace-store.d.ts.map +1 -1
- package/dist/learning/postgres-trace-store.js +73 -20
- package/dist/learning/postgres-trace-store.js.map +1 -1
- package/dist/learning/prompt-logger.d.ts +3 -2
- package/dist/learning/prompt-logger.d.ts.map +1 -1
- package/dist/learning/prompt-logger.js +11 -3
- package/dist/learning/prompt-logger.js.map +1 -1
- package/dist/learning/prompt-monitor.js +2 -2
- package/dist/learning/prompt-monitor.js.map +1 -1
- package/dist/learning/prompts.d.ts.map +1 -1
- package/dist/learning/prompts.js +2 -1
- package/dist/learning/prompts.js.map +1 -1
- package/dist/learning/trace-store.d.ts +2 -0
- package/dist/learning/trace-store.d.ts.map +1 -1
- package/dist/learning/trace-store.js +25 -11
- package/dist/learning/trace-store.js.map +1 -1
- package/dist/llm/anthropic.d.ts +2 -1
- package/dist/llm/anthropic.d.ts.map +1 -1
- package/dist/llm/anthropic.js +35 -9
- package/dist/llm/anthropic.js.map +1 -1
- package/dist/llm/azure.d.ts +1 -1
- package/dist/llm/azure.js +1 -1
- package/dist/llm/bedrock.d.ts +1 -1
- package/dist/llm/bedrock.d.ts.map +1 -1
- package/dist/llm/bedrock.js +25 -8
- package/dist/llm/bedrock.js.map +1 -1
- package/dist/llm/debug.d.ts +1 -0
- package/dist/llm/debug.d.ts.map +1 -1
- package/dist/llm/debug.js +42 -10
- package/dist/llm/debug.js.map +1 -1
- package/dist/llm/errors.d.ts.map +1 -1
- package/dist/llm/errors.js +5 -4
- package/dist/llm/errors.js.map +1 -1
- package/dist/llm/google.d.ts +3 -1
- package/dist/llm/google.d.ts.map +1 -1
- package/dist/llm/google.js +127 -63
- package/dist/llm/google.js.map +1 -1
- package/dist/llm/index.d.ts +11 -11
- package/dist/llm/index.d.ts.map +1 -1
- package/dist/llm/index.js +49 -19
- package/dist/llm/index.js.map +1 -1
- package/dist/llm/ollama.d.ts +1 -1
- package/dist/llm/ollama.d.ts.map +1 -1
- package/dist/llm/ollama.js +97 -47
- package/dist/llm/ollama.js.map +1 -1
- package/dist/llm/openai-compatible-base.d.ts +2 -2
- package/dist/llm/openai-compatible-base.d.ts.map +1 -1
- package/dist/llm/openai-compatible-base.js +27 -8
- package/dist/llm/openai-compatible-base.js.map +1 -1
- package/dist/llm/openai.d.ts +4 -2
- package/dist/llm/openai.d.ts.map +1 -1
- package/dist/llm/openai.js +3 -2
- package/dist/llm/openai.js.map +1 -1
- package/dist/observability/index.d.ts +2 -2
- package/dist/observability/index.js +2 -2
- package/dist/observability/langfuse.d.ts +3 -0
- package/dist/observability/langfuse.d.ts.map +1 -1
- package/dist/observability/langfuse.js +24 -1
- package/dist/observability/langfuse.js.map +1 -1
- package/dist/observability/opentelemetry.d.ts +1 -0
- package/dist/observability/opentelemetry.d.ts.map +1 -1
- package/dist/observability/opentelemetry.js +5 -0
- package/dist/observability/opentelemetry.js.map +1 -1
- package/dist/reasoning/branch-evaluator.d.ts +1 -1
- package/dist/reasoning/branch-evaluator.d.ts.map +1 -1
- package/dist/reasoning/branch-evaluator.js +9 -4
- package/dist/reasoning/branch-evaluator.js.map +1 -1
- package/dist/reasoning/branch-generator.js +1 -1
- package/dist/reasoning/index.d.ts +4 -4
- package/dist/reasoning/index.js +4 -4
- package/dist/reasoning/prompts.d.ts.map +1 -1
- package/dist/reasoning/prompts.js +13 -0
- package/dist/reasoning/prompts.js.map +1 -1
- package/dist/reasoning/thought-tree.d.ts +1 -1
- package/dist/reasoning/thought-tree.js +3 -3
- package/dist/reflection/index.d.ts +3 -3
- package/dist/reflection/index.js +3 -3
- package/dist/reflection/reflection-engine.d.ts.map +1 -1
- package/dist/reflection/reflection-engine.js +10 -4
- package/dist/reflection/reflection-engine.js.map +1 -1
- package/dist/registry.js +1 -1
- package/dist/runtime.d.ts +8 -5
- package/dist/runtime.d.ts.map +1 -1
- package/dist/runtime.js +110 -23
- package/dist/runtime.js.map +1 -1
- package/dist/security/classifiers/index.d.ts +2 -2
- package/dist/security/classifiers/index.js +2 -2
- package/dist/security/classifiers/llm-classifier.d.ts.map +1 -1
- package/dist/security/classifiers/llm-classifier.js +5 -9
- package/dist/security/classifiers/llm-classifier.js.map +1 -1
- package/dist/security/classifiers/local-classifier.d.ts.map +1 -1
- package/dist/security/classifiers/local-classifier.js +3 -1
- package/dist/security/classifiers/local-classifier.js.map +1 -1
- package/dist/security/index.d.ts +4 -4
- package/dist/security/index.js +3 -3
- package/dist/security/patterns.d.ts.map +1 -1
- package/dist/security/patterns.js +2 -0
- package/dist/security/patterns.js.map +1 -1
- package/dist/security/prompt-injection-detector.js +2 -2
- package/dist/skill.d.ts +8 -0
- package/dist/skill.d.ts.map +1 -0
- package/dist/skill.js +77 -0
- package/dist/skill.js.map +1 -0
- package/dist/time-travel/checkpoint-store.d.ts +3 -0
- package/dist/time-travel/checkpoint-store.d.ts.map +1 -1
- package/dist/time-travel/checkpoint-store.js +76 -40
- package/dist/time-travel/checkpoint-store.js.map +1 -1
- package/dist/time-travel/comparator.d.ts.map +1 -1
- package/dist/time-travel/comparator.js +7 -1
- package/dist/time-travel/comparator.js.map +1 -1
- package/dist/time-travel/forker.d.ts +3 -3
- package/dist/time-travel/forker.d.ts.map +1 -1
- package/dist/time-travel/forker.js +4 -1
- package/dist/time-travel/forker.js.map +1 -1
- package/dist/time-travel/index.d.ts +5 -5
- package/dist/time-travel/index.js +5 -5
- package/dist/time-travel/replayer.d.ts +2 -2
- package/dist/time-travel/replayer.d.ts.map +1 -1
- package/dist/time-travel/replayer.js.map +1 -1
- package/dist/time-travel/time-travel.d.ts +2 -2
- package/dist/time-travel/time-travel.d.ts.map +1 -1
- package/dist/time-travel/time-travel.js +13 -13
- package/dist/time-travel/time-travel.js.map +1 -1
- package/dist/tool.d.ts.map +1 -1
- package/dist/tool.js +26 -8
- package/dist/tool.js.map +1 -1
- package/dist/tools/audio-generate.d.ts.map +1 -1
- package/dist/tools/audio-generate.js +33 -16
- package/dist/tools/audio-generate.js.map +1 -1
- package/dist/tools/audio-transcribe.d.ts.map +1 -1
- package/dist/tools/audio-transcribe.js +26 -9
- package/dist/tools/audio-transcribe.js.map +1 -1
- package/dist/tools/base64.d.ts +4 -4
- package/dist/tools/base64.d.ts.map +1 -1
- package/dist/tools/base64.js +22 -7
- package/dist/tools/base64.js.map +1 -1
- package/dist/tools/calculator.d.ts.map +1 -1
- package/dist/tools/calculator.js +13 -2
- package/dist/tools/calculator.js.map +1 -1
- package/dist/tools/capabilities-tool.d.ts +8 -0
- package/dist/tools/capabilities-tool.d.ts.map +1 -0
- package/dist/tools/capabilities-tool.js +35 -0
- package/dist/tools/capabilities-tool.js.map +1 -0
- package/dist/tools/datetime.js +1 -1
- package/dist/tools/device-tools.d.ts +3 -0
- package/dist/tools/device-tools.d.ts.map +1 -0
- package/dist/tools/device-tools.js +91 -0
- package/dist/tools/device-tools.js.map +1 -0
- package/dist/tools/email.d.ts.map +1 -1
- package/dist/tools/email.js +39 -11
- package/dist/tools/email.js.map +1 -1
- package/dist/tools/exec.d.ts +9 -9
- package/dist/tools/exec.d.ts.map +1 -1
- package/dist/tools/exec.js +16 -3
- package/dist/tools/exec.js.map +1 -1
- package/dist/tools/filesystem.js +1 -1
- package/dist/tools/github.d.ts.map +1 -1
- package/dist/tools/github.js +29 -21
- package/dist/tools/github.js.map +1 -1
- package/dist/tools/hash.js +1 -1
- package/dist/tools/http.d.ts.map +1 -1
- package/dist/tools/http.js +9 -8
- package/dist/tools/http.js.map +1 -1
- package/dist/tools/image-analyze.js +1 -1
- package/dist/tools/image-generate.d.ts.map +1 -1
- package/dist/tools/image-generate.js +35 -18
- package/dist/tools/image-generate.js.map +1 -1
- package/dist/tools/index.d.ts +58 -42
- package/dist/tools/index.d.ts.map +1 -1
- package/dist/tools/index.js +45 -40
- package/dist/tools/index.js.map +1 -1
- package/dist/tools/json.js +1 -1
- package/dist/tools/memory-tools.d.ts +50 -0
- package/dist/tools/memory-tools.d.ts.map +1 -0
- package/dist/tools/memory-tools.js +125 -0
- package/dist/tools/memory-tools.js.map +1 -0
- package/dist/tools/random.d.ts +7 -1
- package/dist/tools/random.d.ts.map +1 -1
- package/dist/tools/random.js +11 -4
- package/dist/tools/random.js.map +1 -1
- package/dist/tools/regex.d.ts.map +1 -1
- package/dist/tools/regex.js +4 -1
- package/dist/tools/regex.js.map +1 -1
- package/dist/tools/scheduler-tools.d.ts +32 -0
- package/dist/tools/scheduler-tools.d.ts.map +1 -0
- package/dist/tools/scheduler-tools.js +104 -0
- package/dist/tools/scheduler-tools.js.map +1 -0
- package/dist/tools/self-config.d.ts +9 -0
- package/dist/tools/self-config.d.ts.map +1 -0
- package/dist/tools/self-config.js +62 -0
- package/dist/tools/self-config.js.map +1 -0
- package/dist/tools/self-tools.d.ts +7 -0
- package/dist/tools/self-tools.d.ts.map +1 -0
- package/dist/tools/self-tools.js +172 -0
- package/dist/tools/self-tools.js.map +1 -0
- package/dist/tools/sleep.d.ts +5 -0
- package/dist/tools/sleep.d.ts.map +1 -1
- package/dist/tools/sleep.js +20 -3
- package/dist/tools/sleep.js.map +1 -1
- package/dist/tools/sql-query.d.ts.map +1 -1
- package/dist/tools/sql-query.js +148 -13
- package/dist/tools/sql-query.js.map +1 -1
- package/dist/tools/uuid.js +1 -1
- package/dist/tools/vector-search.d.ts.map +1 -1
- package/dist/tools/vector-search.js +82 -62
- package/dist/tools/vector-search.js.map +1 -1
- package/dist/tools/web-scrape.d.ts.map +1 -1
- package/dist/tools/web-scrape.js +9 -8
- package/dist/tools/web-scrape.js.map +1 -1
- package/dist/tools/web-search.d.ts.map +1 -1
- package/dist/tools/web-search.js +32 -14
- package/dist/tools/web-search.js.map +1 -1
- package/dist/utils/abort.d.ts +9 -0
- package/dist/utils/abort.d.ts.map +1 -0
- package/dist/utils/abort.js +43 -0
- package/dist/utils/abort.js.map +1 -0
- package/dist/utils/audio-fetch.d.ts +5 -1
- package/dist/utils/audio-fetch.d.ts.map +1 -1
- package/dist/utils/audio-fetch.js +13 -6
- package/dist/utils/audio-fetch.js.map +1 -1
- package/dist/utils/fallback.d.ts +2 -2
- package/dist/utils/fallback.js +1 -1
- package/dist/utils/image-fetch.d.ts +4 -0
- package/dist/utils/image-fetch.d.ts.map +1 -1
- package/dist/utils/image-fetch.js +13 -6
- package/dist/utils/image-fetch.js.map +1 -1
- package/dist/utils/index.d.ts +5 -5
- package/dist/utils/index.js +5 -5
- package/dist/utils/retry.d.ts.map +1 -1
- package/dist/utils/retry.js +3 -0
- package/dist/utils/retry.js.map +1 -1
- package/package.json +7 -3
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
import type { Message, RunOptions, MemoryAdapter, ToolCall, ToolResult } from '@cogitator-ai/types';
|
|
2
2
|
import { ContextBuilder } from '@cogitator-ai/memory';
|
|
3
|
-
import type { Agent } from '../agent';
|
|
4
|
-
import type { ReflectionEngine } from '../reflection/index';
|
|
3
|
+
import type { Agent } from '../agent.js';
|
|
4
|
+
import type { ReflectionEngine } from '../reflection/index.js';
|
|
5
5
|
import type { AgentContext } from '@cogitator-ai/types';
|
|
6
6
|
export declare function buildInitialMessages(agent: Agent, options: RunOptions, threadId: string, memoryAdapter: MemoryAdapter | undefined, contextBuilder: ContextBuilder | undefined): Promise<Message[]>;
|
|
7
7
|
export declare function saveEntry(threadId: string, agentId: string, message: Message, memoryAdapter: MemoryAdapter | undefined, toolCalls?: ToolCall[], toolResults?: ToolResult[], onError?: (error: Error, operation: 'save' | 'load') => void): Promise<void>;
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import type { LLMBackend, Message, ToolCall } from '@cogitator-ai/types';
|
|
2
|
-
import { ToolRegistry } from '../registry';
|
|
3
|
-
import type { Agent } from '../agent';
|
|
2
|
+
import { ToolRegistry } from '../registry.js';
|
|
3
|
+
import type { Agent } from '../agent.js';
|
|
4
4
|
export interface StreamChatResult {
|
|
5
5
|
id: string;
|
|
6
6
|
content: string;
|
|
@@ -12,5 +12,5 @@ export interface StreamChatResult {
|
|
|
12
12
|
totalTokens: number;
|
|
13
13
|
};
|
|
14
14
|
}
|
|
15
|
-
export declare function streamChat(backend: LLMBackend, model: string, messages: Message[], registry: ToolRegistry, agent: Agent, onToken: (token: string) => void): Promise<StreamChatResult>;
|
|
15
|
+
export declare function streamChat(backend: LLMBackend, model: string, messages: Message[], registry: ToolRegistry, agent: Agent, onToken: (token: string) => void, signal?: AbortSignal): Promise<StreamChatResult>;
|
|
16
16
|
//# sourceMappingURL=streaming.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"streaming.d.ts","sourceRoot":"","sources":["../../src/cogitator/streaming.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAEzE,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,UAAU,CAAC;AAEtC,MAAM,WAAW,gBAAgB;IAC/B,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,QAAQ,EAAE,CAAC;IACvB,YAAY,EAAE,MAAM,GAAG,YAAY,GAAG,QAAQ,GAAG,OAAO,CAAC;IACzD,KAAK,EAAE;QACL,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,WAAW,EAAE,MAAM,CAAC;KACrB,CAAC;CACH;AAED,wBAAsB,UAAU,CAC9B,OAAO,EAAE,UAAU,EACnB,KAAK,EAAE,MAAM,EACb,QAAQ,EAAE,OAAO,EAAE,EACnB,QAAQ,EAAE,YAAY,EACtB,KAAK,EAAE,KAAK,EACZ,OAAO,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,
|
|
1
|
+
{"version":3,"file":"streaming.d.ts","sourceRoot":"","sources":["../../src/cogitator/streaming.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAEzE,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,UAAU,CAAC;AAEtC,MAAM,WAAW,gBAAgB;IAC/B,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,QAAQ,EAAE,CAAC;IACvB,YAAY,EAAE,MAAM,GAAG,YAAY,GAAG,QAAQ,GAAG,OAAO,CAAC;IACzD,KAAK,EAAE;QACL,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,WAAW,EAAE,MAAM,CAAC;KACrB,CAAC;CACH;AAED,wBAAsB,UAAU,CAC9B,OAAO,EAAE,UAAU,EACnB,KAAK,EAAE,MAAM,EACb,QAAQ,EAAE,OAAO,EAAE,EACnB,QAAQ,EAAE,YAAY,EACtB,KAAK,EAAE,KAAK,EACZ,OAAO,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,EAChC,MAAM,CAAC,EAAE,WAAW,GACnB,OAAO,CAAC,gBAAgB,CAAC,CAgF3B"}
|
|
@@ -1,6 +1,7 @@
|
|
|
1
1
|
import { nanoid } from 'nanoid';
|
|
2
2
|
import { countMessagesTokens } from '@cogitator-ai/memory';
|
|
3
|
-
export async function streamChat(backend, model, messages, registry, agent, onToken) {
|
|
3
|
+
export async function streamChat(backend, model, messages, registry, agent, onToken, signal) {
|
|
4
|
+
throwIfStreamAborted(signal);
|
|
4
5
|
let content = '';
|
|
5
6
|
let toolCalls;
|
|
6
7
|
let finishReason = 'stop';
|
|
@@ -15,8 +16,10 @@ export async function streamChat(backend, model, messages, registry, agent, onTo
|
|
|
15
16
|
topP: agent.config.topP,
|
|
16
17
|
maxTokens: agent.config.maxTokens,
|
|
17
18
|
stop: agent.config.stopSequences,
|
|
19
|
+
signal,
|
|
18
20
|
});
|
|
19
21
|
for await (const chunk of stream) {
|
|
22
|
+
throwIfStreamAborted(signal);
|
|
20
23
|
if (chunk.delta.content) {
|
|
21
24
|
content += chunk.delta.content;
|
|
22
25
|
onToken(chunk.delta.content);
|
|
@@ -40,6 +43,10 @@ export async function streamChat(backend, model, messages, registry, agent, onTo
|
|
|
40
43
|
});
|
|
41
44
|
}
|
|
42
45
|
}
|
|
46
|
+
else if (toolCalls.length > 0 && partial.arguments) {
|
|
47
|
+
const last = toolCalls[toolCalls.length - 1];
|
|
48
|
+
last.arguments = { ...last.arguments, ...partial.arguments };
|
|
49
|
+
}
|
|
43
50
|
}
|
|
44
51
|
}
|
|
45
52
|
if (chunk.finishReason) {
|
|
@@ -55,6 +62,9 @@ export async function streamChat(backend, model, messages, registry, agent, onTo
|
|
|
55
62
|
inputTokens = countMessagesTokens(messages);
|
|
56
63
|
outputTokens = Math.ceil(content.length / 4);
|
|
57
64
|
}
|
|
65
|
+
if (toolCalls && toolCalls.length > 0 && finishReason !== 'tool_calls') {
|
|
66
|
+
finishReason = 'tool_calls';
|
|
67
|
+
}
|
|
58
68
|
return {
|
|
59
69
|
id: `stream_${nanoid(8)}`,
|
|
60
70
|
content,
|
|
@@ -67,4 +77,13 @@ export async function streamChat(backend, model, messages, registry, agent, onTo
|
|
|
67
77
|
},
|
|
68
78
|
};
|
|
69
79
|
}
|
|
80
|
+
function throwIfStreamAborted(signal) {
|
|
81
|
+
if (!signal?.aborted) {
|
|
82
|
+
return;
|
|
83
|
+
}
|
|
84
|
+
if (signal.reason instanceof Error) {
|
|
85
|
+
throw signal.reason;
|
|
86
|
+
}
|
|
87
|
+
throw new Error(signal.reason === undefined ? 'Stream aborted' : String(signal.reason));
|
|
88
|
+
}
|
|
70
89
|
//# sourceMappingURL=streaming.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"streaming.js","sourceRoot":"","sources":["../../src/cogitator/streaming.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAEhC,OAAO,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAgB3D,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,OAAmB,EACnB,KAAa,EACb,QAAmB,EACnB,QAAsB,EACtB,KAAY,EACZ,OAAgC;
|
|
1
|
+
{"version":3,"file":"streaming.js","sourceRoot":"","sources":["../../src/cogitator/streaming.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAEhC,OAAO,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAgB3D,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,OAAmB,EACnB,KAAa,EACb,QAAmB,EACnB,QAAsB,EACtB,KAAY,EACZ,OAAgC,EAChC,MAAoB;IAEpB,oBAAoB,CAAC,MAAM,CAAC,CAAC;IAE7B,IAAI,OAAO,GAAG,EAAE,CAAC;IACjB,IAAI,SAAiC,CAAC;IACtC,IAAI,YAAY,GAA+C,MAAM,CAAC;IACtE,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,IAAI,kBAAkB,GAAG,KAAK,CAAC;IAE/B,MAAM,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC;QAChC,KAAK;QACL,QAAQ;QACR,KAAK,EAAE,QAAQ,CAAC,UAAU,EAAE;QAC5B,WAAW,EAAE,KAAK,CAAC,MAAM,CAAC,WAAW;QACrC,IAAI,EAAE,KAAK,CAAC,MAAM,CAAC,IAAI;QACvB,SAAS,EAAE,KAAK,CAAC,MAAM,CAAC,SAAS;QACjC,IAAI,EAAE,KAAK,CAAC,MAAM,CAAC,aAAa;QAChC,MAAM;KACP,CAAC,CAAC;IAEH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QACjC,oBAAoB,CAAC,MAAM,CAAC,CAAC;QAE7B,IAAI,KAAK,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;YACxB,OAAO,IAAI,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC;YAC/B,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC;QACD,IAAI,KAAK,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;YAC1B,IAAI,CAAC,SAAS;gBAAE,SAAS,GAAG,EAAE,CAAC;YAC/B,KAAK,MAAM,OAAO,IAAI,KAAK,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;gBAC5C,IAAI,OAAO,CAAC,EAAE,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;oBAC/B,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,OAAO,CAAC,EAAE,CAAC,CAAC;oBAC9D,IAAI,QAAQ,EAAE,CAAC;wBACb,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;4BACtB,QAAQ,CAAC,SAAS,GAAG,EAAE,GAAG,QAAQ,CAAC,SAAS,EAAE,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;wBACvE,CAAC;oBACH,CAAC;yBAAM,CAAC;wBACN,SAAS,CAAC,IAAI,CAAC;4BACb,EAAE,EAAE,OAAO,CAAC,EAAE;4BACd,IAAI,EAAE,OAAO,CAAC,IAAI;4BAClB,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,EAAE;yBACnC,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;qBAAM,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;oBACrD,MAAM,IAAI,GAAG,SAAS,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;oBAC7C,IAAI,CAAC,SAAS,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,EAAE,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;gBAC/D,CAAC;YACH,CAAC;QACH,CAAC;QACD,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;YACvB,YAAY,GAAG,KAAK,CAAC,YAAY,CAAC;QACpC,CAAC;QACD,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;YAChB,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC;YACtC,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC;YACxC,kBAAkB,GAAG,IAAI,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACxB,WAAW,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QAC5C,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAC/C,CAAC;IAED,IAAI,SAAS,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,YAAY,KAAK,YAAY,EAAE,CAAC;QACvE,YAAY,GAAG,YAAY,CAAC;IAC9B,CAAC;IAED,OAAO;QACL,EAAE,EAAE,UAAU,MAAM,CAAC,CAAC,CAAC,EAAE;QACzB,OAAO;QACP,SAAS;QACT,YAAY;QACZ,KAAK,EAAE;YACL,WAAW;YACX,YAAY;YACZ,WAAW,EAAE,WAAW,GAAG,YAAY;SACxC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,oBAAoB,CAAC,MAAoB;IAChD,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;QACrB,OAAO;IACT,CAAC;IAED,IAAI,MAAM,CAAC,MAAM,YAAY,KAAK,EAAE,CAAC;QACnC,MAAM,MAAM,CAAC,MAAM,CAAC;IACtB,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;AAC1F,CAAC"}
|
|
@@ -1,7 +1,12 @@
|
|
|
1
1
|
import type { ToolCall, ToolResult, Message } from '@cogitator-ai/types';
|
|
2
|
-
import { ToolRegistry } from '../registry';
|
|
3
|
-
import type { SandboxManager } from './initializers';
|
|
4
|
-
import type { ConstitutionalAI } from '../constitutional/index';
|
|
5
|
-
export declare function executeTool(registry: ToolRegistry, toolCall: ToolCall, runId: string, agentId: string, sandboxManager: SandboxManager | undefined, constitutionalAI: ConstitutionalAI | undefined, filterToolCalls: boolean, initializeSandbox: () => Promise<void>, signal?: AbortSignal
|
|
2
|
+
import { ToolRegistry } from '../registry.js';
|
|
3
|
+
import type { SandboxManager } from './initializers.js';
|
|
4
|
+
import type { ConstitutionalAI } from '../constitutional/index.js';
|
|
5
|
+
export declare function executeTool(registry: ToolRegistry, toolCall: ToolCall, runId: string, agentId: string, sandboxManager: SandboxManager | undefined, constitutionalAI: ConstitutionalAI | undefined, filterToolCalls: boolean, initializeSandbox: () => Promise<void>, signal?: AbortSignal, extraContext?: {
|
|
6
|
+
threadId?: string;
|
|
7
|
+
userId?: string;
|
|
8
|
+
channelType?: string;
|
|
9
|
+
channelId?: string;
|
|
10
|
+
}): Promise<ToolResult>;
|
|
6
11
|
export declare function createToolMessage(toolCall: ToolCall, result: ToolResult): Message;
|
|
7
12
|
//# sourceMappingURL=tool-executor.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"tool-executor.d.ts","sourceRoot":"","sources":["../../src/cogitator/tool-executor.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAQ,QAAQ,EAAE,UAAU,EAAe,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC5F,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAE3C,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AACrD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAEhE,wBAAsB,WAAW,CAC/B,QAAQ,EAAE,YAAY,EACtB,QAAQ,EAAE,QAAQ,EAClB,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,MAAM,EACf,cAAc,EAAE,cAAc,GAAG,SAAS,EAC1C,gBAAgB,EAAE,gBAAgB,GAAG,SAAS,EAC9C,eAAe,EAAE,OAAO,EACxB,iBAAiB,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,EACtC,MAAM,CAAC,EAAE,WAAW,
|
|
1
|
+
{"version":3,"file":"tool-executor.d.ts","sourceRoot":"","sources":["../../src/cogitator/tool-executor.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAQ,QAAQ,EAAE,UAAU,EAAe,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC5F,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAE3C,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AACrD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAEhE,wBAAsB,WAAW,CAC/B,QAAQ,EAAE,YAAY,EACtB,QAAQ,EAAE,QAAQ,EAClB,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,MAAM,EACf,cAAc,EAAE,cAAc,GAAG,SAAS,EAC1C,gBAAgB,EAAE,gBAAgB,GAAG,SAAS,EAC9C,eAAe,EAAE,OAAO,EACxB,iBAAiB,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,EACtC,MAAM,CAAC,EAAE,WAAW,EACpB,YAAY,CAAC,EAAE;IAAE,QAAQ,CAAC,EAAE,MAAM,CAAC;IAAC,MAAM,CAAC,EAAE,MAAM,CAAC;IAAC,WAAW,CAAC,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,GAC9F,OAAO,CAAC,UAAU,CAAC,CAqFrB;AA0GD,wBAAgB,iBAAiB,CAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,GAAG,OAAO,CAUjF"}
|
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import { getLogger } from '../logger';
|
|
2
|
-
export async function executeTool(registry, toolCall, runId, agentId, sandboxManager, constitutionalAI, filterToolCalls, initializeSandbox, signal) {
|
|
1
|
+
import { getLogger } from '../logger.js';
|
|
2
|
+
export async function executeTool(registry, toolCall, runId, agentId, sandboxManager, constitutionalAI, filterToolCalls, initializeSandbox, signal, extraContext) {
|
|
3
3
|
const tool = registry.get(toolCall.name);
|
|
4
4
|
if (!tool) {
|
|
5
5
|
return {
|
|
@@ -9,22 +9,28 @@ export async function executeTool(registry, toolCall, runId, agentId, sandboxMan
|
|
|
9
9
|
error: `Tool not found: ${toolCall.name}`,
|
|
10
10
|
};
|
|
11
11
|
}
|
|
12
|
-
const
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
12
|
+
const isZod = typeof tool.parameters.safeParse === 'function';
|
|
13
|
+
let validatedArgs = toolCall.arguments;
|
|
14
|
+
if (isZod) {
|
|
15
|
+
const parseResult = tool.parameters.safeParse(toolCall.arguments);
|
|
16
|
+
if (!parseResult.success) {
|
|
17
|
+
return {
|
|
18
|
+
callId: toolCall.id,
|
|
19
|
+
name: toolCall.name,
|
|
20
|
+
result: null,
|
|
21
|
+
error: `Invalid arguments: ${parseResult.error.message}`,
|
|
22
|
+
};
|
|
23
|
+
}
|
|
24
|
+
validatedArgs = parseResult.data;
|
|
20
25
|
}
|
|
21
26
|
if (constitutionalAI && filterToolCalls) {
|
|
22
27
|
const context = {
|
|
23
28
|
agentId,
|
|
24
29
|
runId,
|
|
25
30
|
signal: signal ?? new AbortController().signal,
|
|
31
|
+
...extraContext,
|
|
26
32
|
};
|
|
27
|
-
const guardResult = await constitutionalAI.guardTool(tool,
|
|
33
|
+
const guardResult = await constitutionalAI.guardTool(tool, validatedArgs, context);
|
|
28
34
|
if (!guardResult.approved) {
|
|
29
35
|
return {
|
|
30
36
|
callId: toolCall.id,
|
|
@@ -35,15 +41,16 @@ export async function executeTool(registry, toolCall, runId, agentId, sandboxMan
|
|
|
35
41
|
}
|
|
36
42
|
}
|
|
37
43
|
if (tool.sandbox?.type === 'docker' || tool.sandbox?.type === 'wasm') {
|
|
38
|
-
return executeInSandbox(tool, { ...toolCall, arguments:
|
|
44
|
+
return executeInSandbox(tool, { ...toolCall, arguments: validatedArgs }, runId, agentId, sandboxManager, initializeSandbox, signal, extraContext);
|
|
39
45
|
}
|
|
40
46
|
const context = {
|
|
41
47
|
agentId,
|
|
42
48
|
runId,
|
|
43
49
|
signal: signal ?? new AbortController().signal,
|
|
50
|
+
...extraContext,
|
|
44
51
|
};
|
|
45
52
|
try {
|
|
46
|
-
const result = await tool.execute(
|
|
53
|
+
const result = await tool.execute(validatedArgs, context);
|
|
47
54
|
return {
|
|
48
55
|
callId: toolCall.id,
|
|
49
56
|
name: toolCall.name,
|
|
@@ -59,14 +66,15 @@ export async function executeTool(registry, toolCall, runId, agentId, sandboxMan
|
|
|
59
66
|
};
|
|
60
67
|
}
|
|
61
68
|
}
|
|
62
|
-
async function executeInSandbox(tool, toolCall, runId, agentId, sandboxManager, initializeSandbox) {
|
|
69
|
+
async function executeInSandbox(tool, toolCall, runId, agentId, sandboxManager, initializeSandbox, signal, extraContext) {
|
|
63
70
|
await initializeSandbox();
|
|
64
71
|
if (!sandboxManager) {
|
|
65
72
|
getLogger().warn('Sandbox unavailable, executing natively', { tool: tool.name });
|
|
66
73
|
const context = {
|
|
67
74
|
agentId,
|
|
68
75
|
runId,
|
|
69
|
-
signal: new AbortController().signal,
|
|
76
|
+
signal: signal ?? new AbortController().signal,
|
|
77
|
+
...extraContext,
|
|
70
78
|
};
|
|
71
79
|
try {
|
|
72
80
|
const result = await tool.execute(toolCall.arguments, context);
|
|
@@ -105,6 +113,14 @@ async function executeInSandbox(tool, toolCall, runId, agentId, sandboxManager,
|
|
|
105
113
|
error: result.error,
|
|
106
114
|
};
|
|
107
115
|
}
|
|
116
|
+
if (!result.data) {
|
|
117
|
+
return {
|
|
118
|
+
callId: toolCall.id,
|
|
119
|
+
name: toolCall.name,
|
|
120
|
+
result: null,
|
|
121
|
+
error: 'Sandbox returned success but no data',
|
|
122
|
+
};
|
|
123
|
+
}
|
|
108
124
|
if (isWasm) {
|
|
109
125
|
try {
|
|
110
126
|
const parsed = JSON.parse(result.data.stdout);
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"tool-executor.js","sourceRoot":"","sources":["../../src/cogitator/tool-executor.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAItC,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,QAAsB,EACtB,QAAkB,EAClB,KAAa,EACb,OAAe,EACf,cAA0C,EAC1C,gBAA8C,EAC9C,eAAwB,EACxB,iBAAsC,EACtC,MAAoB;
|
|
1
|
+
{"version":3,"file":"tool-executor.js","sourceRoot":"","sources":["../../src/cogitator/tool-executor.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAItC,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,QAAsB,EACtB,QAAkB,EAClB,KAAa,EACb,OAAe,EACf,cAA0C,EAC1C,gBAA8C,EAC9C,eAAwB,EACxB,iBAAsC,EACtC,MAAoB,EACpB,YAA+F;IAE/F,MAAM,IAAI,GAAG,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IAEzC,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO;YACL,MAAM,EAAE,QAAQ,CAAC,EAAE;YACnB,IAAI,EAAE,QAAQ,CAAC,IAAI;YACnB,MAAM,EAAE,IAAI;YACZ,KAAK,EAAE,mBAAmB,QAAQ,CAAC,IAAI,EAAE;SAC1C,CAAC;IACJ,CAAC;IAED,MAAM,KAAK,GAAG,OAAO,IAAI,CAAC,UAAU,CAAC,SAAS,KAAK,UAAU,CAAC;IAC9D,IAAI,aAAa,GAAY,QAAQ,CAAC,SAAS,CAAC;IAEhD,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,WAAW,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAClE,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;YACzB,OAAO;gBACL,MAAM,EAAE,QAAQ,CAAC,EAAE;gBACnB,IAAI,EAAE,QAAQ,CAAC,IAAI;gBACnB,MAAM,EAAE,IAAI;gBACZ,KAAK,EAAE,sBAAsB,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE;aACzD,CAAC;QACJ,CAAC;QACD,aAAa,GAAG,WAAW,CAAC,IAAI,CAAC;IACnC,CAAC;IAED,IAAI,gBAAgB,IAAI,eAAe,EAAE,CAAC;QACxC,MAAM,OAAO,GAAgB;YAC3B,OAAO;YACP,KAAK;YACL,MAAM,EAAE,MAAM,IAAI,IAAI,eAAe,EAAE,CAAC,MAAM;YAC9C,GAAG,YAAY;SAChB,CAAC;QACF,MAAM,WAAW,GAAG,MAAM,gBAAgB,CAAC,SAAS,CAClD,IAAI,EACJ,aAAwC,EACxC,OAAO,CACR,CAAC;QACF,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC;YAC1B,OAAO;gBACL,MAAM,EAAE,QAAQ,CAAC,EAAE;gBACnB,IAAI,EAAE,QAAQ,CAAC,IAAI;gBACnB,MAAM,EAAE,IAAI;gBACZ,KAAK,EAAE,iBAAiB,WAAW,CAAC,MAAM,IAAI,kBAAkB,EAAE;aACnE,CAAC;QACJ,CAAC;IACH,CAAC;IAED,IAAI,IAAI,CAAC,OAAO,EAAE,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,OAAO,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;QACrE,OAAO,gBAAgB,CACrB,IAAI,EACJ,EAAE,GAAG,QAAQ,EAAE,SAAS,EAAE,aAAwC,EAAE,EACpE,KAAK,EACL,OAAO,EACP,cAAc,EACd,iBAAiB,EACjB,MAAM,EACN,YAAY,CACb,CAAC;IACJ,CAAC;IAED,MAAM,OAAO,GAAgB;QAC3B,OAAO;QACP,KAAK;QACL,MAAM,EAAE,MAAM,IAAI,IAAI,eAAe,EAAE,CAAC,MAAM;QAC9C,GAAG,YAAY;KAChB,CAAC;IAEF,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;QAC1D,OAAO;YACL,MAAM,EAAE,QAAQ,CAAC,EAAE;YACnB,IAAI,EAAE,QAAQ,CAAC,IAAI;YACnB,MAAM;SACP,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,MAAM,EAAE,QAAQ,CAAC,EAAE;YACnB,IAAI,EAAE,QAAQ,CAAC,IAAI;YACnB,MAAM,EAAE,IAAI;YACZ,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;SAC9D,CAAC;IACJ,CAAC;AACH,CAAC;AAED,KAAK,UAAU,gBAAgB,CAC7B,IAAU,EACV,QAAkB,EAClB,KAAa,EACb,OAAe,EACf,cAA0C,EAC1C,iBAAsC,EACtC,MAAoB,EACpB,YAA+F;IAE/F,MAAM,iBAAiB,EAAE,CAAC;IAE1B,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,SAAS,EAAE,CAAC,IAAI,CAAC,yCAAyC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QACjF,MAAM,OAAO,GAAgB;YAC3B,OAAO;YACP,KAAK;YACL,MAAM,EAAE,MAAM,IAAI,IAAI,eAAe,EAAE,CAAC,MAAM;YAC9C,GAAG,YAAY;SAChB,CAAC;QACF,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAC/B,QAAQ,CAAC,SAA+C,EACxD,OAAO,CACR,CAAC;YACF,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC;QAC9D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,MAAM,EAAE,QAAQ,CAAC,EAAE;gBACnB,IAAI,EAAE,QAAQ,CAAC,IAAI;gBACnB,MAAM,EAAE,IAAI;gBACZ,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAC9D,CAAC;QACJ,CAAC;IACH,CAAC;IAED,MAAM,IAAI,GAAG,QAAQ,CAAC,SAAS,CAAC;IAChC,MAAM,aAAa,GAAG,IAAI,CAAC,OAAQ,CAAC;IAEpC,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,KAAK,MAAM,CAAC;IAC7C,MAAM,OAAO,GAAG,MAAM;QACpB,CAAC,CAAC;YACE,OAAO,EAAE,EAAE;YACX,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;YAC3B,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB;QACH,CAAC,CAAC;YACE,OAAO,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;YACjD,GAAG,EAAE,IAAI,CAAC,GAAyB;YACnC,GAAG,EAAE,IAAI,CAAC,GAAyC;YACnD,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC;IAEN,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,OAAO,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;IAEpE,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QACpB,OAAO;YACL,MAAM,EAAE,QAAQ,CAAC,EAAE;YACnB,IAAI,EAAE,QAAQ,CAAC,IAAI;YACnB,MAAM,EAAE,IAAI;YACZ,KAAK,EAAE,MAAM,CAAC,KAAK;SACpB,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QACjB,OAAO;YACL,MAAM,EAAE,QAAQ,CAAC,EAAE;YACnB,IAAI,EAAE,QAAQ,CAAC,IAAI;YACnB,MAAM,EAAE,IAAI;YACZ,KAAK,EAAE,sCAAsC;SAC9C,CAAC;IACJ,CAAC;IAED,IAAI,MAAM,EAAE,CAAC;QACX,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC9C,OAAO;gBACL,MAAM,EAAE,QAAQ,CAAC,EAAE;gBACnB,IAAI,EAAE,QAAQ,CAAC,IAAI;gBACnB,MAAM,EAAE,MAAM;aACf,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,OAAO;gBACL,MAAM,EAAE,QAAQ,CAAC,EAAE;gBACnB,IAAI,EAAE,QAAQ,CAAC,IAAI;gBACnB,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM;aAC3B,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO;QACL,MAAM,EAAE,QAAQ,CAAC,EAAE;QACnB,IAAI,EAAE,QAAQ,CAAC,IAAI;QACnB,MAAM,EAAE;YACN,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM;YAC1B,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM;YAC1B,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,QAAQ;YAC9B,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,QAAQ;YAC9B,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,QAAQ;YAC9B,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,QAAkB,EAAE,MAAkB;IACtE,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK;QAC1B,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;QACzC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC;IAC1C,OAAO;QACL,IAAI,EAAE,MAAM;QACZ,OAAO;QACP,UAAU,EAAE,QAAQ,CAAC,EAAE;QACvB,IAAI,EAAE,QAAQ,CAAC,IAAI;KACpB,CAAC;AACJ,CAAC"}
|
package/dist/cogitator.d.ts
CHANGED
|
@@ -1,2 +1,2 @@
|
|
|
1
|
-
export { Cogitator } from './runtime';
|
|
1
|
+
export { Cogitator } from './runtime.js';
|
|
2
2
|
//# sourceMappingURL=cogitator.d.ts.map
|
package/dist/cogitator.js
CHANGED
|
@@ -1,2 +1,2 @@
|
|
|
1
|
-
export { Cogitator } from './runtime';
|
|
1
|
+
export { Cogitator } from './runtime.js';
|
|
2
2
|
//# sourceMappingURL=cogitator.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"constitutional-ai.d.ts","sourceRoot":"","sources":["../../src/constitutional/constitutional-ai.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,eAAe,EACf,YAAY,EACZ,uBAAuB,EACvB,YAAY,EACZ,eAAe,EACf,cAAc,EACd,UAAU,EACV,OAAO,EACP,IAAI,EACJ,WAAW,EACX,WAAW,EACZ,MAAM,qBAAqB,CAAC;AAQ7B,MAAM,WAAW,uBAAuB;IACtC,GAAG,EAAE,UAAU,CAAC;IAChB,YAAY,CAAC,EAAE,YAAY,CAAC;IAC5B,MAAM,CAAC,EAAE,OAAO,CAAC,eAAe,CAAC,CAAC;CACnC;AAED,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,YAAY,CAAe;IACnC,OAAO,CAAC,SAAS,CAAY;IAC7B,OAAO,CAAC,eAAe,CAAkB;IACzC,OAAO,CAAC,OAAO,CAAkB;IACjC,OAAO,CAAC,aAAa,CAAe;IACpC,OAAO,CAAC,YAAY,CAA4E;IAChG,OAAO,CAAC,MAAM,CAAwD;gBAE1D,OAAO,EAAE,uBAAuB;IAmD5C,IAAI,MAAM,IAAI,eAAe,CAE5B;IAED,IAAI,YAAY,IAAI,YAAY,CAE/B;IAEK,WAAW,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAUnE,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,YAAY,CAAC;IAsBvE,SAAS,CACb,IAAI,EAAE,IAAI,EACV,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,OAAO,EAAE,WAAW,GACnB,OAAO,CAAC,eAAe,CAAC;IAarB,iBAAiB,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,cAAc,CAAC;
|
|
1
|
+
{"version":3,"file":"constitutional-ai.d.ts","sourceRoot":"","sources":["../../src/constitutional/constitutional-ai.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,eAAe,EACf,YAAY,EACZ,uBAAuB,EACvB,YAAY,EACZ,eAAe,EACf,cAAc,EACd,UAAU,EACV,OAAO,EACP,IAAI,EACJ,WAAW,EACX,WAAW,EACZ,MAAM,qBAAqB,CAAC;AAQ7B,MAAM,WAAW,uBAAuB;IACtC,GAAG,EAAE,UAAU,CAAC;IAChB,YAAY,CAAC,EAAE,YAAY,CAAC;IAC5B,MAAM,CAAC,EAAE,OAAO,CAAC,eAAe,CAAC,CAAC;CACnC;AAED,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,YAAY,CAAe;IACnC,OAAO,CAAC,SAAS,CAAY;IAC7B,OAAO,CAAC,eAAe,CAAkB;IACzC,OAAO,CAAC,OAAO,CAAkB;IACjC,OAAO,CAAC,aAAa,CAAe;IACpC,OAAO,CAAC,YAAY,CAA4E;IAChG,OAAO,CAAC,MAAM,CAAwD;gBAE1D,OAAO,EAAE,uBAAuB;IAmD5C,IAAI,MAAM,IAAI,eAAe,CAE5B;IAED,IAAI,YAAY,IAAI,YAAY,CAE/B;IAEK,WAAW,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAUnE,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,YAAY,CAAC;IAsBvE,SAAS,CACb,IAAI,EAAE,IAAI,EACV,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,OAAO,EAAE,WAAW,GACnB,OAAO,CAAC,eAAe,CAAC;IAarB,iBAAiB,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,cAAc,CAAC;IAatF,eAAe,CAAC,YAAY,EAAE,YAAY,GAAG,IAAI;IASjD,YAAY,CAAC,SAAS,EAAE,uBAAuB,GAAG,IAAI;IActD,eAAe,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI;IAQjC,eAAe,IAAI,YAAY;IAI/B,SAAS,IAAI,eAAe;IAI5B,eAAe,IAAI,KAAK,CAAC;QAAE,SAAS,EAAE,IAAI,CAAC;QAAC,KAAK,EAAE,WAAW,CAAC;QAAC,MAAM,EAAE,YAAY,CAAA;KAAE,CAAC;IAIvF,iBAAiB,IAAI,IAAI;IAIzB,OAAO,CAAC,YAAY;CAiBrB"}
|
|
@@ -1,9 +1,9 @@
|
|
|
1
|
-
import { DEFAULT_CONSTITUTION } from './constitution';
|
|
2
|
-
import { InputFilter } from './input-filter';
|
|
3
|
-
import { OutputFilter } from './output-filter';
|
|
4
|
-
import { ToolGuard } from './tool-guard';
|
|
5
|
-
import { CritiqueReviser } from './critique-reviser';
|
|
6
|
-
import { getLogger } from '../logger';
|
|
1
|
+
import { DEFAULT_CONSTITUTION } from './constitution.js';
|
|
2
|
+
import { InputFilter } from './input-filter.js';
|
|
3
|
+
import { OutputFilter } from './output-filter.js';
|
|
4
|
+
import { ToolGuard } from './tool-guard.js';
|
|
5
|
+
import { CritiqueReviser } from './critique-reviser.js';
|
|
6
|
+
import { getLogger } from '../logger.js';
|
|
7
7
|
export class ConstitutionalAI {
|
|
8
8
|
inputFilter;
|
|
9
9
|
outputFilter;
|
|
@@ -65,7 +65,7 @@ export class ConstitutionalAI {
|
|
|
65
65
|
return this._constitution;
|
|
66
66
|
}
|
|
67
67
|
async filterInput(input, context) {
|
|
68
|
-
if (!this._config.filterInput) {
|
|
68
|
+
if (!this._config.enabled || !this._config.filterInput) {
|
|
69
69
|
return { allowed: true, harmScores: [] };
|
|
70
70
|
}
|
|
71
71
|
const result = await this.inputFilter.filter(input, context);
|
|
@@ -73,7 +73,7 @@ export class ConstitutionalAI {
|
|
|
73
73
|
return result;
|
|
74
74
|
}
|
|
75
75
|
async filterOutput(output, context) {
|
|
76
|
-
if (!this._config.filterOutput) {
|
|
76
|
+
if (!this._config.enabled || !this._config.filterOutput) {
|
|
77
77
|
return { allowed: true, harmScores: [] };
|
|
78
78
|
}
|
|
79
79
|
const result = await this.outputFilter.filter(output, context);
|
|
@@ -91,7 +91,7 @@ export class ConstitutionalAI {
|
|
|
91
91
|
return result;
|
|
92
92
|
}
|
|
93
93
|
async guardTool(tool, args, context) {
|
|
94
|
-
if (!this._config.filterToolCalls) {
|
|
94
|
+
if (!this._config.enabled || !this._config.filterToolCalls) {
|
|
95
95
|
return {
|
|
96
96
|
approved: true,
|
|
97
97
|
requiresConfirmation: false,
|
|
@@ -102,6 +102,14 @@ export class ConstitutionalAI {
|
|
|
102
102
|
return this.toolGuard.evaluate(tool, args, context);
|
|
103
103
|
}
|
|
104
104
|
async critiqueAndRevise(response, context) {
|
|
105
|
+
if (!this._config.enabled) {
|
|
106
|
+
return {
|
|
107
|
+
original: response,
|
|
108
|
+
revised: response,
|
|
109
|
+
iterations: 0,
|
|
110
|
+
critiqueHistory: [],
|
|
111
|
+
};
|
|
112
|
+
}
|
|
105
113
|
return this.critiqueReviser.critiqueAndRevise(response, context);
|
|
106
114
|
}
|
|
107
115
|
setConstitution(constitution) {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"constitutional-ai.js","sourceRoot":"","sources":["../../src/constitutional/constitutional-ai.ts"],"names":[],"mappings":"AAaA,OAAO,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAC;AACtD,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAC/C,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACzC,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAQtC,MAAM,OAAO,gBAAgB;IACnB,WAAW,CAAc;IACzB,YAAY,CAAe;IAC3B,SAAS,CAAY;IACrB,eAAe,CAAkB;IACjC,OAAO,CAAkB;IACzB,aAAa,CAAe;IAC5B,YAAY,GAAyE,EAAE,CAAC;IACxF,MAAM,GAAG,SAAS,EAAE,CAAC,KAAK,CAAC,EAAE,SAAS,EAAE,kBAAkB,EAAE,CAAC,CAAC;IAEtE,YAAY,OAAgC;QAC1C,MAAM,aAAa,GAAoB;YACrC,OAAO,EAAE,IAAI;YACb,WAAW,EAAE,IAAI;YACjB,YAAY,EAAE,IAAI;YAClB,eAAe,EAAE,IAAI;YACrB,iBAAiB,EAAE,KAAK;YACxB,sBAAsB,EAAE,IAAI;YAC5B,qBAAqB,EAAE,CAAC;YACxB,2BAA2B,EAAE,IAAI;YACjC,UAAU,EAAE;gBACV,QAAQ,EAAE,QAAQ;gBAClB,IAAI,EAAE,KAAK;gBACX,MAAM,EAAE,QAAQ;gBAChB,WAAW,EAAE,KAAK;gBAClB,OAAO,EAAE,KAAK;gBACd,OAAO,EAAE,QAAQ;gBACjB,cAAc,EAAE,MAAM;gBACtB,YAAY,EAAE,QAAQ;aACvB;YACD,UAAU,EAAE,KAAK;YACjB,aAAa,EAAE,IAAI;SACpB,CAAC;QAEF,IAAI,CAAC,OAAO,GAAG,EAAE,GAAG,aAAa,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;QACvD,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,YAAY,IAAI,oBAAoB,CAAC;QAElE,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,CAAC;YACjC,GAAG,EAAE,OAAO,CAAC,GAAG;YAChB,MAAM,EAAE,IAAI,CAAC,OAAO;YACpB,YAAY,EAAE,IAAI,CAAC,aAAa;SACjC,CAAC,CAAC;QAEH,IAAI,CAAC,YAAY,GAAG,IAAI,YAAY,CAAC;YACnC,GAAG,EAAE,OAAO,CAAC,GAAG;YAChB,MAAM,EAAE,IAAI,CAAC,OAAO;YACpB,YAAY,EAAE,IAAI,CAAC,aAAa;SACjC,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,GAAG,IAAI,SAAS,CAAC;YAC7B,MAAM,EAAE,IAAI,CAAC,OAAO;YACpB,YAAY,EAAE,IAAI,CAAC,aAAa;SACjC,CAAC,CAAC;QAEH,IAAI,CAAC,eAAe,GAAG,IAAI,eAAe,CAAC;YACzC,GAAG,EAAE,OAAO,CAAC,GAAG;YAChB,MAAM,EAAE,IAAI,CAAC,OAAO;YACpB,YAAY,EAAE,IAAI,CAAC,aAAa;SACjC,CAAC,CAAC;IACL,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAAa,EAAE,OAAgB;QAC/C,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;
|
|
1
|
+
{"version":3,"file":"constitutional-ai.js","sourceRoot":"","sources":["../../src/constitutional/constitutional-ai.ts"],"names":[],"mappings":"AAaA,OAAO,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAC;AACtD,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAC/C,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACzC,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAQtC,MAAM,OAAO,gBAAgB;IACnB,WAAW,CAAc;IACzB,YAAY,CAAe;IAC3B,SAAS,CAAY;IACrB,eAAe,CAAkB;IACjC,OAAO,CAAkB;IACzB,aAAa,CAAe;IAC5B,YAAY,GAAyE,EAAE,CAAC;IACxF,MAAM,GAAG,SAAS,EAAE,CAAC,KAAK,CAAC,EAAE,SAAS,EAAE,kBAAkB,EAAE,CAAC,CAAC;IAEtE,YAAY,OAAgC;QAC1C,MAAM,aAAa,GAAoB;YACrC,OAAO,EAAE,IAAI;YACb,WAAW,EAAE,IAAI;YACjB,YAAY,EAAE,IAAI;YAClB,eAAe,EAAE,IAAI;YACrB,iBAAiB,EAAE,KAAK;YACxB,sBAAsB,EAAE,IAAI;YAC5B,qBAAqB,EAAE,CAAC;YACxB,2BAA2B,EAAE,IAAI;YACjC,UAAU,EAAE;gBACV,QAAQ,EAAE,QAAQ;gBAClB,IAAI,EAAE,KAAK;gBACX,MAAM,EAAE,QAAQ;gBAChB,WAAW,EAAE,KAAK;gBAClB,OAAO,EAAE,KAAK;gBACd,OAAO,EAAE,QAAQ;gBACjB,cAAc,EAAE,MAAM;gBACtB,YAAY,EAAE,QAAQ;aACvB;YACD,UAAU,EAAE,KAAK;YACjB,aAAa,EAAE,IAAI;SACpB,CAAC;QAEF,IAAI,CAAC,OAAO,GAAG,EAAE,GAAG,aAAa,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;QACvD,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,YAAY,IAAI,oBAAoB,CAAC;QAElE,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,CAAC;YACjC,GAAG,EAAE,OAAO,CAAC,GAAG;YAChB,MAAM,EAAE,IAAI,CAAC,OAAO;YACpB,YAAY,EAAE,IAAI,CAAC,aAAa;SACjC,CAAC,CAAC;QAEH,IAAI,CAAC,YAAY,GAAG,IAAI,YAAY,CAAC;YACnC,GAAG,EAAE,OAAO,CAAC,GAAG;YAChB,MAAM,EAAE,IAAI,CAAC,OAAO;YACpB,YAAY,EAAE,IAAI,CAAC,aAAa;SACjC,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,GAAG,IAAI,SAAS,CAAC;YAC7B,MAAM,EAAE,IAAI,CAAC,OAAO;YACpB,YAAY,EAAE,IAAI,CAAC,aAAa;SACjC,CAAC,CAAC;QAEH,IAAI,CAAC,eAAe,GAAG,IAAI,eAAe,CAAC;YACzC,GAAG,EAAE,OAAO,CAAC,GAAG;YAChB,MAAM,EAAE,IAAI,CAAC,OAAO;YACpB,YAAY,EAAE,IAAI,CAAC,aAAa;SACjC,CAAC,CAAC;IACL,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAAa,EAAE,OAAgB;QAC/C,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;YACvD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;QAC3C,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAC7D,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACnC,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAc,EAAE,OAAkB;QACnD,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;YACxD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;QAC3C,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAE/D,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,sBAAsB,EAAE,CAAC;YAC3D,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAC/D,IAAI,QAAQ,CAAC,OAAO,KAAK,QAAQ,CAAC,QAAQ,EAAE,CAAC;gBAC3C,OAAO;oBACL,OAAO,EAAE,IAAI;oBACb,UAAU,EAAE,MAAM,CAAC,UAAU;oBAC7B,iBAAiB,EAAE,QAAQ,CAAC,OAAO;iBACpC,CAAC;YACJ,CAAC;QACH,CAAC;QAED,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QACpC,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,SAAS,CACb,IAAU,EACV,IAA6B,EAC7B,OAAoB;QAEpB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC;YAC3D,OAAO;gBACL,QAAQ,EAAE,IAAI;gBACd,oBAAoB,EAAE,KAAK;gBAC3B,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;gBACnC,SAAS,EAAE,KAAK;aACjB,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,QAAgB,EAAE,OAAkB;QAC1D,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YAC1B,OAAO;gBACL,QAAQ,EAAE,QAAQ;gBAClB,OAAO,EAAE,QAAQ;gBACjB,UAAU,EAAE,CAAC;gBACb,eAAe,EAAE,EAAE;aACpB,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACnE,CAAC;IAED,eAAe,CAAC,YAA0B;QACxC,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;QAClC,IAAI,CAAC,WAAW,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC;QAClD,IAAI,CAAC,YAAY,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC;QACnD,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC;QAChD,IAAI,CAAC,eAAe,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC;QACtD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,EAAE,EAAE,cAAc,EAAE,YAAY,CAAC,EAAE,EAAE,CAAC,CAAC;IAChF,CAAC;IAED,YAAY,CAAC,SAAkC;QAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,SAAS,CAAC,EAAE,CAAC,CAAC;QAChF,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,oCAAoC,EAAE,EAAE,WAAW,EAAE,SAAS,CAAC,EAAE,EAAE,CAAC,CAAC;YACtF,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAiB;YAC5B,GAAG,IAAI,CAAC,aAAa;YACrB,UAAU,EAAE,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,SAAS,CAAC;SAC1D,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;IAChC,CAAC;IAED,eAAe,CAAC,EAAU;QACxB,MAAM,OAAO,GAAiB;YAC5B,GAAG,IAAI,CAAC,aAAa;YACrB,UAAU,EAAE,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC;SACrE,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;IAChC,CAAC;IAED,eAAe;QACb,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,SAAS;QACP,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;IAC7B,CAAC;IAED,eAAe;QACb,OAAO,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC;IAChC,CAAC;IAED,iBAAiB;QACf,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;IACzB,CAAC;IAEO,YAAY,CAAC,KAAkB,EAAE,MAAoB;QAC3D,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpD,IAAI,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;gBAC/B,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;gBACjE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,8BAA8B,EAAE;oBAC/C,KAAK;oBACL,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,SAAS,EAAE,MAAM,CAAC,UAAU,CAAC,MAAM;oBACnC,UAAU,EAAE,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC;iBACrD,CAAC,CAAC;YACL,CAAC;YAED,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;gBAC7B,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC;IACH,CAAC;CACF"}
|
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import { buildCritiquePrompt, buildRevisionPrompt, parseCritiqueResponse } from './prompts';
|
|
2
|
-
import { filterPrinciplesByLayer } from './constitution';
|
|
1
|
+
import { buildCritiquePrompt, buildRevisionPrompt, parseCritiqueResponse } from './prompts.js';
|
|
2
|
+
import { filterPrinciplesByLayer } from './constitution.js';
|
|
3
3
|
export class CritiqueReviser {
|
|
4
4
|
llm;
|
|
5
5
|
config;
|
|
@@ -1,13 +1,13 @@
|
|
|
1
|
-
export { ConstitutionalAI } from './constitutional-ai';
|
|
2
|
-
export type { ConstitutionalAIOptions } from './constitutional-ai';
|
|
3
|
-
export { InputFilter } from './input-filter';
|
|
4
|
-
export type { InputFilterOptions } from './input-filter';
|
|
5
|
-
export { OutputFilter } from './output-filter';
|
|
6
|
-
export type { OutputFilterOptions } from './output-filter';
|
|
7
|
-
export { ToolGuard } from './tool-guard';
|
|
8
|
-
export type { ToolGuardOptions } from './tool-guard';
|
|
9
|
-
export { CritiqueReviser } from './critique-reviser';
|
|
10
|
-
export type { CritiqueReviserOptions } from './critique-reviser';
|
|
11
|
-
export { DEFAULT_CONSTITUTION, DEFAULT_PRINCIPLES, createConstitution, extendConstitution, filterPrinciplesByLayer, getPrinciplesByCategory, getPrinciplesBySeverity, } from './constitution';
|
|
12
|
-
export { buildInputEvaluationPrompt, buildOutputEvaluationPrompt, buildCritiquePrompt, buildRevisionPrompt, parseEvaluationResponse, parseCritiqueResponse, } from './prompts';
|
|
1
|
+
export { ConstitutionalAI } from './constitutional-ai.js';
|
|
2
|
+
export type { ConstitutionalAIOptions } from './constitutional-ai.js';
|
|
3
|
+
export { InputFilter } from './input-filter.js';
|
|
4
|
+
export type { InputFilterOptions } from './input-filter.js';
|
|
5
|
+
export { OutputFilter } from './output-filter.js';
|
|
6
|
+
export type { OutputFilterOptions } from './output-filter.js';
|
|
7
|
+
export { ToolGuard } from './tool-guard.js';
|
|
8
|
+
export type { ToolGuardOptions } from './tool-guard.js';
|
|
9
|
+
export { CritiqueReviser } from './critique-reviser.js';
|
|
10
|
+
export type { CritiqueReviserOptions } from './critique-reviser.js';
|
|
11
|
+
export { DEFAULT_CONSTITUTION, DEFAULT_PRINCIPLES, createConstitution, extendConstitution, filterPrinciplesByLayer, getPrinciplesByCategory, getPrinciplesBySeverity, } from './constitution.js';
|
|
12
|
+
export { buildInputEvaluationPrompt, buildOutputEvaluationPrompt, buildCritiquePrompt, buildRevisionPrompt, parseEvaluationResponse, parseCritiqueResponse, } from './prompts.js';
|
|
13
13
|
//# sourceMappingURL=index.d.ts.map
|
|
@@ -1,8 +1,8 @@
|
|
|
1
|
-
export { ConstitutionalAI } from './constitutional-ai';
|
|
2
|
-
export { InputFilter } from './input-filter';
|
|
3
|
-
export { OutputFilter } from './output-filter';
|
|
4
|
-
export { ToolGuard } from './tool-guard';
|
|
5
|
-
export { CritiqueReviser } from './critique-reviser';
|
|
6
|
-
export { DEFAULT_CONSTITUTION, DEFAULT_PRINCIPLES, createConstitution, extendConstitution, filterPrinciplesByLayer, getPrinciplesByCategory, getPrinciplesBySeverity, } from './constitution';
|
|
7
|
-
export { buildInputEvaluationPrompt, buildOutputEvaluationPrompt, buildCritiquePrompt, buildRevisionPrompt, parseEvaluationResponse, parseCritiqueResponse, } from './prompts';
|
|
1
|
+
export { ConstitutionalAI } from './constitutional-ai.js';
|
|
2
|
+
export { InputFilter } from './input-filter.js';
|
|
3
|
+
export { OutputFilter } from './output-filter.js';
|
|
4
|
+
export { ToolGuard } from './tool-guard.js';
|
|
5
|
+
export { CritiqueReviser } from './critique-reviser.js';
|
|
6
|
+
export { DEFAULT_CONSTITUTION, DEFAULT_PRINCIPLES, createConstitution, extendConstitution, filterPrinciplesByLayer, getPrinciplesByCategory, getPrinciplesBySeverity, } from './constitution.js';
|
|
7
|
+
export { buildInputEvaluationPrompt, buildOutputEvaluationPrompt, buildCritiquePrompt, buildRevisionPrompt, parseEvaluationResponse, parseCritiqueResponse, } from './prompts.js';
|
|
8
8
|
//# sourceMappingURL=index.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"input-filter.d.ts","sourceRoot":"","sources":["../../src/constitutional/input-filter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,eAAe,EACf,YAAY,EAGZ,YAAY,EACZ,UAAU,EACX,MAAM,qBAAqB,CAAC;AAI7B,MAAM,WAAW,kBAAkB;IACjC,GAAG,EAAE,UAAU,CAAC;IAChB,MAAM,EAAE,eAAe,CAAC;IACxB,YAAY,EAAE,YAAY,CAAC;CAC5B;AAED,qBAAa,WAAW;IACtB,OAAO,CAAC,GAAG,CAAa;IACxB,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,YAAY,CAAe;IACnC,OAAO,CAAC,UAAU,CAA4B;gBAElC,OAAO,EAAE,kBAAkB;IAOjC,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAcpE,OAAO,CAAC,SAAS;YA8BH,eAAe;
|
|
1
|
+
{"version":3,"file":"input-filter.d.ts","sourceRoot":"","sources":["../../src/constitutional/input-filter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,eAAe,EACf,YAAY,EAGZ,YAAY,EACZ,UAAU,EACX,MAAM,qBAAqB,CAAC;AAI7B,MAAM,WAAW,kBAAkB;IACjC,GAAG,EAAE,UAAU,CAAC;IAChB,MAAM,EAAE,eAAe,CAAC;IACxB,YAAY,EAAE,YAAY,CAAC;CAC5B;AAED,qBAAa,WAAW;IACtB,OAAO,CAAC,GAAG,CAAa;IACxB,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,YAAY,CAAe;IACnC,OAAO,CAAC,UAAU,CAA4B;gBAElC,OAAO,EAAE,kBAAkB;IAOjC,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAcpE,OAAO,CAAC,SAAS;YA8BH,eAAe;IAwB7B,OAAO,CAAC,eAAe;IAqBvB,kBAAkB,CAAC,YAAY,EAAE,YAAY,GAAG,IAAI;CAIrD"}
|
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import { buildInputEvaluationPrompt, parseEvaluationResponse } from './prompts';
|
|
2
|
-
import { filterPrinciplesByLayer } from './constitution';
|
|
1
|
+
import { buildInputEvaluationPrompt, parseEvaluationResponse } from './prompts.js';
|
|
2
|
+
import { filterPrinciplesByLayer } from './constitution.js';
|
|
3
3
|
export class InputFilter {
|
|
4
4
|
llm;
|
|
5
5
|
config;
|
|
@@ -58,6 +58,16 @@ export class InputFilter {
|
|
|
58
58
|
maxTokens: 500,
|
|
59
59
|
});
|
|
60
60
|
const result = parseEvaluationResponse(response.content);
|
|
61
|
+
if (result.isHarmful && result.harmScores.length === 0) {
|
|
62
|
+
return [
|
|
63
|
+
{
|
|
64
|
+
category: 'manipulation',
|
|
65
|
+
severity: 'high',
|
|
66
|
+
confidence: 1,
|
|
67
|
+
reasoning: 'Safety evaluator marked input as harmful without category details',
|
|
68
|
+
},
|
|
69
|
+
];
|
|
70
|
+
}
|
|
61
71
|
return result.harmScores;
|
|
62
72
|
}
|
|
63
73
|
applyThresholds(harmScores) {
|
|
@@ -70,15 +80,12 @@ export class InputFilter {
|
|
|
70
80
|
if (violations.length === 0) {
|
|
71
81
|
return { allowed: true, harmScores };
|
|
72
82
|
}
|
|
73
|
-
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
|
|
79
|
-
};
|
|
80
|
-
}
|
|
81
|
-
return { allowed: true, harmScores };
|
|
83
|
+
const categories = [...new Set(violations.map((v) => v.category))];
|
|
84
|
+
return {
|
|
85
|
+
allowed: false,
|
|
86
|
+
harmScores,
|
|
87
|
+
blockedReason: `Input violates safety policies: ${categories.join(', ')}`,
|
|
88
|
+
};
|
|
82
89
|
}
|
|
83
90
|
updateConstitution(constitution) {
|
|
84
91
|
this.constitution = constitution;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"input-filter.js","sourceRoot":"","sources":["../../src/constitutional/input-filter.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,0BAA0B,EAAE,uBAAuB,EAAE,MAAM,WAAW,CAAC;AAChF,OAAO,EAAE,uBAAuB,EAAE,MAAM,gBAAgB,CAAC;AAQzD,MAAM,OAAO,WAAW;IACd,GAAG,CAAa;IAChB,MAAM,CAAkB;IACxB,YAAY,CAAe;IAC3B,UAAU,CAA4B;IAE9C,YAAY,OAA2B;QACrC,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;QACzC,IAAI,CAAC,UAAU,GAAG,uBAAuB,CAAC,IAAI,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;IACxE,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,OAAgB;QAC1C,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC1C,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;YACzB,OAAO,WAAW,CAAC;QACrB,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;QAC3C,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAC9D,OAAO,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;IAC1C,CAAC;IAEO,SAAS,CAAC,KAAa;QAC7B,MAAM,OAAO,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;QACpC,MAAM,QAAQ,GAAG;YACf,EAAE,OAAO,EAAE,+CAA+C,EAAE,QAAQ,EAAE,UAAmB,EAAE;YAC3F,EAAE,OAAO,EAAE,qCAAqC,EAAE,QAAQ,EAAE,UAAmB,EAAE;YACjF,EAAE,OAAO,EAAE,6BAA6B,EAAE,QAAQ,EAAE,SAAkB,EAAE;YACxE,EAAE,OAAO,EAAE,yBAAyB,EAAE,QAAQ,EAAE,QAAiB,EAAE;YACnE,EAAE,OAAO,EAAE,+BAA+B,EAAE,QAAQ,EAAE,WAAoB,EAAE;SAC7E,CAAC;QAEF,KAAK,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,QAAQ,EAAE,CAAC;YAC7C,IAAI,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC1B,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,UAAU,EAAE;wBACV;4BACE,QAAQ;4BACR,QAAQ,EAAE,MAAM;4BAChB,UAAU,EAAE,IAAI;4BAChB,SAAS,EAAE,kCAAkC;yBAC9C;qBACF;oBACD,aAAa,EAAE,+CAA+C,QAAQ,GAAG;iBAC1E,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;IAC3C,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,KAAa,EAAE,QAAiB;QAC5D,MAAM,MAAM,GAAG,0BAA0B,CAAC,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QAElE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC;YACnC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,aAAa;YACzC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;YAC7C,WAAW,EAAE,CAAC;YACd,SAAS,EAAE,GAAG;SACf,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,uBAAuB,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QACzD,OAAO,MAAM,CAAC,UAAU,CAAC;IAC3B,CAAC;IAEO,eAAe,CAAC,UAAuB;QAC7C,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;QAC1C,MAAM,aAAa,GAA2B,EAAE,GAAG,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;QAE7E,MAAM,UAAU,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YAC7C,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,MAAM,CAAC;YACvD,OAAO,aAAa,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,aAAa,CAAC,SAAS,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;QACvC,CAAC;QAED,
|
|
1
|
+
{"version":3,"file":"input-filter.js","sourceRoot":"","sources":["../../src/constitutional/input-filter.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,0BAA0B,EAAE,uBAAuB,EAAE,MAAM,WAAW,CAAC;AAChF,OAAO,EAAE,uBAAuB,EAAE,MAAM,gBAAgB,CAAC;AAQzD,MAAM,OAAO,WAAW;IACd,GAAG,CAAa;IAChB,MAAM,CAAkB;IACxB,YAAY,CAAe;IAC3B,UAAU,CAA4B;IAE9C,YAAY,OAA2B;QACrC,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;QACzC,IAAI,CAAC,UAAU,GAAG,uBAAuB,CAAC,IAAI,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;IACxE,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,OAAgB;QAC1C,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC1C,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;YACzB,OAAO,WAAW,CAAC;QACrB,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;QAC3C,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAC9D,OAAO,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;IAC1C,CAAC;IAEO,SAAS,CAAC,KAAa;QAC7B,MAAM,OAAO,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;QACpC,MAAM,QAAQ,GAAG;YACf,EAAE,OAAO,EAAE,+CAA+C,EAAE,QAAQ,EAAE,UAAmB,EAAE;YAC3F,EAAE,OAAO,EAAE,qCAAqC,EAAE,QAAQ,EAAE,UAAmB,EAAE;YACjF,EAAE,OAAO,EAAE,6BAA6B,EAAE,QAAQ,EAAE,SAAkB,EAAE;YACxE,EAAE,OAAO,EAAE,yBAAyB,EAAE,QAAQ,EAAE,QAAiB,EAAE;YACnE,EAAE,OAAO,EAAE,+BAA+B,EAAE,QAAQ,EAAE,WAAoB,EAAE;SAC7E,CAAC;QAEF,KAAK,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,QAAQ,EAAE,CAAC;YAC7C,IAAI,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC1B,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,UAAU,EAAE;wBACV;4BACE,QAAQ;4BACR,QAAQ,EAAE,MAAM;4BAChB,UAAU,EAAE,IAAI;4BAChB,SAAS,EAAE,kCAAkC;yBAC9C;qBACF;oBACD,aAAa,EAAE,+CAA+C,QAAQ,GAAG;iBAC1E,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;IAC3C,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,KAAa,EAAE,QAAiB;QAC5D,MAAM,MAAM,GAAG,0BAA0B,CAAC,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QAElE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC;YACnC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,aAAa;YACzC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;YAC7C,WAAW,EAAE,CAAC;YACd,SAAS,EAAE,GAAG;SACf,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,uBAAuB,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QACzD,IAAI,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvD,OAAO;gBACL;oBACE,QAAQ,EAAE,cAAc;oBACxB,QAAQ,EAAE,MAAM;oBAChB,UAAU,EAAE,CAAC;oBACb,SAAS,EAAE,mEAAmE;iBAC/E;aACF,CAAC;QACJ,CAAC;QACD,OAAO,MAAM,CAAC,UAAU,CAAC;IAC3B,CAAC;IAEO,eAAe,CAAC,UAAuB;QAC7C,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;QAC1C,MAAM,aAAa,GAA2B,EAAE,GAAG,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;QAE7E,MAAM,UAAU,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YAC7C,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,MAAM,CAAC;YACvD,OAAO,aAAa,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,aAAa,CAAC,SAAS,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;QACvC,CAAC;QAED,MAAM,UAAU,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QACnE,OAAO;YACL,OAAO,EAAE,KAAK;YACd,UAAU;YACV,aAAa,EAAE,mCAAmC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;SAC1E,CAAC;IACJ,CAAC;IAED,kBAAkB,CAAC,YAA0B;QAC3C,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,UAAU,GAAG,uBAAuB,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;IACnE,CAAC;CACF"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"output-filter.d.ts","sourceRoot":"","sources":["../../src/constitutional/output-filter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,eAAe,EACf,YAAY,EAGZ,YAAY,EACZ,UAAU,EACV,OAAO,EACR,MAAM,qBAAqB,CAAC;AAI7B,MAAM,WAAW,mBAAmB;IAClC,GAAG,EAAE,UAAU,CAAC;IAChB,MAAM,EAAE,eAAe,CAAC;IACxB,YAAY,EAAE,YAAY,CAAC;CAC5B;AAED,qBAAa,YAAY;IACvB,OAAO,CAAC,GAAG,CAAa;IACxB,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,YAAY,CAAe;IACnC,OAAO,CAAC,UAAU,CAA4B;gBAElC,OAAO,EAAE,mBAAmB;IAOlC,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,YAAY,CAAC;IAcvE,OAAO,CAAC,SAAS;YA6BH,eAAe;
|
|
1
|
+
{"version":3,"file":"output-filter.d.ts","sourceRoot":"","sources":["../../src/constitutional/output-filter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,eAAe,EACf,YAAY,EAGZ,YAAY,EACZ,UAAU,EACV,OAAO,EACR,MAAM,qBAAqB,CAAC;AAI7B,MAAM,WAAW,mBAAmB;IAClC,GAAG,EAAE,UAAU,CAAC;IAChB,MAAM,EAAE,eAAe,CAAC;IACxB,YAAY,EAAE,YAAY,CAAC;CAC5B;AAED,qBAAa,YAAY;IACvB,OAAO,CAAC,GAAG,CAAa;IACxB,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,YAAY,CAAe;IACnC,OAAO,CAAC,UAAU,CAA4B;gBAElC,OAAO,EAAE,mBAAmB;IAOlC,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,YAAY,CAAC;IAcvE,OAAO,CAAC,SAAS;YA6BH,eAAe;IAwB7B,OAAO,CAAC,eAAe;IAqBvB,kBAAkB,CAAC,YAAY,EAAE,YAAY,GAAG,IAAI;CAIrD"}
|
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import { buildOutputEvaluationPrompt, parseEvaluationResponse } from './prompts';
|
|
2
|
-
import { filterPrinciplesByLayer } from './constitution';
|
|
1
|
+
import { buildOutputEvaluationPrompt, parseEvaluationResponse } from './prompts.js';
|
|
2
|
+
import { filterPrinciplesByLayer } from './constitution.js';
|
|
3
3
|
export class OutputFilter {
|
|
4
4
|
llm;
|
|
5
5
|
config;
|
|
@@ -56,6 +56,16 @@ export class OutputFilter {
|
|
|
56
56
|
maxTokens: 500,
|
|
57
57
|
});
|
|
58
58
|
const result = parseEvaluationResponse(response.content);
|
|
59
|
+
if (result.isHarmful && result.harmScores.length === 0) {
|
|
60
|
+
return [
|
|
61
|
+
{
|
|
62
|
+
category: 'manipulation',
|
|
63
|
+
severity: 'high',
|
|
64
|
+
confidence: 1,
|
|
65
|
+
reasoning: 'Safety evaluator marked output as harmful without category details',
|
|
66
|
+
},
|
|
67
|
+
];
|
|
68
|
+
}
|
|
59
69
|
return result.harmScores;
|
|
60
70
|
}
|
|
61
71
|
applyThresholds(harmScores) {
|
|
@@ -68,15 +78,12 @@ export class OutputFilter {
|
|
|
68
78
|
if (violations.length === 0) {
|
|
69
79
|
return { allowed: true, harmScores };
|
|
70
80
|
}
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
};
|
|
78
|
-
}
|
|
79
|
-
return { allowed: true, harmScores };
|
|
81
|
+
const categories = [...new Set(violations.map((v) => v.category))];
|
|
82
|
+
return {
|
|
83
|
+
allowed: false,
|
|
84
|
+
harmScores,
|
|
85
|
+
blockedReason: `Output violates safety policies: ${categories.join(', ')}`,
|
|
86
|
+
};
|
|
80
87
|
}
|
|
81
88
|
updateConstitution(constitution) {
|
|
82
89
|
this.constitution = constitution;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"output-filter.js","sourceRoot":"","sources":["../../src/constitutional/output-filter.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,MAAM,WAAW,CAAC;AACjF,OAAO,EAAE,uBAAuB,EAAE,MAAM,gBAAgB,CAAC;AAQzD,MAAM,OAAO,YAAY;IACf,GAAG,CAAa;IAChB,MAAM,CAAkB;IACxB,YAAY,CAAe;IAC3B,UAAU,CAA4B;IAE9C,YAAY,OAA4B;QACtC,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;QACzC,IAAI,CAAC,UAAU,GAAG,uBAAuB,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;IACzE,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAc,EAAE,OAAkB;QAC7C,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC3C,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;YACzB,OAAO,WAAW,CAAC;QACrB,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;QAC3C,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC/D,OAAO,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;IAC1C,CAAC;IAEO,SAAS,CAAC,MAAc;QAC9B,MAAM,OAAO,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC;QAErC,MAAM,QAAQ,GAAG;YACf,EAAE,OAAO,EAAE,sBAAsB,EAAE,QAAQ,EAAE,SAAkB,EAAE;YACjE,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,SAAkB,EAAE;YACzD,EAAE,OAAO,EAAE,kBAAkB,EAAE,QAAQ,EAAE,SAAkB,EAAE;SAC9D,CAAC;QAEF,KAAK,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,QAAQ,EAAE,CAAC;YAC7C,IAAI,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC1B,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,UAAU,EAAE;wBACV;4BACE,QAAQ;4BACR,QAAQ,EAAE,MAAM;4BAChB,UAAU,EAAE,IAAI;4BAChB,SAAS,EAAE,mCAAmC;yBAC/C;qBACF;oBACD,aAAa,EAAE,gDAAgD;iBAChE,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;IAC3C,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,MAAc,EAAE,OAAkB;QAC9D,MAAM,MAAM,GAAG,2BAA2B,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QAE7E,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC;YACnC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,aAAa;YACzC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;YAC7C,WAAW,EAAE,CAAC;YACd,SAAS,EAAE,GAAG;SACf,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,uBAAuB,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QACzD,OAAO,MAAM,CAAC,UAAU,CAAC;IAC3B,CAAC;IAEO,eAAe,CAAC,UAAuB;QAC7C,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;QAC1C,MAAM,aAAa,GAA2B,EAAE,GAAG,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;QAE7E,MAAM,UAAU,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YAC7C,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,MAAM,CAAC;YACvD,OAAO,aAAa,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,aAAa,CAAC,SAAS,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;QACvC,CAAC;QAED,
|
|
1
|
+
{"version":3,"file":"output-filter.js","sourceRoot":"","sources":["../../src/constitutional/output-filter.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,MAAM,WAAW,CAAC;AACjF,OAAO,EAAE,uBAAuB,EAAE,MAAM,gBAAgB,CAAC;AAQzD,MAAM,OAAO,YAAY;IACf,GAAG,CAAa;IAChB,MAAM,CAAkB;IACxB,YAAY,CAAe;IAC3B,UAAU,CAA4B;IAE9C,YAAY,OAA4B;QACtC,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;QACzC,IAAI,CAAC,UAAU,GAAG,uBAAuB,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;IACzE,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAc,EAAE,OAAkB;QAC7C,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC3C,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;YACzB,OAAO,WAAW,CAAC;QACrB,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;QAC3C,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC/D,OAAO,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;IAC1C,CAAC;IAEO,SAAS,CAAC,MAAc;QAC9B,MAAM,OAAO,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC;QAErC,MAAM,QAAQ,GAAG;YACf,EAAE,OAAO,EAAE,sBAAsB,EAAE,QAAQ,EAAE,SAAkB,EAAE;YACjE,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,SAAkB,EAAE;YACzD,EAAE,OAAO,EAAE,kBAAkB,EAAE,QAAQ,EAAE,SAAkB,EAAE;SAC9D,CAAC;QAEF,KAAK,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,QAAQ,EAAE,CAAC;YAC7C,IAAI,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC1B,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,UAAU,EAAE;wBACV;4BACE,QAAQ;4BACR,QAAQ,EAAE,MAAM;4BAChB,UAAU,EAAE,IAAI;4BAChB,SAAS,EAAE,mCAAmC;yBAC/C;qBACF;oBACD,aAAa,EAAE,gDAAgD;iBAChE,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;IAC3C,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,MAAc,EAAE,OAAkB;QAC9D,MAAM,MAAM,GAAG,2BAA2B,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QAE7E,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC;YACnC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,aAAa;YACzC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;YAC7C,WAAW,EAAE,CAAC;YACd,SAAS,EAAE,GAAG;SACf,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,uBAAuB,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QACzD,IAAI,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvD,OAAO;gBACL;oBACE,QAAQ,EAAE,cAAc;oBACxB,QAAQ,EAAE,MAAM;oBAChB,UAAU,EAAE,CAAC;oBACb,SAAS,EAAE,oEAAoE;iBAChF;aACF,CAAC;QACJ,CAAC;QACD,OAAO,MAAM,CAAC,UAAU,CAAC;IAC3B,CAAC;IAEO,eAAe,CAAC,UAAuB;QAC7C,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;QAC1C,MAAM,aAAa,GAA2B,EAAE,GAAG,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;QAE7E,MAAM,UAAU,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YAC7C,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,MAAM,CAAC;YACvD,OAAO,aAAa,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,aAAa,CAAC,SAAS,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;QACvC,CAAC;QAED,MAAM,UAAU,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QACnE,OAAO;YACL,OAAO,EAAE,KAAK;YACd,UAAU;YACV,aAAa,EAAE,oCAAoC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;SAC3E,CAAC;IACJ,CAAC;IAED,kBAAkB,CAAC,YAA0B;QAC3C,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,UAAU,GAAG,uBAAuB,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;IACpE,CAAC;CACF"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"context-manager.d.ts","sourceRoot":"","sources":["../../src/context/context-manager.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,OAAO,EACP,UAAU,EACV,oBAAoB,EACpB,iBAAiB,EACjB,YAAY,EAGb,MAAM,qBAAqB,CAAC;AAW7B,MAAM,WAAW,kBAAkB;IACjC,UAAU,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,UAAU,CAAC;CAC5C;AAED,qBAAa,cAAc;IACzB,OAAO,CAAC,MAAM,
|
|
1
|
+
{"version":3,"file":"context-manager.d.ts","sourceRoot":"","sources":["../../src/context/context-manager.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,OAAO,EACP,UAAU,EACV,oBAAoB,EACpB,iBAAiB,EACjB,YAAY,EAGb,MAAM,qBAAqB,CAAC;AAW7B,MAAM,WAAW,kBAAkB;IACjC,UAAU,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,UAAU,CAAC;CAC5C;AAED,qBAAa,cAAc;IACzB,OAAO,CAAC,MAAM,CAAwD;IACtE,OAAO,CAAC,QAAQ,CAA6B;IAC7C,OAAO,CAAC,IAAI,CAAqB;gBAErB,MAAM,EAAE,oBAAoB,EAAE,IAAI,GAAE,kBAAuB;IAcvE,OAAO,CAAC,cAAc;IActB,oBAAoB,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM;IA2BjD,UAAU,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,WAAW,EAAE,MAAM,GAAG,YAAY;IAmBlE,cAAc,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,WAAW,EAAE,MAAM,GAAG,OAAO;IAM3D,QAAQ,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,CAAC;CAiCrF"}
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import { getModel } from '@cogitator-ai/models';
|
|
2
|
-
import { TruncateStrategy, SlidingWindowStrategy, SummarizeStrategy, HybridStrategy, countMessagesTokens, } from './strategies/index';
|
|
3
|
-
import { parseModel } from '../llm/index';
|
|
2
|
+
import { TruncateStrategy, SlidingWindowStrategy, SummarizeStrategy, HybridStrategy, countMessagesTokens, } from './strategies/index.js';
|
|
3
|
+
import { parseModel } from '../llm/index.js';
|
|
4
4
|
export class ContextManager {
|
|
5
5
|
config;
|
|
6
6
|
strategy;
|
|
@@ -13,7 +13,6 @@ export class ContextManager {
|
|
|
13
13
|
outputReserve: config.outputReserve ?? 0.15,
|
|
14
14
|
summaryModel: config.summaryModel ?? '',
|
|
15
15
|
windowSize: config.windowSize ?? 10,
|
|
16
|
-
windowOverlap: config.windowOverlap ?? 2,
|
|
17
16
|
};
|
|
18
17
|
this.deps = deps;
|
|
19
18
|
this.strategy = this.createStrategy(this.config.strategy);
|
|
@@ -84,7 +83,7 @@ export class ContextManager {
|
|
|
84
83
|
}
|
|
85
84
|
async compress(messages, modelString) {
|
|
86
85
|
const state = this.checkState(messages, modelString);
|
|
87
|
-
if (!state.needsCompression) {
|
|
86
|
+
if (!this.config.enabled || !state.needsCompression) {
|
|
88
87
|
return {
|
|
89
88
|
messages,
|
|
90
89
|
originalTokens: state.currentTokens,
|