Skip to content

Conversation

SAKURA-CAT
Copy link
Member

@SAKURA-CAT SAKURA-CAT commented Feb 12, 2025

该拉取请求(Pull Request)对 swanlab 项目进行了重要更新,重点涉及 HostFormatter 类及其在多个模块中的集成。主要改动包括 HostFormatter 类的迁移与增强、登录功能的更新,以及额外的测试以确保系统的健壮性。

主要变更:

重构与增强:

  • HostFormatter 类已从 swanlab/data/sdk.py 移动到 swanlab/package.py,并增强了对 hostweb_host 参数的处理能力。该类现在负责验证和格式化主机 URL,并设置相关的环境变量。[1] [2]

登录功能:

  • 更新了 swanlab/cli/commands/auth/login.py 中的 login 函数,新增 hostweb_host 选项,允许用户在登录时指定服务器和 Web 主机。
  • 在重新登录时新增逻辑以清除环境变量,并使用 HostFormatter 类进行主机地址的验证和格式化。

代码优化:

  • 移除了 swanlab/data/sdk.py 中冗余的 HostFormatter 类定义,并更新 login 函数以使用 swanlab/package.py 中的新 HostFormatter 类。[1] [2]

测试:

  • test/unit/cli/test_cli_login.pytest/unit/test_package.py 中新增测试用例,以验证 HostFormatter 类的功能以及不同主机输入下的登录过程。[1] [2]
  • 移除了 test/unit/data/test_sdk.py 中针对旧 HostFormatter 类的过时测试用例。

导入与依赖项:

  • 在多个文件中添加了 ossysre 模块的必要导入,以支持新功能。[1] [2] [3] [4]

注意:在交互上,重新登录需要使用--relogin参数,此时host也会被重置为官方的。

close #792

@SAKURA-CAT SAKURA-CAT requested a review from Zeyi-Lin February 12, 2025 06:06
@SAKURA-CAT SAKURA-CAT self-assigned this Feb 12, 2025
@SAKURA-CAT SAKURA-CAT added the 💪 enhancement New feature or request label Feb 12, 2025
@SAKURA-CAT SAKURA-CAT merged commit 6d49c88 into main Feb 12, 2025
5 checks passed
@SAKURA-CAT SAKURA-CAT deleted the feat/login-cli branch February 12, 2025 07:05
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

💪 enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[REQUEST] 将web host、api host相关环境变量集成进 swanlab.login

2 participants