Skip to content

Commit 3afecef

Browse files
committed
chore: Refactor all uses of config to use the config object from the config module
1 parent ea57173 commit 3afecef

38 files changed

+79
-74
lines changed

openadapt/adapters/__init__.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2,13 +2,9 @@
22

33
from types import ModuleType
44

5-
from openadapt import config
6-
from . import anthropic
7-
from . import openai
8-
from . import replicate
9-
from . import som
10-
from . import ultralytics
11-
from . import google
5+
from openadapt.config import config
6+
7+
from . import anthropic, google, openai, replicate, som, ultralytics
128

139

1410
def get_default_prompt_adapter() -> ModuleType:

openadapt/adapters/anthropic.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,8 @@
55
from loguru import logger
66
import anthropic
77

8-
from openadapt import cache, config
9-
8+
from openadapt import cache
9+
from openadapt.config import config
1010

1111
MAX_TOKENS = 4096
1212
# from https://docs.anthropic.com/claude/docs/vision

openadapt/adapters/google.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,8 @@
99
import fire
1010
import google.generativeai as genai
1111

12-
from openadapt import cache, config, utils
13-
12+
from openadapt import cache, utils
13+
from openadapt.config import config
1414

1515
MAX_TOKENS = 2**20 # 1048576
1616
MODEL_NAME = [

openadapt/adapters/openai.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,12 @@
44
"""
55

66
from pprint import pformat
7-
import requests
87

98
from loguru import logger
9+
import requests
1010

11-
from openadapt import cache, config
11+
from openadapt import cache
12+
from openadapt.config import config
1213

1314
MODEL_NAME = [
1415
"gpt-4-vision-preview",

openadapt/adapters/replicate.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,8 @@
99
import replicate
1010
import requests
1111

12-
from openadapt import cache, config, utils
12+
from openadapt import cache, utils
13+
from openadapt.config import config
1314

1415

1516
@cache.cache()
@@ -28,10 +29,8 @@ def _fetch_segmented_image(image_uri: str) -> Image.Image:
2829
os.environ["REPLICATE_API_TOKEN"] = config.REPLICATE_API_TOKEN
2930
logger.info("segmenting image...")
3031
segmented_image_url = replicate.run(
31-
(
32-
"pablodawson/segment-anything-automatic:"
33-
"14fbb04535964b3d0c7fad03bb4ed272130f15b956cbedb7b2f20b5b8a2dbaa0"
34-
),
32+
"pablodawson/segment-anything-automatic:"
33+
"14fbb04535964b3d0c7fad03bb4ed272130f15b956cbedb7b2f20b5b8a2dbaa0",
3534
input={
3635
"image": image_uri,
3736
# "resize_width": 1080,

openadapt/adapters/som.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
import fire
1212
import gradio_client
1313

14-
from openadapt import config
14+
from openadapt.config import config
1515

1616

1717
def save_image_to_temp_file(img: Image.Image) -> str:

openadapt/adapters/ultralytics.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@
33
See https://docs.ultralytics.com/models/fast-sam/#predict-usage for details.
44
"""
55

6-
from tempfile import TemporaryDirectory
76
from pathlib import Path
7+
from tempfile import TemporaryDirectory
88
import os
99

1010
from loguru import logger
@@ -16,7 +16,6 @@
1616

1717
from openadapt import cache
1818

19-
2019
MODEL_NAMES = (
2120
"FastSAM-x.pt",
2221
"FastSAM-s.pt",

openadapt/app/dashboard/api/index.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,10 +10,10 @@
1010
from loguru import logger
1111
import uvicorn
1212

13-
from openadapt import config
1413
from openadapt.app.dashboard.api.recordings import RecordingsAPI
1514
from openadapt.app.dashboard.api.settings import SettingsAPI
1615
from openadapt.build_utils import is_running_from_executable
16+
from openadapt.config import config
1717

1818
app = FastAPI()
1919

openadapt/app/dashboard/run.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,8 @@
88

99
from loguru import logger
1010

11-
from openadapt import config
1211
from openadapt.build_utils import is_running_from_executable
12+
from openadapt.config import config
1313
from openadapt.extensions.thread import Thread
1414

1515
from .api.index import run_app

openadapt/app/main.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,16 +12,18 @@
1212
from subprocess import Popen
1313
import base64
1414
import os
15+
import pathlib
1516

1617
from nicegui import app, ui
1718

18-
from openadapt import config, replay
19+
from openadapt import replay
1920
from openadapt.app.cards import recording_prompt, select_import, settings
2021
from openadapt.app.objects.console import Console
2122
from openadapt.app.util import clear_db, on_export, on_import
23+
from openadapt.config import config
2224

2325
SERVER = "127.0.0.1:8000/upload"
24-
FPATH = os.path.dirname(__file__)
26+
FPATH = pathlib.Path(__file__).parent
2527

2628
# Recording description autocomplete
2729
OPTIONS = ["test"]

0 commit comments

Comments
 (0)