Fix/get multiple flows working (#355)

* Reduce log output

* Fix problems
This commit is contained in:
cybermaggedon 2025-04-29 00:06:41 +01:00 committed by GitHub
parent 9508ac6c69
commit 027b52cd7c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 19 additions and 10 deletions

View file

@ -72,7 +72,7 @@ class Processor(FlowProcessor):
for k, v in config.items()
}
def get_prices(self, prices, modelname):
def get_prices(self, modelname):
if modelname in self.prices:
model = self.prices[modelname]
@ -90,9 +90,7 @@ class Processor(FlowProcessor):
__class__.input_token_metric.inc(num_in)
__class__.output_token_metric.inc(num_out)
model_input_price, model_output_price = self.get_prices(
price_list, modelname
)
model_input_price, model_output_price = self.get_prices(modelname)
if model_input_price == None:
cost_per_call = f"Model Not Found in Price list"