orgo 0.0.12__tar.gz → 0.0.13__tar.gz

{orgo-0.0.12 → orgo-0.0.13}/PKG-INFO

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

{orgo-0.0.12 → orgo-0.0.13}/pyproject.toml

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

{orgo-0.0.12 → orgo-0.0.13}/src/orgo/computer.py

@@ -1,15 +1,17 @@
 """Computer class for interacting with Orgo virtual environments"""
-# src/orgo/computer.py
-
 import os
 import io
 import base64
+import logging
 from typing import Dict, List, Any, Optional, Callable, Union
 from PIL import Image
+from requests.exceptions import RequestException
 
 from .api.client import ApiClient
 from .prompt import get_provider
+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):
@@ -26,16 +28,31 @@ class Computer:
         self.base_api_url = base_api_url
         self.api = ApiClient(self.api_key, self.base_api_url)
         
+        # Look for a saved project ID if none was provided
+        if project_id is None:
+            project_id = ProjectManager.load_project_id()
+        
         if project_id:
-            self.project_id = project_id
-            self._info = self.api.connect_computer(project_id)
+            try:
+                self.project_id = project_id
+                self._info = self.api.connect_computer(project_id)
+            except (RequestException, ValueError) as e:
+                logger.warning(f"Could not connect to saved project {project_id}: {e}")
+                self._create_new_computer(config)
         else:
-            response = self.api.create_computer(config)
-            self.project_id = response.get("name")
-            self._info = response
+            self._create_new_computer(config)
             
+    def _create_new_computer(self, config=None):
+        """Create a new computer instance and save its ID"""
+        response = self.api.create_computer(config)
+        self.project_id = response.get("name")
+        self._info = response
+        
         if not self.project_id:
             raise ValueError("Failed to initialize computer: No project ID returned")
+            
+        # Save the project ID for future use
+        ProjectManager.save_project_id(self.project_id)
     
     def status(self) -> Dict[str, Any]:
         """Get current computer status"""
@@ -130,28 +147,6 @@ class Computer:
             
         Returns:
             List of messages from the conversation
-            
-        Examples:
-            # Simple usage with environment variables
-            computer.prompt("Open Firefox and search for Python tutorials")
-            
-            # With explicit API key
-            computer.prompt("Open Terminal and list files", api_key="your-anthropic-key")
-            
-            # With callback for progress updates
-            computer.prompt("Create a new text file", callback=my_callback_function)
-            
-            # With thinking enabled (Claude 3.7 Sonnet)
-            computer.prompt(
-                "Analyze a complex webpage", 
-                thinking_enabled=True
-            )
-            
-            # With custom screenshot management
-            computer.prompt(
-                "Perform a complex multi-step task",
-                max_saved_screenshots=10  # Keep more screenshots for complex tasks
-            )
         """
         # Get the provider instance
         provider_instance = get_provider(provider)
@@ -173,4 +168,4 @@ class Computer:
             # Pass through the Orgo API client configuration
             orgo_api_key=self.api_key,
             orgo_base_url=self.base_api_url
-        )
+        )

orgo-0.0.13/src/orgo/project.py

@@ -0,0 +1,50 @@
+"""Project management for Orgo virtual environments"""
+import os
+import json
+import logging
+from typing import Optional
+
+logger = logging.getLogger(__name__)
+
+class ProjectManager:
+    """Manages project persistence for Orgo computers"""
+    
+    @staticmethod
+    def load_project_id() -> Optional[str]:
+        """Load project ID from local config file"""
+        config_path = ProjectManager._get_config_path()
+        
+        if not os.path.exists(config_path):
+            return None
+            
+        try:
+            with open(config_path, 'r') as f:
+                data = json.load(f)
+                return data.get('project_id')
+        except (json.JSONDecodeError, IOError, OSError) as e:
+            logger.warning(f"Error loading project config: {str(e)}")
+            return None
+    
+    @staticmethod
+    def save_project_id(project_id: str) -> None:
+        """Save project ID to local config file"""
+        config_dir = ProjectManager._get_project_dir()
+        config_path = ProjectManager._get_config_path()
+        
+        try:
+            os.makedirs(config_dir, exist_ok=True)
+            with open(config_path, 'w') as f:
+                json.dump({'project_id': project_id}, f, indent=2)
+        except (IOError, OSError) as e:
+            logger.error(f"Failed to save project ID: {str(e)}")
+            raise RuntimeError(f"Failed to save project configuration: {str(e)}") from e
+    
+    @staticmethod
+    def _get_project_dir() -> str:
+        """Get the project directory path"""
+        return os.path.join(os.getcwd(), ".orgo")
+    
+    @staticmethod
+    def _get_config_path() -> str:
+        """Get the full path to the config file"""
+        return os.path.join(ProjectManager._get_project_dir(), "project.json")

{orgo-0.0.12 → orgo-0.0.13}/src/orgo.egg-info/PKG-INFO

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

{orgo-0.0.12 → orgo-0.0.13}/src/orgo.egg-info/SOURCES.txt

@@ -2,6 +2,7 @@ README.md
 pyproject.toml
 src/orgo/__init__.py
 src/orgo/computer.py
+src/orgo/project.py
 src/orgo/prompt.py
 src/orgo.egg-info/PKG-INFO
 src/orgo.egg-info/SOURCES.txt