taskai-cli 1.0.0__tar.gz → 1.0.1__tar.gz

{taskai_cli-1.0.0 → taskai_cli-1.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: taskai-cli
-Version: 1.0.0
+Version: 1.0.1
 Author-email: Alex Paskal <alexcpaskal@gmail.com>
 Requires-Python: >=3.12
 Requires-Dist: google-genai>=2.8

{taskai_cli-1.0.0 → taskai_cli-1.0.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "taskai-cli"
-version = "1.0.0"
+version = "1.0.1"
 description = ""
 authors = [{name = "Alex Paskal", email = "alexcpaskal@gmail.com"}]
 readme = "README.md"

{taskai_cli-1.0.0 → taskai_cli-1.0.1}/taskai/services/user_setup.py

@@ -14,20 +14,20 @@ def user_setup_service(
     db: JsonDirectoryDatabase
 ):
     
-    config = db.config
+    config = db.get_config()
     # setup gemini model
     print("Beginning setup")
     if "GEMINI_API_KEY" not in config:
         api_key = _get_gemini_api_key()
         if api_key:
-            config["GEMINI_API_KEY"] = api_key
+            db.update_config(GEMINI_API_KEY=api_key)
     else:
         print("gemini key already specified")
     
     if "GEMINI_MODEL" not in config:
         model = _select_gemini_model()
         if model:
-            config["GEMINI_MODEL"] = model
+            db.update_config(GEMINI_MODEL=model)
     else:
         print("gemini model already specified")
     print("Setup complete! Use 'task config set|get|list' to interact with your configuration options")
@@ -60,4 +60,4 @@ if __name__ == "__main__":
         ".taskai/task_db", user=os.getenv("USER")
     )
     db.connect()
-    user_setup_service(db)
+    user_setup_service(db)