fix: Add missing import for count_messages_tokens in providers.py

- Import count_messages_tokens from utils module
- Fixes 'name count_messages_tokens is not defined' error in Google streaming handler
parent a41de233
...@@ -33,6 +33,7 @@ from anthropic import Anthropic ...@@ -33,6 +33,7 @@ from anthropic import Anthropic
from pydantic import BaseModel from pydantic import BaseModel
from .models import Provider, Model, ErrorTracking from .models import Provider, Model, ErrorTracking
from .config import config from .config import config
from .utils import count_messages_tokens
# Check if debug mode is enabled # Check if debug mode is enabled
AISBF_DEBUG = os.environ.get('AISBF_DEBUG', '').lower() in ('true', '1', 'yes') AISBF_DEBUG = os.environ.get('AISBF_DEBUG', '').lower() in ('true', '1', 'yes')
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment