fix: test files

fix: enable opamp for llmpricing
2026-05-13 13:40:30 +01:00 · 2026-05-13 15:01:41 +05:30 · 2026-05-13 14:58:33 +05:30
6 changed files with 3 additions and 7 deletions
--- a/ee/query-service/app/server.go
+++ b/ee/query-service/app/server.go
@@ -114,7 +114,7 @@ func NewServer(config signoz.Config, signoz *signoz.SigNoz) (*Server, error) {
 	// initiate agent config handler
 	agentConfMgr, err := agentConf.Initiate(&agentConf.ManagerOptions{
 		Store:         signoz.SQLStore,
-		AgentFeatures: []agentConf.AgentFeature{logParsingPipelineController},
+		AgentFeatures: []agentConf.AgentFeature{logParsingPipelineController, signoz.Modules.LLMPricingRule},
 	})
 	if err != nil {
 		return nil, err
--- a/pkg/query-service/app/server.go
+++ b/pkg/query-service/app/server.go
@@ -135,6 +135,7 @@ func NewServer(config signoz.Config, signoz *signoz.SigNoz) (*Server, error) {
 			Store: signoz.SQLStore,
 			AgentFeatures: []agentConf.AgentFeature{
 				logParsingPipelineController,
+				signoz.Modules.LLMPricingRule,
 			},
 		},
 	)
--- a/pkg/types/llmpricingruletypes/processorconfig.go
+++ b/pkg/types/llmpricingruletypes/processorconfig.go
@@ -26,9 +26,8 @@ type LLMPricingRuleProcessorAttrs struct {
 	CacheWrite string `yaml:"cache_write" json:"cache_write"`
 }

-// LLMPricingRuleProcessorDefaultPricing holds the pricing unit and the list of model-specific rules.
+// LLMPricingRuleProcessorDefaultPricing holds the list of model-specific rules.
 type LLMPricingRuleProcessorDefaultPricing struct {
-	Unit  string                    `yaml:"unit" json:"unit"`
 	Rules []LLMPricingRuleProcessor `yaml:"rules" json:"rules"`
 }

@@ -87,7 +86,6 @@ func buildProcessorConfig(rules []*LLMPricingRule) *LLMPricingRuleProcessorConfi
 			CacheWrite: GenAIUsageCacheCreationInputTokens,
 		},
 		DefaultPricing: LLMPricingRuleProcessorDefaultPricing{
-			Unit:  UnitPerMillionTokens.StringValue(),
 			Rules: pricingRules,
 		},
 		OutputAttrs: LLMPricingRuleProcessorOutputAttrs{
--- a/pkg/types/llmpricingruletypes/testdata/collector_baseline.yaml
+++ b/pkg/types/llmpricingruletypes/testdata/collector_baseline.yaml
@@ -11,7 +11,6 @@ processors:
      cache_read: gen_ai.usage.cache_read.input_tokens
      cache_write: gen_ai.usage.cache_creation.input_tokens
    default_pricing:
-      unit: per_million_tokens
      rules: []
    output_attrs:
      in: _signoz.gen_ai.cost_input
--- a/pkg/types/llmpricingruletypes/testdata/collector_no_rules.yaml
+++ b/pkg/types/llmpricingruletypes/testdata/collector_no_rules.yaml
@@ -11,7 +11,6 @@ processors:
            cache_read: gen_ai.usage.cache_read.input_tokens
            cache_write: gen_ai.usage.cache_creation.input_tokens
        default_pricing:
-            unit: per_million_tokens
            rules: []
        output_attrs:
            in: _signoz.gen_ai.cost_input
--- a/pkg/types/llmpricingruletypes/testdata/collector_with_rule.yaml
+++ b/pkg/types/llmpricingruletypes/testdata/collector_with_rule.yaml
@@ -11,7 +11,6 @@ processors:
            cache_read: gen_ai.usage.cache_read.input_tokens
            cache_write: gen_ai.usage.cache_creation.input_tokens
        default_pricing:
-            unit: per_million_tokens
            rules:
                - name: gpt-4o
                  pattern:
Author	SHA1	Message	Date
nityanandagohain	f48ea7f406	fix: test files	2026-05-13 15:01:41 +05:30
nityanandagohain	dfa49b7bd7	fix: enable opamp for llmpricing	2026-05-13 14:58:33 +05:30