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
« prev ^ index » next coverage.py v7.6.8, created at 2024-12-06 09:34 +0800
1from pathlib import Path
3from ..settings import get_settings
4from ._codegen import generate_modules, ruff_check
5from ._metadata import WithMetadata as WithMetadata
6from .user import APIKey, User
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
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")
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
33__all__ = [
34 "APIKey",
35 "User",
36 "Assistant",
37 "Message",
38 "Run",
39 "RunStep",
40 "Thread",
41 "FileObject",
42]