feat:精简

backend/open_webui/routers/ollama.py

@@ -41,8 +41,6 @@ from starlette.background import BackgroundTask
 
 
 from open_webui.models.models import Models
-from open_webui.utils.credit.usage import CreditDeduct
-from open_webui.utils.credit.utils import check_credit_by_user_id
 from open_webui.utils.misc import (
     calculate_sha256,
 )
@@ -1077,10 +1075,6 @@ async def embeddings(
 ):
     log.info(f"generate_ollama_embeddings {form_data}")
 
-    # check credit
-    if user:
-        check_credit_by_user_id(user_id=user.id, form_data={}, is_embedding=True)
-
     if url_idx is None:
         await get_all_models(request, user=user)
         models = request.app.state.OLLAMA_MODELS
@@ -1128,18 +1122,6 @@ async def embeddings(
 
         data = r.json()
 
-        # calculate usage
-        if user:
-            input_text = form_data.prompt
-            with CreditDeduct(
-                user=user,
-                model_id=form_data.model,
-                body={"messages": [{"role": "user", "content": input_text}]},
-                is_stream=False,
-                is_embedding=True,
-            ) as credit_deduct:
-                credit_deduct.run(input_text)
-
         return data
     except Exception as e:
         log.exception(e)
@@ -1469,8 +1451,6 @@ async def generate_openai_chat_completion(
     url_idx: Optional[int] = None,
     user=Depends(get_verified_user),
 ):
-    check_credit_by_user_id(user_id=user.id, form_data=form_data)
-
     metadata = form_data.pop("metadata", None)
 
     try: