Coverage for intelligence_toolkit/tests/unit/AI/test_types.py: 100%

1# Copyright (c) 2024 Microsoft Corporation. All rights reserved. 

2# Licensed under the MIT license. See LICENSE file in the project. 

3# 

4from intelligence_toolkit.AI.types import OpenAIClientTypes 

5from openai import AsyncAzureOpenAI, AsyncOpenAI 

6 

7 

8def test_openai_client_types_includes_async_openai(): 

9 # Test that OpenAIClientTypes accepts AsyncOpenAI 

10 # This is a type union test, so we just verify the types are imported correctly 

11 assert AsyncOpenAI is not None 

12 assert AsyncAzureOpenAI is not None 

13 # Verify the type alias is defined 

14 assert OpenAIClientTypes is not None