Coverage for src/fastoai/models/__init__.py: 100%

21 statements  

« prev     ^ index     » next       coverage.py v7.6.8, created at 2024-12-06 09:34 +0800

1from pathlib import Path 

2 

3from ..settings import get_settings 

4from ._codegen import generate_modules, ruff_check 

5from ._metadata import WithMetadata as WithMetadata 

6from .user import APIKey, User 

7 

8if get_settings().generate_models: 

9 import openai.types.beta.assistant 

10 import openai.types.beta.thread 

11 import openai.types.beta.threads.message 

12 import openai.types.beta.threads.run 

13 import openai.types.beta.threads.runs.run_step 

14 import openai.types.file_object 

15 

16 generate_modules( 

17 openai.types.beta.assistant, 

18 openai.types.beta.thread, 

19 openai.types.beta.threads.message, 

20 openai.types.beta.threads.run, 

21 openai.types.beta.threads.runs.run_step, 

22 openai.types.file_object, 

23 ) 

24 ruff_check(Path(__file__).parent / "generated") 

25 

26from .generated.assistant import Assistant 

27from .generated.file_object import FileObject 

28from .generated.message import Message 

29from .generated.run import Run 

30from .generated.run_step import RunStep 

31from .generated.thread import Thread 

32 

33__all__ = [ 

34 "APIKey", 

35 "User", 

36 "Assistant", 

37 "Message", 

38 "Run", 

39 "RunStep", 

40 "Thread", 

41 "FileObject", 

42]