PikoAi 0.1.5__py3-none-any.whl → 0.1.8__py3-none-any.whl

cli.py

@@ -296,5 +296,64 @@ def list_models():
         for model_name_full in model_list: # model_name_full is "openai/gpt-4o", etc.
             click.echo(f"  - {model_name_full}")
 
+@cli.command('set-api-key')
+@click.option('--provider', '-p', type=click.Choice(list(AVAILABLE_MODELS.keys())), 
+              help='The LLM provider to set API key for')
+@click.option('--key', '-k', help='The API key to set (if not provided, will prompt for it)')
+def set_api_key(provider, key):
+    """Set or update API key for a specific LLM provider"""
+    if not provider:
+        # If no provider specified, ask user to choose
+        questions = [
+            inquirer.List('provider_key',
+                message="Select LLM Provider to update API key",
+                choices=list(AVAILABLE_MODELS.keys())
+            )
+        ]
+        provider = inquirer.prompt(questions)['provider_key']
+    
+    # Get the environment variable name for this provider
+    env_var = API_KEYS.get(provider)
+    if not env_var:
+        raise ValueError(f"Unknown provider: {provider}")
+    
+    # Get the API key (either from command line or prompt)
+    if not key:
+        questions = [
+            inquirer.Text('api_key',
+                message=f"Enter your {provider.upper()} API key",
+                validate=lambda _, x: len(x.strip()) > 0
+            )
+        ]
+        key = inquirer.prompt(questions)['api_key']
+    
+    # Get the path to .env file
+    env_path = os.path.join(os.path.dirname(__file__), '../.env')
+    
+    # Read existing .env file if it exists
+    lines = []
+    if os.path.exists(env_path):
+        with open(env_path, 'r') as f:
+            lines = f.readlines()
+    
+    # Update or add the API key
+    key_line = f"{env_var}={key}\n"
+    key_exists = False
+    
+    for i, line in enumerate(lines):
+        if line.strip().startswith(f"{env_var}=") or line.strip().startswith(f"#{env_var}="):
+            lines[i] = key_line
+            key_exists = True
+            break
+    
+    if not key_exists:
+        lines.append(key_line)
+    
+    # Write back to .env file
+    with open(env_path, 'w') as f:
+        f.writelines(lines)
+    
+    click.echo(f"API key for {provider.upper()} has been updated successfully in {env_path}")
+
 if __name__ == '__main__':
     cli() 

{pikoai-0.1.5.dist-info → pikoai-0.1.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PikoAi
-Version: 0.1.5
+Version: 0.1.8
 Summary: An AI-powered task automation tool
 Home-page: https://github.com/nihaaaar22/OS-Assistant
 Author: Nihar S

{pikoai-0.1.5.dist-info → pikoai-0.1.8.dist-info}/RECORD

@@ -1,5 +1,5 @@
 OpenCopilot.py,sha256=6210OONj0yV-LPzkCdZ1RUiHC5NZvLsD-y93Y1zurX0,10687
-cli.py,sha256=o2V_DETYXEiYm6617Xg8rDcAgb0ofmojPvdlOpNHwMY,11809
+cli.py,sha256=hY6KUxvKvJOFThZT--r6m2nzOEMIOVtiVptewsi9Z9w,13868
 Agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 Agents/Executor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 Agents/Executor/executor.py,sha256=zaFO4IQNE1PscvGPV8zoRsQAJAYVTz2JsD_zUBLyYwg,12360
@@ -24,9 +24,9 @@ Utils/executor_utils.py,sha256=eVVxZDcAGxENRV5DhCkgh1AKiJEgfV_1q-iSQNWR5kI,1180
 Utils/ter_interface.py,sha256=Zay9mwyAyKYTNQAKOWXHAa3upo9TWprSf26STiHXk0g,6255
 llm_interface/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 llm_interface/llm.py,sha256=tI_KDOW14QLWowA7bB3GPe2qjlk0sjS5fBavs9XD1fo,5185
-pikoai-0.1.5.dist-info/licenses/LICENSE,sha256=cELUVOboOAderKFp8bdtcM5VyJi61YH1oDbRhOuoQZw,1067
-pikoai-0.1.5.dist-info/METADATA,sha256=X5Tah8uN8Qse-K9lCNezMn3UPeGNi_-9WUqpg6faSKU,2961
-pikoai-0.1.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pikoai-0.1.5.dist-info/entry_points.txt,sha256=QVeDO6N3nO3UScMb2ksusQWPgcVn86vXosgL-8gu6fo,33
-pikoai-0.1.5.dist-info/top_level.txt,sha256=hWzBNE7UQsuNcENIOksGcJED08k3ZGRRn2X5jnStICU,53
-pikoai-0.1.5.dist-info/RECORD,,
+pikoai-0.1.8.dist-info/licenses/LICENSE,sha256=cELUVOboOAderKFp8bdtcM5VyJi61YH1oDbRhOuoQZw,1067
+pikoai-0.1.8.dist-info/METADATA,sha256=BtVoSGxZ_-ZKZ7k1rgSEvkcCsj_5OpHokiHvOOnBkGU,2961
+pikoai-0.1.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pikoai-0.1.8.dist-info/entry_points.txt,sha256=xjZnheDymNDnQ0o84R0jZKEITrhNbzQWN-AhqfA_d6s,50
+pikoai-0.1.8.dist-info/top_level.txt,sha256=hWzBNE7UQsuNcENIOksGcJED08k3ZGRRn2X5jnStICU,53
+pikoai-0.1.8.dist-info/RECORD,,

{pikoai-0.1.5.dist-info → pikoai-0.1.8.dist-info}/entry_points.txt

@@ -1,2 +1,3 @@
 [console_scripts]
 piko = cli:cli
+pikoai = cli:cli