mirror of
https://github.com/codeflash-ai/codeflash-internal.git
synced 2026-05-04 18:25:18 +00:00
style: auto-fix linting issues
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
parent
ffa7989c78
commit
2f04efb87d
2 changed files with 4 additions and 4 deletions
|
|
@ -118,8 +118,7 @@ class LLMClient:
|
|||
if bearer_token := os.environ.get("AWS_BEARER_TOKEN_BEDROCK"):
|
||||
# Use bearer token authentication
|
||||
self.anthropic_client = AsyncAnthropicBedrock(
|
||||
aws_session_token=bearer_token,
|
||||
aws_region=os.environ.get("AWS_REGION", "us-east-1"),
|
||||
aws_session_token=bearer_token, aws_region=os.environ.get("AWS_REGION", "us-east-1")
|
||||
)
|
||||
else:
|
||||
# Use standard AWS access key credentials
|
||||
|
|
|
|||
|
|
@ -1,4 +1,5 @@
|
|||
"""Tests for LLM client event loop handling."""
|
||||
|
||||
import asyncio
|
||||
from unittest import mock
|
||||
|
||||
|
|
@ -6,7 +7,7 @@ import pytest
|
|||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_llm_client_handles_closed_event_loop_on_client_close():
|
||||
async def test_llm_client_handles_closed_event_loop_on_client_close() -> None:
|
||||
"""Test that LLMClient handles RuntimeError when closing clients on closed event loops."""
|
||||
from aiservice.llm import LLMClient
|
||||
|
||||
|
|
@ -53,7 +54,7 @@ async def test_llm_client_handles_closed_event_loop_on_client_close():
|
|||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_llm_client_recreates_clients_on_loop_change():
|
||||
async def test_llm_client_recreates_clients_on_loop_change() -> None:
|
||||
"""Test that LLMClient recreates provider clients when event loop changes."""
|
||||
from aiservice.llm import LLMClient
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue