diff --git a/.github/workflows/docker-ci.yaml b/.github/workflows/docker-ci.yaml.hint similarity index 100% rename from .github/workflows/docker-ci.yaml rename to .github/workflows/docker-ci.yaml.hint diff --git a/llmkira/extra/user/schema.py b/llmkira/extra/user/schema.py index dc2174bbe..606f3684e 100644 --- a/llmkira/extra/user/schema.py +++ b/llmkira/extra/user/schema.py @@ -11,7 +11,7 @@ from pydantic import field_validator, ConfigDict, BaseModel, Field from pydantic_settings import BaseSettings, SettingsConfigDict -from ...sdk.endpoint import Driver +from llmkira.sdk.endpoint.tee import Driver class UserDriverMode(Enum): @@ -74,7 +74,7 @@ def create_from_function( return cls( request_id=request_id, uid=uid, - cost=cls.Cost.by_function( + cost=Cost.by_function( function_name=cost_by, token_usage=token_usage, token_uuid=token_uuid,