@@ -42,34 +42,34 @@ async def load_settings(request: Request) -> GETSettingsModel | JSONResponse:
42
42
43
43
44
44
@app .post ('/unset-settings-tokens' , response_model = dict [str , str ])
45
- async def unset_settings_tokens (
46
- request : Request
47
- ) -> JSONResponse :
45
+ async def unset_settings_tokens (request : Request ) -> JSONResponse :
48
46
try :
49
47
settings_store = await SettingsStoreImpl .get_instance (
50
48
config , get_user_id (request )
51
49
)
52
50
53
51
existing_settings = await settings_store .load ()
54
52
if existing_settings :
55
- settings = existing_settings .model_copy (update = {'secrets_store' : SecretStore ()})
53
+ settings = existing_settings .model_copy (
54
+ update = {'secrets_store' : SecretStore ()}
55
+ )
56
56
await settings_store .store (settings )
57
-
57
+
58
58
return JSONResponse (
59
59
status_code = status .HTTP_200_OK ,
60
60
content = {'message' : 'Settings stored' },
61
61
)
62
-
62
+
63
63
except Exception as e :
64
64
logger .warning (f'Something went wrong unsetting tokens: { e } ' )
65
65
return JSONResponse (
66
66
status_code = status .HTTP_500_INTERNAL_SERVER_ERROR ,
67
67
content = {'error' : 'Something went wrong unsetting tokens' },
68
68
)
69
+
70
+
69
71
@app .post ('/reset-settings' , response_model = dict [str , str ])
70
- async def reset_settings (
71
- request : Request
72
- ) -> JSONResponse :
72
+ async def reset_settings (request : Request ) -> JSONResponse :
73
73
"""
74
74
Resets user settings.
75
75
"""
@@ -79,23 +79,28 @@ async def reset_settings(
79
79
)
80
80
81
81
existing_settings = await settings_store .load ()
82
- settings = Settings (language = "en" ,
83
- agent = "CodeActAgent" ,
84
- security_analyzer = "" ,
85
- confirmation_mode = False ,
86
- llm_model = "anthropic/claude-3-5-sonnet-20241022" ,
87
- llm_api_key = "" ,
88
- llm_base_url = "" ,
89
- remote_runtime_resource_factor = 1 ,
90
- enable_default_condenser = True ,
91
- enable_sound_notifications = False ,
92
- user_consents_to_analytics = existing_settings .user_consents_to_analytics if existing_settings else False
93
- )
94
-
82
+ settings = Settings (
83
+ language = 'en' ,
84
+ agent = 'CodeActAgent' ,
85
+ security_analyzer = '' ,
86
+ confirmation_mode = False ,
87
+ llm_model = 'anthropic/claude-3-5-sonnet-20241022' ,
88
+ llm_api_key = '' ,
89
+ llm_base_url = '' ,
90
+ remote_runtime_resource_factor = 1 ,
91
+ enable_default_condenser = True ,
92
+ enable_sound_notifications = False ,
93
+ user_consents_to_analytics = existing_settings .user_consents_to_analytics
94
+ if existing_settings
95
+ else False ,
96
+ )
97
+
95
98
server_config_values = server_config .get_config ()
96
- is_hide_llm_settings_enabled = server_config_values .get ("FEATURE_FLAGS" , {}).get ("HIDE_LLM_SETTINGS" , False )
99
+ is_hide_llm_settings_enabled = server_config_values .get (
100
+ 'FEATURE_FLAGS' , {}
101
+ ).get ('HIDE_LLM_SETTINGS' , False )
97
102
# We don't want the user to be able to modify these settings in SaaS
98
- if ( server_config .app_mode == AppMode .SAAS and is_hide_llm_settings_enabled ) :
103
+ if server_config .app_mode == AppMode .SAAS and is_hide_llm_settings_enabled :
99
104
if existing_settings :
100
105
settings .llm_api_key = existing_settings .llm_api_key
101
106
settings .llm_base_url = existing_settings .llm_base_url
@@ -106,14 +111,15 @@ async def reset_settings(
106
111
status_code = status .HTTP_200_OK ,
107
112
content = {'message' : 'Settings stored' },
108
113
)
109
-
114
+
110
115
except Exception as e :
111
116
logger .warning (f'Something went wrong resetting settings: { e } ' )
112
117
return JSONResponse (
113
118
status_code = status .HTTP_500_INTERNAL_SERVER_ERROR ,
114
119
content = {'error' : 'Something went wrong resetting settings' },
115
120
)
116
121
122
+
117
123
@app .post ('/settings' , response_model = dict [str , str ])
118
124
async def store_settings (
119
125
request : Request ,
0 commit comments