aion-agent 0.1.0__tar.gz

aion_agent-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Sheikh Abdul Munim
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

aion_agent-0.1.0/PKG-INFO

@@ -0,0 +1,219 @@
+Metadata-Version: 2.4
+Name: aion-agent
+Version: 0.1.0
+Summary: AI calendar scheduling agent for Google Calendar
+Author: Sheikh Abdul Munim
+License: MIT
+Project-URL: Homepage, https://github.com/sheikhmunim/Aion
+Project-URL: Repository, https://github.com/sheikhmunim/Aion
+Project-URL: Issues, https://github.com/sheikhmunim/Aion/issues
+Keywords: calendar,scheduling,google-calendar,cli,agent
+Classifier: Development Status :: 3 - Alpha
+Classifier: Environment :: Console
+Classifier: Intended Audience :: End Users/Desktop
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Office/Business :: Scheduling
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: clingo>=5.6.0
+Requires-Dist: httpx>=0.25.0
+Requires-Dist: rich>=13.0.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: ruff>=0.1.0; extra == "dev"
+Dynamic: license-file
+
+# Aion
+
+**AI-powered calendar agent for Google Calendar.**
+
+Schedule, list, reschedule, and find free time — all from natural language in your terminal.
+
+```
+  aion > schedule gym tomorrow morning
+  Finding optimal slot for 'gym'...
+  Schedule 'gym' on February 19, 2026 at 07:00 for 60 min? [y/n]: y
+  ✔ Created! 'gym' on 2026-02-19 at 07:00
+```
+
+Part of [A.U.R.A](https://github.com/sheikhmunim) (Autonomous Unified Reasoning Assistant).
+
+---
+
+## Features
+
+- **Natural language** — "schedule dentist friday at 2pm for 45 min", "what's on tomorrow?"
+- **Smart scheduling** — ASP/Clingo constraint solver finds optimal slots avoiding conflicts
+- **Google Calendar sync** — reads and writes real events via Calendar API v3
+- **Conflict detection** — warns on overlaps, offers alternatives
+- **User preferences** — block time slots (lunch, sleep), set default morning/afternoon/evening
+- **Timezone-aware** — auto-detects your timezone from Google Calendar on login
+- **Ollama NLU** (optional) — local LLM fallback for complex commands, auto-installs on first run
+
+---
+
+## Architecture
+
+```
+User Input
+    │
+    ▼
+┌──────────────┐    ┌──────────────┐
+│ Regex NLU    │───▶│ Ollama LLM   │  (optional fallback)
+│ (intent.py)  │    │ (ollama.py)  │
+└──────┬───────┘    └──────────────┘
+       │
+       ▼
+┌──────────────┐    ┌──────────────┐
+│ ASP Solver   │───▶│    Clingo    │  (constraint solving)
+│ (solver.py)  │    │              │
+└──────┬───────┘    └──────────────┘
+       │
+       ▼
+┌──────────────┐
+│ Google Cal   │  (httpx async)
+│ (google_cal) │
+└──────────────┘
+```
+
+---
+
+## Quick Start
+
+```bash
+pip install aion-agent
+aion login
+aion
+```
+
+That's it. `aion login` opens your browser for Google sign-in. Your timezone is auto-detected. No API keys or configuration needed.
+
+On first run, Aion offers to install [Ollama](https://ollama.com) for smarter natural language understanding — this is optional.
+
+---
+
+## Installation
+
+**From PyPI:**
+
+```bash
+pip install aion-agent
+```
+
+**From source:**
+
+```bash
+git clone https://github.com/sheikhmunim/Aion.git
+cd Aion
+pip install -e .
+```
+
+Requires **Python 3.10+**.
+
+---
+
+## Usage
+
+Start the interactive CLI:
+
+```bash
+aion
+```
+
+### Commands
+
+| Action | Examples |
+|--------|----------|
+| **Schedule** | `schedule gym tomorrow morning`, `add meeting at 3pm for 90 min` |
+| **List** | `what's on today?`, `show my calendar this week`, `what tomorrow?` |
+| **Delete** | `cancel gym tomorrow`, `delete meeting` |
+| **Update** | `move gym to 3pm`, `reschedule meeting to friday` |
+| **Free slots** | `when am I free tomorrow?`, `free slots this week` |
+| **Best time** | `best time for a 2h study session` |
+| **Preferences** | `preferences` — manage blocked times and defaults |
+| **Login/Logout** | `login`, `logout` |
+| **Help** | `help` |
+| **Quit** | `quit` or `exit` |
+
+### Preferences
+
+Block recurring time slots and set defaults:
+
+```
+aion > preferences
+  ┌─────────────────────────────────────────────┐
+  │ 1. Add a blocked time slot                  │
+  │ 2. Remove a blocked slot                    │
+  │ 3. Change default time preference           │
+  │ 4. Back                                     │
+  └─────────────────────────────────────────────┘
+```
+
+Blocked slots (e.g. lunch 12:00-13:00 on weekdays) are respected by the scheduler — it won't suggest times during those windows.
+
+---
+
+## Configuration
+
+Config lives at `~/.aion/config.json`. All options can also be set via environment variables with `AION_` prefix.
+
+| Key | Env var | Default | Description |
+|-----|---------|---------|-------------|
+| `google_client_id` | `AION_GOOGLE_CLIENT_ID` | Built-in | OAuth client ID (override with your own if needed) |
+| `google_client_secret` | `AION_GOOGLE_CLIENT_SECRET` | Built-in | OAuth client secret |
+| `timezone` | `AION_TIMEZONE` | `UTC` | IANA timezone (auto-detected on login) |
+| `default_duration` | `AION_DEFAULT_DURATION` | `60` | Default event duration in minutes |
+| `ollama_url` | `AION_OLLAMA_URL` | `http://localhost:11434` | Ollama server URL |
+| `ollama_model` | `AION_OLLAMA_MODEL` | `qwen2.5:0.5b` | Ollama model for NLU |
+
+---
+
+## Development
+
+```bash
+# Install with dev dependencies
+pip install -e ".[dev]"
+
+# Run tests
+pytest tests/ -v
+
+# Lint
+ruff check aion/
+```
+
+---
+
+## How it works
+
+1. **Intent classification** — Regex patterns match commands (schedule, list, delete, etc.) with confidence scores. Falls back to Ollama LLM for ambiguous input.
+
+2. **Date parsing** — Handles "today", "tomorrow", weekday names, "this/next week", specific dates like "March 5th", and common typos.
+
+3. **Constraint solving** — The ASP/Clingo solver models the day as 30-minute slots (6AM-10PM), marks busy times from existing events and user preferences, then finds optimal placements with time-of-day preferences.
+
+4. **Google Calendar API** — All reads/writes go through Calendar API v3 via httpx async. Token refresh is automatic.
+
+---
+
+## Dependencies
+
+| Package | Purpose |
+|---------|---------|
+| [clingo](https://potassco.org/clingo/) | ASP constraint solver |
+| [httpx](https://www.python-httpx.org/) | Async HTTP client (Google Calendar + Ollama) |
+| [rich](https://rich.readthedocs.io/) | Terminal UI |
+
+---
+
+## Privacy
+
+Aion runs entirely on your machine. No calendar data is sent to external servers. See [PRIVACY.md](PRIVACY.md) for details.
+
+## License
+
+MIT License. See [LICENSE](LICENSE).

aion_agent-0.1.0/README.md

@@ -0,0 +1,189 @@
+# Aion
+
+**AI-powered calendar agent for Google Calendar.**
+
+Schedule, list, reschedule, and find free time — all from natural language in your terminal.
+
+```
+  aion > schedule gym tomorrow morning
+  Finding optimal slot for 'gym'...
+  Schedule 'gym' on February 19, 2026 at 07:00 for 60 min? [y/n]: y
+  ✔ Created! 'gym' on 2026-02-19 at 07:00
+```
+
+Part of [A.U.R.A](https://github.com/sheikhmunim) (Autonomous Unified Reasoning Assistant).
+
+---
+
+## Features
+
+- **Natural language** — "schedule dentist friday at 2pm for 45 min", "what's on tomorrow?"
+- **Smart scheduling** — ASP/Clingo constraint solver finds optimal slots avoiding conflicts
+- **Google Calendar sync** — reads and writes real events via Calendar API v3
+- **Conflict detection** — warns on overlaps, offers alternatives
+- **User preferences** — block time slots (lunch, sleep), set default morning/afternoon/evening
+- **Timezone-aware** — auto-detects your timezone from Google Calendar on login
+- **Ollama NLU** (optional) — local LLM fallback for complex commands, auto-installs on first run
+
+---
+
+## Architecture
+
+```
+User Input
+    │
+    ▼
+┌──────────────┐    ┌──────────────┐
+│ Regex NLU    │───▶│ Ollama LLM   │  (optional fallback)
+│ (intent.py)  │    │ (ollama.py)  │
+└──────┬───────┘    └──────────────┘
+       │
+       ▼
+┌──────────────┐    ┌──────────────┐
+│ ASP Solver   │───▶│    Clingo    │  (constraint solving)
+│ (solver.py)  │    │              │
+└──────┬───────┘    └──────────────┘
+       │
+       ▼
+┌──────────────┐
+│ Google Cal   │  (httpx async)
+│ (google_cal) │
+└──────────────┘
+```
+
+---
+
+## Quick Start
+
+```bash
+pip install aion-agent
+aion login
+aion
+```
+
+That's it. `aion login` opens your browser for Google sign-in. Your timezone is auto-detected. No API keys or configuration needed.
+
+On first run, Aion offers to install [Ollama](https://ollama.com) for smarter natural language understanding — this is optional.
+
+---
+
+## Installation
+
+**From PyPI:**
+
+```bash
+pip install aion-agent
+```
+
+**From source:**
+
+```bash
+git clone https://github.com/sheikhmunim/Aion.git
+cd Aion
+pip install -e .
+```
+
+Requires **Python 3.10+**.
+
+---
+
+## Usage
+
+Start the interactive CLI:
+
+```bash
+aion
+```
+
+### Commands
+
+| Action | Examples |
+|--------|----------|
+| **Schedule** | `schedule gym tomorrow morning`, `add meeting at 3pm for 90 min` |
+| **List** | `what's on today?`, `show my calendar this week`, `what tomorrow?` |
+| **Delete** | `cancel gym tomorrow`, `delete meeting` |
+| **Update** | `move gym to 3pm`, `reschedule meeting to friday` |
+| **Free slots** | `when am I free tomorrow?`, `free slots this week` |
+| **Best time** | `best time for a 2h study session` |
+| **Preferences** | `preferences` — manage blocked times and defaults |
+| **Login/Logout** | `login`, `logout` |
+| **Help** | `help` |
+| **Quit** | `quit` or `exit` |
+
+### Preferences
+
+Block recurring time slots and set defaults:
+
+```
+aion > preferences
+  ┌─────────────────────────────────────────────┐
+  │ 1. Add a blocked time slot                  │
+  │ 2. Remove a blocked slot                    │
+  │ 3. Change default time preference           │
+  │ 4. Back                                     │
+  └─────────────────────────────────────────────┘
+```
+
+Blocked slots (e.g. lunch 12:00-13:00 on weekdays) are respected by the scheduler — it won't suggest times during those windows.
+
+---
+
+## Configuration
+
+Config lives at `~/.aion/config.json`. All options can also be set via environment variables with `AION_` prefix.
+
+| Key | Env var | Default | Description |
+|-----|---------|---------|-------------|
+| `google_client_id` | `AION_GOOGLE_CLIENT_ID` | Built-in | OAuth client ID (override with your own if needed) |
+| `google_client_secret` | `AION_GOOGLE_CLIENT_SECRET` | Built-in | OAuth client secret |
+| `timezone` | `AION_TIMEZONE` | `UTC` | IANA timezone (auto-detected on login) |
+| `default_duration` | `AION_DEFAULT_DURATION` | `60` | Default event duration in minutes |
+| `ollama_url` | `AION_OLLAMA_URL` | `http://localhost:11434` | Ollama server URL |
+| `ollama_model` | `AION_OLLAMA_MODEL` | `qwen2.5:0.5b` | Ollama model for NLU |
+
+---
+
+## Development
+
+```bash
+# Install with dev dependencies
+pip install -e ".[dev]"
+
+# Run tests
+pytest tests/ -v
+
+# Lint
+ruff check aion/
+```
+
+---
+
+## How it works
+
+1. **Intent classification** — Regex patterns match commands (schedule, list, delete, etc.) with confidence scores. Falls back to Ollama LLM for ambiguous input.
+
+2. **Date parsing** — Handles "today", "tomorrow", weekday names, "this/next week", specific dates like "March 5th", and common typos.
+
+3. **Constraint solving** — The ASP/Clingo solver models the day as 30-minute slots (6AM-10PM), marks busy times from existing events and user preferences, then finds optimal placements with time-of-day preferences.
+
+4. **Google Calendar API** — All reads/writes go through Calendar API v3 via httpx async. Token refresh is automatic.
+
+---
+
+## Dependencies
+
+| Package | Purpose |
+|---------|---------|
+| [clingo](https://potassco.org/clingo/) | ASP constraint solver |
+| [httpx](https://www.python-httpx.org/) | Async HTTP client (Google Calendar + Ollama) |
+| [rich](https://rich.readthedocs.io/) | Terminal UI |
+
+---
+
+## Privacy
+
+Aion runs entirely on your machine. No calendar data is sent to external servers. See [PRIVACY.md](PRIVACY.md) for details.
+
+## License
+
+MIT License. See [LICENSE](LICENSE).

aion_agent-0.1.0/aion/__init__.py

@@ -0,0 +1,3 @@
+"""Aion — AI calendar scheduling agent for Google Calendar."""
+
+__version__ = "0.1.0"

aion_agent-0.1.0/aion/__main__.py

@@ -0,0 +1,5 @@
+"""Allow running as `python -m aion`."""
+
+from aion.cli import main
+
+main()

aion_agent-0.1.0/aion/asp_model.py

@@ -0,0 +1,200 @@
+"""ASP Model for Calendar Scheduling — generates Answer Set Programs for Clingo."""
+
+from datetime import datetime, timedelta
+
+from aion.config import get_now
+
+
+class ASPModel:
+    """Generates ASP rules for calendar scheduling.
+
+    Time model:
+    - Day divided into 30-minute slots (0-31)
+    - Slot 0 = 6:00 AM, Slot 31 = 9:30 PM
+    - Working hours: slots 6-24 (9:00 AM - 6:00 PM)
+    """
+
+    def __init__(self):
+        self.slots_per_hour = 2  # 30-minute slots
+        self.day_start_hour = 6  # 6:00 AM
+        self.day_end_hour = 22   # 10:00 PM
+        self.total_slots = (self.day_end_hour - self.day_start_hour) * self.slots_per_hour
+
+    def time_to_slot(self, time_str: str) -> int:
+        h, m = map(int, time_str.split(":"))
+        return (h - self.day_start_hour) * self.slots_per_hour + (1 if m >= 30 else 0)
+
+    def slot_to_time(self, slot: int) -> str:
+        h = self.day_start_hour + slot // self.slots_per_hour
+        m = 30 if slot % self.slots_per_hour else 0
+        return f"{h:02d}:{m:02d}"
+
+    def duration_to_slots(self, minutes: int) -> int:
+        return (minutes + 29) // 30
+
+    def date_to_weekday(self, date_str: str) -> str:
+        dt = datetime.strptime(date_str, "%Y-%m-%d")
+        return dt.strftime("%A").lower()
+
+    def get_week_dates(self, start_date: str | None = None) -> list[str]:
+        if start_date:
+            start = datetime.strptime(start_date, "%Y-%m-%d")
+        else:
+            start = get_now()
+        start = start - timedelta(days=start.weekday())
+        return [(start + timedelta(days=i)).strftime("%Y-%m-%d") for i in range(7)]
+
+    def generate_base_program(self) -> str:
+        return f"""
+% Time slots: 0 to {self.total_slots - 1} (30-min blocks from 6AM to 10PM)
+time_slot(0..{self.total_slots - 1}).
+
+% Days of the week
+day(monday; tuesday; wednesday; thursday; friday; saturday; sunday).
+
+% Day types
+weekday(monday; tuesday; wednesday; thursday; friday).
+weekend(saturday; sunday).
+
+% Working hours: 9AM-6PM (slots 6-24)
+working_hour(6..24).
+
+% Morning: 6AM-12PM (slots 0-12)
+morning(0..12).
+
+% Afternoon: 12PM-6PM (slots 12-24)
+afternoon(12..24).
+
+% Evening: 6PM-10PM (slots 24-32)
+evening(24..{self.total_slots - 1}).
+"""
+
+    def generate_busy_constraints(self, events: list[dict], dates: list[str] | None = None) -> str:
+        lines = ["\n% Busy times from existing events"]
+        for event in events:
+            if dates and event["date"] not in dates:
+                continue
+            weekday = self.date_to_weekday(event["date"])
+            start_slot = self.time_to_slot(event["time"])
+            duration_slots = self.duration_to_slots(event["duration"])
+            for slot in range(start_slot, min(start_slot + duration_slots, self.total_slots)):
+                lines.append(f'busy({weekday}, {slot}, "{event["date"]}").')
+        return "\n".join(lines)
+
+    def generate_scheduling_request(self, request: dict) -> str:
+        activity = request.get("activity", "event").replace(" ", "_").lower()
+        duration_slots = self.duration_to_slots(request.get("duration", 60))
+        count = request.get("count", 1)
+        specific_date = request.get("date")
+
+        lines = [f"\n% Scheduling request: {activity}"]
+        lines.append(f'activity("{activity}").')
+        lines.append(f'duration("{activity}", {duration_slots}).')
+        lines.append(f'need_count("{activity}", {count}).')
+
+        if specific_date:
+            weekday = self.date_to_weekday(specific_date)
+            lines.append(f'allowed_day("{activity}", {weekday}).')
+            lines.append(f'target_date("{activity}", "{specific_date}").')
+        else:
+            allowed_days = request.get("days", [
+                "monday", "tuesday", "wednesday", "thursday", "friday", "saturday", "sunday"
+            ])
+            for day in allowed_days:
+                lines.append(f'allowed_day("{activity}", {day}).')
+
+        lines.append(f"""
+% Generate exactly {count} time slot(s) for {activity}
+{count} {{ schedule("{activity}", D, T) : allowed_day("{activity}", D), time_slot(T) }} {count}.
+""")
+
+        lines.append(f"""
+% Cannot overlap with busy times
+:- schedule("{activity}", D, T), busy(D, T, _).
+""")
+
+        if duration_slots > 1:
+            lines.append(f"""
+% Ensure full duration is available
+:- schedule("{activity}", D, T), duration("{activity}", Dur),
+   Offset = 1..Dur-1, busy(D, T+Offset, _).
+""")
+
+        lines.append(f"""
+% Don't exceed end of day
+:- schedule("{activity}", D, T), duration("{activity}", Dur), T + Dur > {self.total_slots}.
+""")
+
+        if count > 1:
+            lines.append(f"""
+% Spread sessions across different days
+:- schedule("{activity}", D, T1), schedule("{activity}", D, T2), T1 != T2.
+""")
+
+        if request.get("avoid_weekends"):
+            lines.append(f"""
+:- schedule("{activity}", D, _), weekend(D).
+""")
+
+        if request.get("working_hours_only"):
+            lines.append(f"""
+:- schedule("{activity}", D, T), not working_hour(T).
+""")
+
+        if request.get("prefer_morning"):
+            lines.append(f"""
+#minimize {{ T@1,D : schedule("{activity}", D, T), not morning(T) }}.
+#minimize {{ T@2,D : schedule("{activity}", D, T) }}.
+""")
+        elif request.get("prefer_afternoon"):
+            lines.append(f"""
+#minimize {{ 1@1,D,T : schedule("{activity}", D, T), not afternoon(T) }}.
+""")
+        elif request.get("prefer_evening"):
+            lines.append(f"""
+#minimize {{ 1@1,D,T : schedule("{activity}", D, T), not evening(T) }}.
+""")
+        else:
+            lines.append(f"""
+#minimize {{ T@1,D : schedule("{activity}", D, T) }}.
+""")
+
+        lines.append("\n#show schedule/3.")
+        return "\n".join(lines)
+
+    def generate_preference_constraints(
+        self, blocked_slots: list[dict], target_date: str | None = None
+    ) -> str:
+        """Convert blocked preference slots into ASP busy facts.
+
+        If target_date is given, only generate facts for that date's weekday.
+        """
+        today = get_now().strftime("%Y-%m-%d")
+        lines = ["\n% Blocked times from user preferences"]
+
+        for block in blocked_slots:
+            until = block.get("until")
+            if until and until < today:
+                continue
+
+            start_slot = self.time_to_slot(block["start"])
+            end_slot = self.time_to_slot(block["end"])
+
+            if target_date:
+                weekday = self.date_to_weekday(target_date)
+                if weekday not in block.get("days", []):
+                    continue
+                for slot in range(start_slot, min(end_slot, self.total_slots)):
+                    lines.append(f'busy({weekday}, {slot}, "preference").')
+            else:
+                for day in block.get("days", []):
+                    for slot in range(start_slot, min(end_slot, self.total_slots)):
+                        lines.append(f'busy({day}, {slot}, "preference").')
+
+        return "\n".join(lines)
+
+    def generate_full_program(self, events: list[dict], request: dict, dates: list[str] | None = None) -> str:
+        program = self.generate_base_program()
+        program += self.generate_busy_constraints(events, dates)
+        program += self.generate_scheduling_request(request)
+        return program