优化三个工具的返回结果。

2025-01-06 17:27:45 +08:00
parent c7d2d482da
commit 4d0c41d222
7 changed files with 266 additions and 8 deletions
--- a/utils.py
+++ b/utils.py
@@ -17,6 +17,7 @@ class Settings(BaseSettings):
    # Material Project
    mp_api_key: Optional[str] = Field(None, env="MP_API_KEY")
    mp_endpoint: Optional[str] = Field(None, env="MP_ENDPOINT")
+    mp_topk: Optional[int] = Field(3, env="MP_TOPK")
    
    # Proxy
    http_proxy: Optional[str] = Field(None, env="HTTP_PROXY")