diff options
| -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: |