orgo 0.0.4__tar.gz → 0.0.5__tar.gz

{orgo-0.0.4 → orgo-0.0.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: orgo
-Version: 0.0.4
+Version: 0.0.5
 Summary: Desktop infrastructure for AI agents
 Author: Orgo Team
 License: MIT

{orgo-0.0.4 → orgo-0.0.5}/pyproject.toml

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

{orgo-0.0.4 → orgo-0.0.5}/src/orgo/computer.py

@@ -3,31 +3,21 @@
 import os
 import io
 import base64
-from typing import Dict, Any, Optional
+from typing import Dict, Any
 from PIL import Image
 
 from .api.client import ApiClient
 
 class Computer:
-    def __init__(self, project_id: Optional[str] = None, api_key: Optional[str] = None, config: Optional[Dict[str, Any]] = None):
-        """
-        Initialize a computer instance.
-        
-        Args:
-            project_id: Existing project ID to connect to
-            api_key: API key for authentication (falls back to ORGO_API_KEY env var)
-            config: Configuration for creating a new computer
-        """
+    def __init__(self, project_id=None, api_key=None, config=None):
         self.api = ApiClient(api_key or os.environ.get("ORGO_API_KEY"))
-        self._info = {}
         
         if project_id:
-            # Simply set the project ID if provided
             self.project_id = project_id
+            self._info = self.api.connect_computer(project_id)
         else:
-            # Create a new computer instance
             response = self.api.create_computer(config)
-            self.project_id = response.get("id")
+            self.project_id = response.get("name")
             self._info = response
             
         if not self.project_id:
@@ -75,13 +65,13 @@ class Computer:
     def screenshot(self) -> Image.Image:
         """Capture screenshot and return as PIL Image"""
         response = self.api.get_screenshot(self.project_id)
-        img_data = base64.b64decode(response.get("image", ""))
+        img_data = base64.b64decode(response.get("screenshot", ""))
         return Image.open(io.BytesIO(img_data))
     
     def screenshot_base64(self) -> str:
         """Capture screenshot and return as base64 string"""
         response = self.api.get_screenshot(self.project_id)
-        return response.get("image", "")
+        return response.get("screenshot", "")
     
     # Execution methods
     def bash(self, command: str) -> str:

{orgo-0.0.4 → orgo-0.0.5}/src/orgo.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: orgo
-Version: 0.0.4
+Version: 0.0.5
 Summary: Desktop infrastructure for AI agents
 Author: Orgo Team
 License: MIT