55302d18f0
The tracker-mapping and common-pattern-enrichment workers ran with the kit/worker defaults (interval 10s, max-concurrency 5 each) and dropped the resolved per-agent max-tokens/temperature, so up to ten LLM pipelines could run unbounded on one OpenAI client. The mapping worker also held a FOR UPDATE transaction across the LLM and Firecrawl calls while its DB search tools acquired a second pooled connection, risking pool exhaustion under concurrency. Plumb max-tokens, temperature, agent timeout, and per-worker max-turns through TrackerAgentsConfig and DisambiguationConfig into all three agent builders, replacing the hard-coded constants with config-fed fields and package fallbacks. Expose worker interval, concurrency, stale-after, agent timeout, and max-turns as config (env, Helm values, deployment template) mirroring the evidence-describer pattern, and apply them at registration. Refactor Process into deterministic-read, agent (no transaction), and persist phases so neither the mapping agent nor disambiguation runs inside an open transaction, removing the row locks held across network latency and the nested-connection pressure. Signed-off-by: Émile Ré <emile@probo.com>
81 lines
3.2 KiB
Go
81 lines
3.2 KiB
Go
// Copyright (c) 2026 Probo Inc <hello@getprobo.com>.
|
|
//
|
|
// Permission to use, copy, modify, and/or distribute this software for any
|
|
// purpose with or without fee is hereby granted, provided that the above
|
|
// copyright notice and this permission notice appear in all copies.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
|
|
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
|
|
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
|
|
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
|
|
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
|
|
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
|
|
// PERFORMANCE OF THIS SOFTWARE.
|
|
|
|
package probodconfig
|
|
|
|
type (
|
|
// FullConfig represents the complete configuration file structure.
|
|
// This is used by bootstrap to generate the YAML config file.
|
|
FullConfig struct {
|
|
Unit UnitConfig `json:"unit"`
|
|
Probod Config `json:"probod"`
|
|
}
|
|
|
|
// UnitConfig contains unit framework configuration.
|
|
UnitConfig struct {
|
|
Metrics MetricsConfig `json:"metrics"`
|
|
Tracing TracingConfig `json:"tracing"`
|
|
}
|
|
|
|
// MetricsConfig contains metrics server configuration.
|
|
MetricsConfig struct {
|
|
Addr string `json:"addr"`
|
|
}
|
|
|
|
// TracingConfig contains tracing configuration.
|
|
TracingConfig struct {
|
|
Addr string `json:"addr"`
|
|
MaxBatchSize int `json:"max-batch-size"`
|
|
BatchTimeout int `json:"batch-timeout"`
|
|
ExportTimeout int `json:"export-timeout"`
|
|
MaxQueueSize int `json:"max-queue-size"`
|
|
}
|
|
|
|
// ESignConfig contains electronic signature configuration.
|
|
ESignConfig struct {
|
|
TSAURL string `json:"tsa-url"`
|
|
}
|
|
|
|
// Config represents the probod application configuration.
|
|
Config struct {
|
|
BaseURL string `json:"base-url"`
|
|
EncryptionKey string `json:"encryption-key"`
|
|
Pg PgConfig `json:"pg"`
|
|
Api APIConfig `json:"api"`
|
|
Auth AuthConfig `json:"auth"`
|
|
TrustCenter TrustCenterConfig `json:"trust-center"`
|
|
AWS AWSConfig `json:"aws"`
|
|
Notifications NotificationsConfig `json:"notifications"`
|
|
Connectors []ConnectorConfig `json:"connectors"`
|
|
Agents AgentsConfig `json:"llm"`
|
|
EvidenceDescriber EvidenceDescriberConfig `json:"evidence-describer"`
|
|
|
|
TrackerMappingWorker TrackerMappingWorkerConfig `json:"tracker-mapping-worker"`
|
|
CommonPatternEnrichmentWorker CommonPatternEnrichmentWorkerConfig `json:"common-pattern-enrichment-worker"`
|
|
|
|
ChromeDPAddr string `json:"chrome-dp-addr"`
|
|
CustomDomains CustomDomainsConfig `json:"custom-domains"`
|
|
SCIMBridge SCIMBridgeConfig `json:"scim-bridge"`
|
|
ESign ESignConfig `json:"esign"`
|
|
Branding bool `json:"branding"`
|
|
}
|
|
|
|
// TrustCenterConfig contains trust center server configuration.
|
|
TrustCenterConfig struct {
|
|
HTTPAddr string `json:"http-addr"`
|
|
HTTPSAddr string `json:"https-addr"`
|
|
ProxyProtocol ProxyProtocolConfig `json:"proxy-protocol"`
|
|
}
|
|
)
|