diff options
| author | Nate Sesti <sestinj@gmail.com> | 2023-07-30 16:30:06 -0700 |
|---|---|---|
| committer | Nate Sesti <sestinj@gmail.com> | 2023-07-30 16:30:06 -0700 |
| commit | 798e94f62b2c64762e2e6f79645e9334013ac7a8 (patch) | |
| tree | cea85e5cacf20af84875c63b5b8b4a4dda229f67 | |
| parent | 6ead69fb71ea01e8b0ab6964d17c5dd058244883 (diff) | |
| download | sncontinue-798e94f62b2c64762e2e6f79645e9334013ac7a8.tar.gz sncontinue-798e94f62b2c64762e2e6f79645e9334013ac7a8.tar.bz2 sncontinue-798e94f62b2c64762e2e6f79645e9334013ac7a8.zip | |
refactor: :construction: rename get_api_key -> get_user_secret
| -rw-r--r-- | continuedev/src/continuedev/core/abstract_sdk.py | 2 | ||||
| -rw-r--r-- | continuedev/src/continuedev/core/sdk.py | 7 |
2 files changed, 3 insertions, 6 deletions
diff --git a/continuedev/src/continuedev/core/abstract_sdk.py b/continuedev/src/continuedev/core/abstract_sdk.py index 94d7be10..e048f877 100644 --- a/continuedev/src/continuedev/core/abstract_sdk.py +++ b/continuedev/src/continuedev/core/abstract_sdk.py @@ -73,7 +73,7 @@ class AbstractContinueSDK(ABC): pass @abstractmethod - async def get_user_secret(self, env_var: str, prompt: str) -> str: + async def get_user_secret(self, env_var: str) -> str: pass config: ContinueConfig diff --git a/continuedev/src/continuedev/core/sdk.py b/continuedev/src/continuedev/core/sdk.py index 7febb932..1dd4b857 100644 --- a/continuedev/src/continuedev/core/sdk.py +++ b/continuedev/src/continuedev/core/sdk.py @@ -69,7 +69,7 @@ class ContinueSDK(AbstractContinueSDK): async def start_model(self, llm: LLM): kwargs = {} if llm.requires_api_key: - kwargs["api_key"] = await self.get_api_key(llm.requires_api_key) + kwargs["api_key"] = await self.get_user_secret(llm.requires_api_key) if llm.requires_unique_id: kwargs["unique_id"] = self.ide.unique_id if llm.requires_write_log: @@ -145,13 +145,10 @@ class ContinueSDK(AbstractContinueSDK): path = await self._ensure_absolute_path(path) return await self.run_step(FileSystemEditStep(edit=DeleteDirectory(path=path))) - async def get_api_key(self, env_var: str) -> str: + async def get_user_secret(self, env_var: str) -> str: # TODO support error prompt dynamically set on env_var return await self.ide.getUserSecret(env_var) - async def get_user_secret(self, env_var: str, prompt: str) -> str: - return await self.ide.getUserSecret(env_var) - _last_valid_config: ContinueConfig = None def _load_config_dot_py(self) -> ContinueConfig: |