[emu] Check chosen credential before create billing event (#920)

* check chosen credential before create billing event

Author: noobHappylife

services/api/src/owl/utils/billing/oss.py

@@ -574,7 +574,6 @@ def create_llm_events(
         model_id: str,
         input_tokens: int,
         output_tokens: int,
-        model_provider: str = "",
         is_byok: bool = False,
         *,
         create_usage: bool = True,
@@ -620,7 +619,6 @@ def create_image_gen_events(
         text_output_token: int,
         image_input_token: int,
         image_output_token: int,
-        model_provider: str = "",
         is_byok: bool = False,
         create_usage: bool = True,
     ) -> None:
@@ -686,7 +684,6 @@ def create_embedding_events(
         self,
         model_id: str,
         token_usage: int,
-        model_provider: str = "",
         is_byok: bool = False,
         *,
         create_usage: bool = True,
@@ -731,7 +728,6 @@ def create_reranker_events(
         self,
         model_id: str,
         num_searches: int,
-        model_provider: str = "",
         is_byok: bool = False,
         *,
         create_usage: bool = True,

services/api/src/owl/utils/lm.py

@@ -1728,13 +1728,12 @@ async def _setup_chat(
         finally:
             # # Consume rate limits
             # await asyncio.gather(rpm_limiter.hit(), tpm_limiter.hit(self._chat_usage.total_tokens))
-            if self.billing is not None:
+            if self.billing is not None and router.chosen_credential is not None:
                 try:
                     self.billing.create_llm_events(
                         model_id=model,
                         input_tokens=self._chat_usage.prompt_tokens,
                         output_tokens=self._chat_usage.completion_tokens,
-                        model_provider=router.chosen_credential.provider,
                         is_byok=router.chosen_credential.is_byok,
                     )
                 except Exception as e:
@@ -2214,7 +2213,6 @@ def _record_image_usage(
         self,
         *,
         model_id: str,
-        model_provider: str,
         is_byok: bool,
         usage: Usage | None,
         source: Literal["image_endpoint", "completion"],
@@ -2260,7 +2258,6 @@ def _record_image_usage(
                 text_output_token=text_output_token,
                 image_input_token=image_input_token,
                 image_output_token=image_output_token,
-                model_provider=model_provider,
                 is_byok=is_byok,
             )
         except Exception as e:
@@ -2290,7 +2287,6 @@ async def image_generation(
             )
             self._record_image_usage(
                 model_id=resolved_model,
-                model_provider=router.chosen_credential.provider,
                 is_byok=router.chosen_credential.is_byok,
                 usage=provider_result.usage,
                 source=provider_result.source,
@@ -2328,7 +2324,6 @@ async def image_edit(
             )
             self._record_image_usage(
                 model_id=resolved_model,
-                model_provider=router.chosen_credential.provider,
                 is_byok=router.chosen_credential.is_byok,
                 usage=provider_result.usage,
                 source=provider_result.source,
@@ -2384,12 +2379,11 @@ async def _setup_embedding(self, model: str):
             # await asyncio.gather(
             #     rpm_limiter.hit(), tpm_limiter.hit(self._embed_usage.total_tokens)
             # )
-            if self.billing is not None:
+            if self.billing is not None and router.chosen_credential is not None:
                 try:
                     self.billing.create_embedding_events(
                         model_id=model,
                         token_usage=self._embed_usage.total_tokens,
-                        model_provider=router.chosen_credential.provider,
                         is_byok=router.chosen_credential.is_byok,
                     )
                 except Exception as e:
@@ -2524,12 +2518,11 @@ async def _setup_reranking(self, model: str):
         finally:
             # # Consume rate limits
             # await asyncio.gather(rpm_limiter.hit(), spm_limiter.hit(self._rerank_usage.documents))
-            if self.billing is not None:
+            if self.billing is not None and router.chosen_credential is not None:
                 try:
                     self.billing.create_reranker_events(
                         model_id=model,
                         num_searches=self._rerank_usage.documents,
-                        model_provider=router.chosen_credential.provider,
                         is_byok=router.chosen_credential.is_byok,
                     )
                 except Exception as e: