specialagent 0.0.1__tar.gz

specialagent-0.0.1/.github/workflows/pypi.yml

@@ -0,0 +1,24 @@
+name: PyPI
+
+on:
+  push:
+    tags:
+      - 'v*.*.*'
+
+jobs:
+  PyPI:
+    runs-on: ubuntu-latest
+    permissions:
+      id-token: write
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v5
+
+      - name: Setup uv
+        uses: astral-sh/setup-uv@v5
+
+      - name: Build
+        run: uv build
+
+      - name: Publish
+        run: uv publish

specialagent-0.0.1/PKG-INFO

@@ -0,0 +1,8 @@
+Metadata-Version: 2.4
+Name: specialagent
+Version: 0.0.1
+Author-email: Jon Craton <jncraton@gmail.com>
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.9

specialagent-0.0.1/agent.py

@@ -0,0 +1,105 @@
+import json
+import os
+import subprocess
+
+def run_bash(command):
+    """
+    Executes a bash command and returns the output.
+    >>> run_bash('echo "hello"')
+    'hello\\n'
+    """
+    print(f"{command}")
+    
+    result = subprocess.run(command, shell=True, capture_output=True, text=True)
+    return f'{result.stdout}{result.stderr}'
+
+def write_file(path, content):
+    """
+    Writes content to a file at the specified path.
+    >>> import tempfile
+    >>> with tempfile.NamedTemporaryFile() as tmp:
+    ...     write_file(tmp.name, 'test')
+    'File written to '...
+    """
+    with open(path, 'w') as f:
+        f.write(content)
+    return f'File written to {path}'
+
+def call_gemini(messages, tools):
+    import urllib.request
+
+    url = 'https://generativelanguage.googleapis.com/v1beta/models/gemma-4-26b-a4b-it:generateContent'
+    api_key = os.environ.get('GEMINI_API_KEY')
+    headers = {'Content-Type': 'application/json', 'x-goog-api-key': api_key}
+    data = json.dumps({'contents': messages, 'tools': [{'function_declarations': tools}]}).encode()
+    
+    req = urllib.request.Request(url, data=data, headers=headers)
+    with urllib.request.urlopen(req) as response:
+        return json.loads(response.read().decode())
+
+def main():
+    tools = [
+        {
+            'name': 'run_bash',
+            'description': 'Execute a command in the bash shell',
+            'parameters': {
+                'type': 'object',
+                'properties': {'command': {'type': 'string'}},
+                'required': ['command']
+            }
+        },
+        {
+            'name': 'write_file',
+            'description': 'Write text content to a file path',
+            'parameters': {
+                'type': 'object',
+                'properties': {
+                    'path': {'type': 'string'},
+                    'content': {'type': 'string'}
+                },
+                'required': ['path', 'content']
+            }
+        }
+    ]
+
+    messages = []
+    user_input = input('Task: ')
+    messages.append({'role': 'user', 'parts': [{'text': user_input}]})
+
+    while True:
+        response = call_gemini(messages, tools)
+        parts = response['candidates'][0]['content']['parts']
+        text_parts = [p for p in parts if "text" in p]
+        function_calls = [p for p in parts if "functionCall" in p]
+
+        messages.append(response['candidates'][0]['content'])
+
+        if len(function_calls) == 0:
+            print(text_parts[0].get('text', 'Task complete.'))
+            break
+
+        for call in function_calls:
+            name = call['functionCall']['name']
+            args = call['functionCall']['args']
+
+            print(f'Executing {name} with {args}...')
+
+            if name == 'run_bash':
+                result = run_bash(args['command'])
+            elif name == 'write_file':
+                result = write_file(args['path'], args['content'])
+            else:
+                result = 'Error: unknown tool'
+
+            messages.append({
+                'role': 'user',
+                'parts': [{
+                    'functionResponse': {
+                        'name': name,
+                        'response': {'content': result}
+                    }
+                }]
+            })
+
+if __name__ == '__main__':
+    main()

specialagent-0.0.1/pyproject.toml

@@ -0,0 +1,25 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "specialagent"
+version = "0.0.1"
+description = ""
+readme = "readme.md"
+authors = [
+    { name = "Jon Craton", email = "jncraton@gmail.com" }
+]
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+]
+requires-python = ">=3.9"
+dependencies = []
+
+[project.scripts]
+json2qti = "specialagent:main"
+
+[tool.hatch.build.targets.wheel]
+packages = ["specialagent.py"]

specialagent-0.0.1/run

@@ -0,0 +1,45 @@
+#!/bin/bash
+
+# Run within firejail sandbox
+
+firejail \
+  --quiet \
+  --noprofile \
+  --caps.drop=all \
+  --seccomp \
+  --machine-id \
+  --disable-mnt \
+  --private-etc=localtime,ca-certificates,ssl,pki \
+  --private-dev \
+  --no3d \
+  --nodbus \
+  --nodvd \
+  --noinput \
+  --nosound \
+  --notv \
+  --nou2f \
+  --novideo \
+  --nonewprivs \
+  --dns=8.8.8.8 \
+  --dns=8.8.4.4 \
+  --protocol=inet,unix \
+  --hostname="hostname" \
+  --private-tmp \
+  --whitelist="$(pwd)" \
+  --whitelist=~/.profile \
+  --read-only=~/.profile \
+  --whitelist=~/.npm \
+  --whitelist=~/.cargo \
+  --read-only=~/.cargo \
+  --whitelist=~/.local/share/pipx \
+  --whitelist=~/.local/bin \
+  --read-only=~/.local/bin \
+  --whitelist=~/.cache/ms-playwright \
+  --whitelist=~/.cache/huggingface \
+  --whitelist=~/.cache/uv \
+  --rmenv=WINDOWID \
+  --rmenv=USER \
+  --rmenv=JOURNAL_STREAM \
+  --rmenv=DISPLAY \
+  --rmenv=TMUX \
+  python3 agent.py