update

Author: cyyeh

wren-ai-service/src/providers/embedder/litellm.py

@@ -70,7 +70,7 @@ async def run(self, text: str):
 
         meta = {
             "model": response.model,
-            "usage": dict(response.usage) if response.usage else {},
+            "usage": dict(response.usage) if hasattr(response, "usage") else {},
         }
 
         return {"embedding": response.data[0]["embedding"], "meta": meta}
@@ -117,7 +117,9 @@ async def _embed_batch(
             if "model" not in meta:
                 meta["model"] = response.model
             if "usage" not in meta:
-                meta["usage"] = dict(response.usage) if response.usage else {}
+                meta["usage"] = (
+                    dict(response.usage) if hasattr(response, "usage") else {}
+                )
             else:
                 meta["usage"]["prompt_tokens"] += response.usage.prompt_tokens
                 meta["usage"]["total_tokens"] += response.usage.total_tokens