llm/SIA
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added restart script and fixed dockerfile and run to match docs

Browse Source
This commit is contained in:
Geens
2025-03-01 13:29:17 +01:00
parent 1d5b687e00
commit b7e95d7398
30 changed files with 3047 additions and 1640 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
sia/config.py
View File

@@ -15,26 +15,26 @@ class Config:
parser.add_argument(
'--system-prompt',
type=Path,
default=os.getenv('SIA_SYSTEM_PROMPT', 'system_prompt.md'),
help='Path to the system prompt file (default: system_prompt.md, env: SIA_SYSTEM_PROMPT)'
default=os.getenv('SIA_SYSTEM_PROMPT', '/root/sia/system_prompt.md'),
help='Path to the system prompt file (default: /root/sia/system_prompt.md, env: SIA_SYSTEM_PROMPT)'
)
parser.add_argument(
'--action-schema',
type=Path,
default=os.getenv('SIA_ACTION_SCHEMA', 'action_schema.xsd'),
help='Path to the action schema file (default: action_schema.xsd, env: SIA_ACTION_SCHEMA)'
default=os.getenv('SIA_ACTION_SCHEMA', '/root/sia/action_schema.xsd'),
help='Path to the action schema file (default: /root/sia/action_schema.xsd, env: SIA_ACTION_SCHEMA)'
)
parser.add_argument(
'--iterations-dir',
type=Path,
default=os.getenv('SIA_ITERATIONS_DIR', 'iterations'),
help='Path to the directory for storing iterations (default: iterations, env: SIA_ITERATIONS_DIR)'
default=os.getenv('SIA_ITERATIONS_DIR', '/root/data/iterations'),
help='Path to the directory for storing iterations (default: /root/data/iterations, env: SIA_ITERATIONS_DIR)'
)
parser.add_argument(
'--work-dir',
type=Path,
default=os.getenv('SIA_WORK_DIR', '/root'),
help='Path to the working directory (default: /root, env: SIA_WORK_DIR)'
default=os.getenv('SIA_WORK_DIR', '/root/desktop'),
help='Path to the working directory (default: /root/desktop, env: SIA_WORK_DIR)'
)
# Web server configuration
@@ -59,8 +59,8 @@ class Config:
parser.add_argument(
'--static-files',
type=Path,
default=self._parse_optional_path('SIA_STATIC_FILES', './static/'),
help='Path to static web files (default: ./static/, env: SIA_STATIC_FILES)'
default=self._parse_optional_path('SIA_STATIC_FILES', '/root/static/'),
help='Path to static web files (default: /root/static/, env: SIA_STATIC_FILES)'
)
# Local LLM configuration
@@ -73,8 +73,8 @@ class Config:
parser.add_argument(
'--local-model',
type=str,
default=os.getenv('SIA_LOCAL_MODEL', '/root/model/'),
help='Path to local model directory (default: /root/model/, env: SIA_LOCAL_MODEL)'
default=os.getenv('SIA_LOCAL_MODEL', '/root/models/current'),
help='Path to local model directory (default: /root/models/current, env: SIA_LOCAL_MODEL)'
)
parser.add_argument(
'--local-temperature',

7
sia/web_agent.py
View File

@@ -1,8 +1,9 @@
from collections import defaultdict
from datetime import datetime, timezone
from enum import Enum, auto
from sys import exit
from threading import Lock
from typing import Callable, Dict, List, Optional
from collections import defaultdict
from .base_agent import BaseAgent
from .command import Command
@@ -160,7 +161,9 @@ class WebAgent(BaseAgent):
if isinstance(parse_result, Command):
result = parse_result.execute(self._working_memory)
self._command_result = result
if not result.should_stop:
if result.should_stop:
exit(42)
else:
self._working_memory.update()
else:
parse_result.update()