orgo 0.0.19__tar.gz → 0.0.21__tar.gz

{orgo-0.0.19 → orgo-0.0.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: orgo
-Version: 0.0.19
+Version: 0.0.21
 Summary: Computers for AI agents
 Author: Orgo Team
 License: MIT

{orgo-0.0.19 → orgo-0.0.21}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "orgo"
-version = "0.0.19"
+version = "0.0.21"
 description = "Computers for AI agents"
 authors = [{name = "Orgo Team"}]
 license = {text = "MIT"}

{orgo-0.0.19 → orgo-0.0.21}/src/orgo/api/client.py

@@ -1,4 +1,3 @@
-# src/orgo/api/client.py
 """API client for Orgo service"""
 
 import requests
@@ -43,7 +42,10 @@ class ApiClient:
     # Computer lifecycle methods
     def create_computer(self, config: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
         """Create a new project with desktop instance"""
-        return self._request("POST", "projects", {"config": config} if config else None)
+        payload = {}
+        if config:
+            payload["config"] = config
+        return self._request("POST", "projects", payload if payload else None)
     
     def connect_computer(self, project_id: str) -> Dict[str, Any]:
         return self._request("GET", f"projects/by-name/{project_id}")

{orgo-0.0.19 → orgo-0.0.21}/src/orgo/computer.py

@@ -1,10 +1,9 @@
-# src/orgo/computer.py
 """Computer class for interacting with Orgo virtual environments"""
 import os
 import io
 import base64
 import logging
-from typing import Dict, List, Any, Optional, Callable
+from typing import Dict, List, Any, Optional, Callable, Literal
 from PIL import Image
 import requests
 from requests.exceptions import RequestException
@@ -16,7 +15,13 @@ from .project import ProjectManager
 logger = logging.getLogger(__name__)
 
 class Computer:
-    def __init__(self, project_id=None, api_key=None, config=None, base_api_url=None):
+    def __init__(self, 
+                 project_id: Optional[str] = None, 
+                 api_key: Optional[str] = None, 
+                 config: Optional[Dict[str, Any]] = None, 
+                 base_api_url: Optional[str] = None,
+                 ram: Optional[Literal[2, 4]] = None,
+                 cpu: Optional[Literal[2, 4]] = None):
         """
         Initialize an Orgo virtual computer.
         
@@ -25,6 +30,8 @@ class Computer:
             api_key: Orgo API key (defaults to ORGO_API_KEY env var)
             config: Configuration for new computer (optional)
             base_api_url: Custom API URL (optional)
+            ram: RAM in GB for new computer (2 or 4) - only used when creating
+            cpu: CPU cores for new computer (2 or 4) - only used when creating
         """
         self.api_key = api_key or os.environ.get("ORGO_API_KEY")
         self.base_api_url = base_api_url
@@ -38,14 +45,39 @@ class Computer:
             try:
                 self.project_id = project_id
                 self._info = self.api.connect_computer(project_id)
+                # Log if ram/cpu were provided but ignored
+                if ram is not None or cpu is not None:
+                    logger.info("Note: ram and cpu parameters are ignored when connecting to existing computer")
             except (RequestException, ValueError) as e:
                 logger.warning(f"Could not connect to saved project {project_id}: {e}")
-                self._create_new_computer(config)
+                self._create_new_computer(config, ram, cpu)
         else:
-            self._create_new_computer(config)
+            self._create_new_computer(config, ram, cpu)
             
-    def _create_new_computer(self, config=None):
+    def _create_new_computer(self, config: Optional[Dict[str, Any]] = None, 
+                           ram: Optional[Literal[2, 4]] = None,
+                           cpu: Optional[Literal[2, 4]] = None):
         """Create a new computer instance and save its ID"""
+        # Validate ram and cpu values if provided
+        if ram is not None and ram not in [2, 4]:
+            raise ValueError("ram must be either 2 or 4 GB")
+        if cpu is not None and cpu not in [2, 4]:
+            raise ValueError("cpu must be either 2 or 4 cores")
+        
+        # Build the config with ram and cpu if provided
+        if ram is not None or cpu is not None:
+            if config is None:
+                config = {}
+            else:
+                # Make a copy to avoid modifying the original
+                config = config.copy()
+            
+            # Add ram and cpu to config
+            if ram is not None:
+                config['ram'] = ram
+            if cpu is not None:
+                config['cpu'] = cpu
+        
         response = self.api.create_computer(config)
         self.project_id = response.get("name")
         self._info = response
@@ -74,7 +106,10 @@ class Computer:
     
     def destroy(self) -> Dict[str, Any]:
         """Terminate and delete the computer instance"""
-        return self.api.delete_computer(self.project_id)
+        result = self.api.delete_computer(self.project_id)
+        # Clear the local project cache after destroying
+        ProjectManager.clear_project_cache()
+        return result
     
     # Navigation methods
     def left_click(self, x: int, y: int) -> Dict[str, Any]:

{orgo-0.0.19 → orgo-0.0.21}/src/orgo/project.py

@@ -2,6 +2,7 @@
 """Project management for Orgo virtual environments"""
 import os
 import json
+import shutil
 import logging
 from typing import Optional
 
@@ -40,6 +41,18 @@ class ProjectManager:
             logger.error(f"Failed to save project ID: {str(e)}")
             raise RuntimeError(f"Failed to save project configuration: {str(e)}") from e
     
+    @staticmethod
+    def clear_project_cache() -> None:
+        """Clear the .orgo folder and all its contents"""
+        project_dir = ProjectManager._get_project_dir()
+        
+        if os.path.exists(project_dir):
+            try:
+                shutil.rmtree(project_dir)
+                logger.info(f"Cleared project cache at {project_dir}")
+            except (IOError, OSError) as e:
+                logger.warning(f"Failed to clear project cache: {str(e)}")
+    
     @staticmethod
     def _get_project_dir() -> str:
         """Get the project directory path"""
@@ -48,4 +61,4 @@ class ProjectManager:
     @staticmethod
     def _get_config_path() -> str:
         """Get the full path to the config file"""
-        return os.path.join(ProjectManager._get_project_dir(), "project.json")
+        return os.path.join(ProjectManager._get_project_dir(), "project.json")

{orgo-0.0.19 → orgo-0.0.21}/src/orgo.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: orgo
-Version: 0.0.19
+Version: 0.0.21
 Summary: Computers for AI agents
 Author: Orgo Team
 License: MIT