Move num_outputs pricing logic to base.py

daras_ai_v2/base.py

@@ -1838,7 +1838,7 @@ def get_price_roundoff(self, state: dict) -> int:
         return max(1, math.ceil(self.get_raw_price(state)))
 
     def get_raw_price(self, state: dict) -> float:
-        return self.price
+        return self.price * state.get("num_outputs", 1)
 
     @classmethod
     def get_example_preferred_fields(cls, state: dict) -> list[str]:

recipes/DocSummary.py

@@ -185,9 +185,6 @@ def run(self, state: dict) -> typing.Iterator[str | None]:
             case _:
                 raise NotImplementedError(f"{chain_type} not implemented")
 
-    def get_raw_price(self, state: dict) -> float:
-        return self.price * state.get("num_outputs", 1)
-
 
 MAP_REDUCE_PROMPT = """
 {documents}

recipes/GoogleGPT.py

@@ -279,6 +279,3 @@ def run_v2(
             max_tokens=request.max_tokens,
             avoid_repetition=request.avoid_repetition,
         )
-
-    def get_raw_price(self, state: dict) -> float:
-        return self.price * state.get("num_outputs", 1)

recipes/GoogleImageGen.py

@@ -238,6 +238,3 @@ def render_example(self, state: dict):
 
     def preview_description(self, state: dict) -> str:
         return "Enter a Google Image Search query + your Img2Img text prompt describing how to alter the result to create a unique, relevant ai generated images for any search query."
-
-    def get_raw_price(self, state: dict) -> float:
-        return super().get_raw_price(state) * state.get("num_outputs", 1)

recipes/SmartGPT.py

@@ -204,9 +204,6 @@ def render_steps(self):
     def preview_description(self, state: dict) -> str:
         return "SmartGPT is a cutting-edge AI technology that can be used to generate natural language responses to any given input. We have combined the power of [CoT](https://arxiv.org/abs/2305.02897), [Reflexion](https://arxiv.org/abs/2303.11366) & [DERA](https://arxiv.org/abs/2303.17071) into one pipeline so that you can use ChatGPT to its full potential! Input your prompt + a reflection/research prompt + a resolver prompt to use SmartGPT for enhanced text generation, natural language and incredible question-answer results."
 
-    def get_raw_price(self, state: dict) -> float:
-        return self.price * state.get("num_outputs", 1)
-
 
 def answers_as_prompt(texts: list[str], sep="\n\n") -> str:
     return sep.join(

recipes/Text2Audio.py

@@ -140,9 +140,6 @@ def render_example(self, state: dict):
     def preview_description(self, state: dict) -> str:
         return "Generate AI Music with text instruction prompts. AudiLDM is capable of generating realistic audio samples by process any text input. Learn more [here](https://huggingface.co/cvssp/audioldm-m-full)."
 
-    def get_raw_price(self, state: dict) -> float:
-        return super().get_raw_price(state) * state.get("num_outputs", 1)
-
 
 def _render_output(state):
     selected_models = state.get("selected_models", [])