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>
58 lines
2.7 KiB
Go
58 lines
2.7 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 probod
|
|
|
|
import "go.probo.inc/probo/pkg/probodconfig"
|
|
|
|
type (
|
|
FullConfig = probodconfig.FullConfig
|
|
Config = probodconfig.Config
|
|
UnitConfig = probodconfig.UnitConfig
|
|
MetricsConfig = probodconfig.MetricsConfig
|
|
TracingConfig = probodconfig.TracingConfig
|
|
ESignConfig = probodconfig.ESignConfig
|
|
TrustCenterConfig = probodconfig.TrustCenterConfig
|
|
APIConfig = probodconfig.APIConfig
|
|
CorsConfig = probodconfig.CorsConfig
|
|
ProxyProtocolConfig = probodconfig.ProxyProtocolConfig
|
|
AuthConfig = probodconfig.AuthConfig
|
|
OAuth2ServerConfig = probodconfig.OAuth2ServerConfig
|
|
OAuth2SigningKeyConfig = probodconfig.OAuth2SigningKeyConfig
|
|
CookieConfig = probodconfig.CookieConfig
|
|
PasswordConfig = probodconfig.PasswordConfig
|
|
AWSConfig = probodconfig.AWSConfig
|
|
ConnectorConfig = probodconfig.ConnectorConfig
|
|
ConnectorConfigOAuth2 = probodconfig.ConnectorConfigOAuth2
|
|
CustomDomainsConfig = probodconfig.CustomDomainsConfig
|
|
ACMEConfig = probodconfig.ACMEConfig
|
|
LLMProviderConfig = probodconfig.LLMProviderConfig
|
|
LLMAgentConfig = probodconfig.LLMAgentConfig
|
|
EvidenceDescriberConfig = probodconfig.EvidenceDescriberConfig
|
|
AgentsConfig = probodconfig.AgentsConfig
|
|
|
|
TrackerMappingWorkerConfig = probodconfig.TrackerMappingWorkerConfig
|
|
CommonPatternEnrichmentWorkerConfig = probodconfig.CommonPatternEnrichmentWorkerConfig
|
|
|
|
MailerConfig = probodconfig.MailerConfig
|
|
SMTPConfig = probodconfig.SMTPConfig
|
|
NotificationsConfig = probodconfig.NotificationsConfig
|
|
WebhookConfig = probodconfig.WebhookConfig
|
|
OIDCProviderConfig = probodconfig.OIDCProviderConfig
|
|
PgConfig = probodconfig.PgConfig
|
|
SAMLConfig = probodconfig.SAMLConfig
|
|
SCIMBridgeConfig = probodconfig.SCIMBridgeConfig
|
|
SlackConfig = probodconfig.SlackConfig
|
|
)
|