orgo 0.0.20__py3-none-any.whl → 0.0.22__py3-none-any.whl

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/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,14 @@ 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,
+                 memory: Optional[Literal[2, 4]] = None,
+                 cpu: Optional[Literal[2, 4]] = None):
         """
         Initialize an Orgo virtual computer.
         
@@ -25,11 +31,20 @@ 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
+            memory: Alternative parameter for RAM in GB (2 or 4) - only used when creating
+            cpu: CPU cores for new computer (2 or 4) - only used when creating
+        
+        Note: If both ram and memory are provided, ram takes precedence.
         """
         self.api_key = api_key or os.environ.get("ORGO_API_KEY")
         self.base_api_url = base_api_url
         self.api = ApiClient(self.api_key, self.base_api_url)
         
+        # Handle memory parameter as an alias for ram
+        if ram is None and memory is not None:
+            ram = memory
+        
         # Look for a saved project ID if none was provided
         if project_id is None:
             project_id = ProjectManager.load_project_id()
@@ -38,14 +53,39 @@ class Computer:
             try:
                 self.project_id = project_id
                 self._info = self.api.connect_computer(project_id)
+                # Log if ram/memory/cpu were provided but ignored
+                if ram is not None or memory is not None or cpu is not None:
+                    logger.info("Note: ram, memory, 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/memory 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

{orgo-0.0.20.dist-info → orgo-0.0.22.dist-info}/METADATA

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

{orgo-0.0.20.dist-info → orgo-0.0.22.dist-info}/RECORD

@@ -1,12 +1,12 @@
 orgo/__init__.py,sha256=TlOzDJqRKotAam631MdZcz8ypAqyWrLo-Px8HWLkrD0,131
-orgo/computer.py,sha256=zp5cqkaSHHMIUTg04VfpUunlSPLKDrzlOPrUToxVQhw,9410
+orgo/computer.py,sha256=AD9k9DijHnu-wIwu4p3nPasgsQBRqdrHCtbvUyDjUaE,11390
 orgo/project.py,sha256=F90mYJKRZAoFSFw8HVAiosMoLyzVlf0QaKJgTyXoJ4A,2267
 orgo/prompt.py,sha256=ynblwXPTDp_aF1MbGBsY0PIEr9naklDaKFcfSE_EZ6E,19781
 orgo/api/__init__.py,sha256=9Tzb_OPJ5DH7Cg7OrHzpZZUT4ip05alpa9RLDYmnId8,113
-orgo/api/client.py,sha256=mCm1m_8k3DCCYSM0JvY29BI6HHlH0PjVXNnENHhvMvU,5799
+orgo/api/client.py,sha256=0pXH69jVwNvtmw2VdFcqkgFpkZANt8O5BjewULOcmJc,5843
 orgo/utils/__init__.py,sha256=W4G_nwGBf_7jy0w_mfcrkllurYHSRU4B5cMTVYH_uCc,123
 orgo/utils/auth.py,sha256=tPLBJY-6gdBQWLUjUbwIwxHphC3KoRT_XgP3Iykw3Mw,509
-orgo-0.0.20.dist-info/METADATA,sha256=3dT1Vy4T4bcQ9RxOb1fqo-dVYCFCSnMcQ8aT7wQvEgo,822
-orgo-0.0.20.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-orgo-0.0.20.dist-info/top_level.txt,sha256=q0rYtFji8GbYuhFW8A5Ab9e0j27761IKPhnL0E9xow4,5
-orgo-0.0.20.dist-info/RECORD,,
+orgo-0.0.22.dist-info/METADATA,sha256=cvlBaNACC6-VaF1PPtwdyBaBRd6q52vNmUGKGlxBXfs,822
+orgo-0.0.22.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+orgo-0.0.22.dist-info/top_level.txt,sha256=q0rYtFji8GbYuhFW8A5Ab9e0j27761IKPhnL0E9xow4,5
+orgo-0.0.22.dist-info/RECORD,,