-
Notifications
You must be signed in to change notification settings - Fork 110
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
# Conflicts:
# agentstack/main.py
# agentstack/templates/crewai/{{cookiecutter.project_metadata.project_slug}}/pyproject.toml- Loading branch information
Showing
67 changed files
with
8,650 additions
and
311 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,43 @@ | ||
| name: Codecov | ||
|
|
||
| on: | ||
| push: | ||
| branches: | ||
| - main | ||
| paths: | ||
| - 'agentstack/**/*.py' | ||
| pull_request: | ||
| branches: | ||
| - main | ||
| paths: | ||
| - 'agentstack/**/*.py' | ||
|
|
||
| jobs: | ||
| test: | ||
| runs-on: ubuntu-latest | ||
|
|
||
| steps: | ||
| - name: Checkout code | ||
| uses: actions/checkout@v2 | ||
|
|
||
| - name: Set up Python | ||
| uses: actions/setup-python@v2 | ||
| with: | ||
| python-version: 3.11 | ||
|
|
||
| - name: Install dependencies | ||
| run: | | ||
| python -m pip install --upgrade pip | ||
| pip install tox | ||
| - name: Run tests with tox | ||
| run: tox | ||
|
|
||
| - name: Upload coverage to Codecov | ||
| uses: codecov/codecov-action@v2 | ||
| with: | ||
| token: ${{ secrets.CODECOV_TOKEN }} | ||
| files: ./coverage.xml | ||
| flags: unittests | ||
| name: codecov-umbrella | ||
| fail_ci_if_error: true |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,38 @@ | ||
| name: Mintlify Documentation Check | ||
| on: | ||
| pull_request: | ||
| paths: | ||
| - 'docs/**' # Only trigger on changes to docs directory | ||
| - 'mint.json' # Also trigger on mintlify config changes | ||
|
|
||
| jobs: | ||
| build-docs: | ||
| name: Build Documentation | ||
| runs-on: ubuntu-latest | ||
|
|
||
| steps: | ||
| - name: Checkout code | ||
| uses: actions/checkout@v4 | ||
|
|
||
| - name: Setup Node.js | ||
| uses: actions/setup-node@v4 | ||
| with: | ||
| node-version: '18' | ||
|
|
||
| - name: Install Mintlify | ||
| run: npm install -g mintlify | ||
|
|
||
| - name: Test documentation | ||
| run: | | ||
| cd docs | ||
| # If mintlify dev has errors, it will exit with status 1 | ||
| # If it starts successfully, kill it after 5 seconds | ||
| timeout 5s mintlify dev || exit_status=$? | ||
| if [ $exit_status -eq 124 ]; then | ||
| # timeout exit code 124 means the process was killed after starting successfully | ||
| echo "Documentation built successfully!" | ||
| exit 0 | ||
| else | ||
| echo "Documentation failed to build! Try running `mintlify dev` from the docs dir locally" | ||
| exit 1 | ||
| fi |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,142 @@ | ||
| from http.server import HTTPServer, BaseHTTPRequestHandler | ||
| from urllib.parse import parse_qs, urlparse | ||
| import webbrowser | ||
| import json | ||
| import os | ||
| import threading | ||
| import socket | ||
| from pathlib import Path | ||
|
|
||
| import inquirer | ||
| from appdirs import user_data_dir | ||
| from agentstack.logger import log | ||
|
|
||
|
|
||
| try: | ||
| base_dir = Path(user_data_dir("agentstack", "agency")) | ||
| # Test if we can write to directory | ||
| test_file = base_dir / '.test_write_permission' | ||
| test_file.touch() | ||
| test_file.unlink() | ||
| except (RuntimeError, OSError, PermissionError): | ||
| # In CI or when directory is not writable, use temp directory | ||
| base_dir = Path(os.getenv('TEMP', '/tmp')) | ||
|
|
||
|
|
||
| class AuthCallbackHandler(BaseHTTPRequestHandler): | ||
| def do_GET(self): | ||
| """Handle the OAuth callback from the browser""" | ||
| try: | ||
| # Parse the query parameters | ||
| query_components = parse_qs(urlparse(self.path).query) | ||
|
|
||
| # Extract the token from query parameters | ||
| token = query_components.get('token', [''])[0] | ||
|
|
||
| if token: | ||
| # Store the token | ||
| base_dir.mkdir(exist_ok=True, parents=True) | ||
|
|
||
| with open(base_dir / 'auth.json', 'w') as f: | ||
| json.dump({'bearer_token': token}, f) | ||
|
|
||
| # Send success response | ||
| self.send_response(200) | ||
| self.send_header('Content-type', 'text/html') | ||
| self.end_headers() | ||
|
|
||
| success_html = """ | ||
| <html> | ||
| <body> | ||
| <script> | ||
| setTimeout(function() { | ||
| window.close(); | ||
| }, 1000); | ||
| </script> | ||
| <h2>Authentication successful! You can close this window.</h2> | ||
| </body> | ||
| </html> | ||
| """ | ||
| self.wfile.write(success_html.encode()) | ||
|
|
||
| # Signal the main thread that we're done | ||
| self.server.authentication_successful = True | ||
| else: | ||
| self.send_response(400) | ||
| self.send_header('Content-type', 'text/html') | ||
| self.end_headers() | ||
| self.wfile.write(b'Authentication failed: No token received') | ||
|
|
||
| except Exception as e: | ||
| self.send_response(500) | ||
| self.send_header('Content-type', 'text/html') | ||
| self.end_headers() | ||
| self.wfile.write(f'Error: {str(e)}'.encode()) | ||
|
|
||
| def find_free_port(): | ||
| """Find a free port on localhost""" | ||
| with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s: | ||
| s.bind(('', 0)) | ||
| s.listen(1) | ||
| port = s.getsockname()[1] | ||
| return port | ||
|
|
||
| def start_auth_server(): | ||
| """Start the local authentication server""" | ||
| port = find_free_port() | ||
| server = HTTPServer(('localhost', port), AuthCallbackHandler) | ||
| server.authentication_successful = False | ||
| return server, port | ||
|
|
||
|
|
||
| def login(): | ||
| """Log in to AgentStack""" | ||
| try: | ||
| # check if already logged in | ||
| token = get_stored_token() | ||
| if token: | ||
| print("You are already authenticated!") | ||
| if not inquirer.confirm('Would you like to log in with a different account?'): | ||
| return | ||
|
|
||
| # Start the local server | ||
| server, port = start_auth_server() | ||
|
|
||
| # Create server thread | ||
| server_thread = threading.Thread(target=server.serve_forever) | ||
| server_thread.daemon = True | ||
| server_thread.start() | ||
|
|
||
| # Open the browser to the login page | ||
| auth_url_base = os.getenv('AGENTSTACK_AUTHORIZATION_BASE_URL', 'https://agentstack.sh') | ||
| auth_url = f"{auth_url_base}/login?callback_port={port}" | ||
| webbrowser.open(auth_url) | ||
|
|
||
| # Wait for authentication to complete | ||
| while not server.authentication_successful: | ||
| pass | ||
|
|
||
| # Cleanup | ||
| server.shutdown() | ||
| server_thread.join() | ||
|
|
||
| print("🔐 Authentication successful! Token has been stored.") | ||
| return True | ||
|
|
||
| except Exception as e: | ||
| log.warn(f"Authentication failed: {str(e)}", err=True) | ||
| return False | ||
|
|
||
|
|
||
| def get_stored_token(): | ||
| """Retrieve the stored bearer token""" | ||
| try: | ||
| auth_path = base_dir / 'auth.json' | ||
| if not auth_path.exists(): | ||
| return None | ||
|
|
||
| with open(auth_path) as f: | ||
| config = json.load(f) | ||
| return config.get('bearer_token') | ||
| except Exception: | ||
| return None |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.