Fix typing in server directory (#8375)

Co-authored-by: openhands <openhands@all-hands.dev>
Co-authored-by: Rohit Malhotra <rohitvinodmalhotra@gmail.com>
This commit is contained in:
Graham Neubig
2025-05-13 17:27:59 -04:00
committed by GitHub
parent f9b0fcd76e
commit 154eed148f
15 changed files with 79 additions and 56 deletions

View File

@@ -29,7 +29,7 @@ class DefaultUserAuth(UserAuth):
"""The default implementation does not support multi tenancy, so access_token is always None"""
return None
async def get_user_settings_store(self):
async def get_user_settings_store(self) -> SettingsStore:
settings_store = self._settings_store
if settings_store:
return settings_store
@@ -37,6 +37,8 @@ class DefaultUserAuth(UserAuth):
settings_store = await shared.SettingsStoreImpl.get_instance(
shared.config, user_id
)
if settings_store is None:
raise ValueError('Failed to get settings store instance')
self._settings_store = settings_store
return settings_store
@@ -49,7 +51,7 @@ class DefaultUserAuth(UserAuth):
self._settings = settings
return settings
async def get_secrets_store(self):
async def get_secrets_store(self) -> SecretsStore:
secrets_store = self._secrets_store
if secrets_store:
return secrets_store
@@ -57,6 +59,8 @@ class DefaultUserAuth(UserAuth):
secret_store = await shared.SecretsStoreImpl.get_instance(
shared.config, user_id
)
if secret_store is None:
raise ValueError('Failed to get secrets store instance')
self._secrets_store = secret_store
return secret_store
@@ -70,9 +74,10 @@ class DefaultUserAuth(UserAuth):
return user_secrets
async def get_provider_tokens(self) -> PROVIDER_TOKEN_TYPE | None:
secrets_store = await self.get_user_secrets()
provider_tokens = getattr(secrets_store, 'provider_tokens', None)
return provider_tokens
user_secrets = await self.get_user_secrets()
if user_secrets is None:
return None
return user_secrets.provider_tokens
@classmethod
async def get_instance(cls, request: Request) -> UserAuth: