Spaces:
Running
Running
Automatically load *_ops.py.
Browse files- server/main.py +7 -7
server/main.py
CHANGED
@@ -1,15 +1,15 @@
|
|
1 |
import dataclasses
|
2 |
import fastapi
|
3 |
import pathlib
|
|
|
4 |
from . import ops
|
5 |
from . import workspace
|
6 |
-
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
|
11 |
-
|
12 |
-
from . import llm_ops
|
13 |
|
14 |
app = fastapi.FastAPI()
|
15 |
|
|
|
1 |
import dataclasses
|
2 |
import fastapi
|
3 |
import pathlib
|
4 |
+
import pkgutil
|
5 |
from . import ops
|
6 |
from . import workspace
|
7 |
+
|
8 |
+
here = pathlib.Path(__file__).parent
|
9 |
+
for _, name, _ in pkgutil.iter_modules([str(here)]):
|
10 |
+
if name.endswith('_ops') and not name.startswith('test_'):
|
11 |
+
print(f'Importing {name}')
|
12 |
+
__import__(f'server.{name}')
|
|
|
13 |
|
14 |
app = fastapi.FastAPI()
|
15 |
|